[英]There is no output to JSON file in my python script
我有一個應該執行SQL查詢的python腳本。 但是,腳本未寫入JSON文件,當我打開JSON文件時,該腳本為空。 但是,腳本運行時我沒有出現任何錯誤。 我嘗試修復此錯誤無濟於事。 非常感謝您的幫助! 我添加了certificate_store = cursor.fetchall(),但即使添加也似乎無法修復我的代碼。
碼
from __future__ import print_function
try:
import psycopg2
except ImportError:
raise ImportError('\n\033[33mpsycopg2 library missing. pip install psycopg2\033[1;m\n')
sys.exit(1)
import re
import sys
import json
import pprint
import time
import psycopg2.extras
outfilepath = "crtsh_output/crtsh_flat_file"
DB_HOST = 'crt.sh'
DB_NAME = 'certwatch'
DB_USER = 'guest'
DELAY = 0
def connect_to_db():
start = 0
offset = 10
flag = True
while flag:
filepath = 'forager.txt'
print('am i stuck')
with open(filepath, "at+") as fp, open(outfilepath, "wt+") as outfile:
try:
for cnt, domain_name in enumerate(fp):
print("Line {}: {}".format(cnt, domain_name))
print(domain_name)
domain_name = domain_name.rstrip()
SQL = '''
SELECT c.id, x509_commonName(c.certificate) as common_name,
x509_issuerName(c.certificate) as issuer_name,
x509_notBefore(c.certificate) as not_before,
x509_notAfter(c.certificate) as not_after,
x509_keyAlgorithm(c.certificate) as key_algorithm,
x509_keySize(c.certificate) as key_size,
x509_serialNumber(c.certificate) as serial_number,
x509_signatureHashAlgorithm(c.certificate) as signature_hash_algorithm,
x509_signatureKeyAlgorithm(c.certificate) as signature_key_algorithm,
x509_subjectName(c.certificate) as subject_name,
x509_name(c.certificate) as name,
x509_altNames(c.certificate) as alt_names
FROM certificate c, certificate_identity ci
WHERE c.id = ci.certificate_id
AND ci.name_type = 'dNSName'
AND lower(ci.name_value) = lower(%s)
AND x509_notAfter(c.certificate) > statement_timestamp()
'''
with psycopg2.connect("dbname=certwatch user=guest host=crt.sh".format(DB_NAME, DB_USER, DB_HOST)) as conn:
cursor = conn.cursor(cursor_factory=psycopg2.extras.NamedTupleCursor)
cursor.execute(SQL, (domain_name,))
certificate_store = cursor.fetchall()
for certificate in certificate_store:
print(certificate.id)
print(certificate.common_name)
print(certificate.issuer_name)
outfile.write(json.dumps(certificate, sort_keys=True, indent=4, default=str, ensure_ascii=False))
# query db with start and offset
# unique_domains = cursor.fetchall()
# if not unique_domains:
# flag = False
# else:
# do processing with your data
# pprint.pprint(unique_domains)
# outfile.write(json.dumps(unique_domains, sort_keys=True, indent=4, default=str, ensure_ascii = False))
# offset += limit
except Exception as error:
print(str(error))
if __name__ == "__main__":
connect_to_db()
寫入文件的行:
outfile.write(...)
受if
條件保護:
if not unique_domains:
flag = False
else:
....
outfile.write(...)
因此unique_domains
必須為True
ish。
但是你的台詞
unique_domains = cursor.fetchall()
是不是數據庫的塊內with
條件:
with psycopg2.connect("dbname=certwatch user=guest host=crt.sh".format(DB_NAME, DB_USER, DB_HOST)) as conn:
另外,您已經獲取(並在打印后丟棄)了
for certificate in cursor.fetchall():
所以,除非你將無法再獲取它們.execute
再次查詢。 一個更好的主意是將數據存儲在第一次提取中。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.