adulau/cpe-guesser
1
0
Fork 0
mirror of https://github.com/cve-search/cpe-guesser.git synced 2024-11-24 15:57:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

fix: [clean-up] use format() instead concatenation

Browse source
This commit is contained in:
Alexandre Dulaunoy 2021-09-21 20:31:32 +02:00
parent 6cf6e05b75
commit 722863c103
Signed by: adulau
GPG key ID: 09E2CD4944E6CBCD
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
bin/import.py
View file

@ -58,7 +58,7 @@ class CPEHandler( xml.sax.ContentHandler ):
self.record = {} self.record = {}
self.itemcount += 1 self.itemcount += 1
if self.itemcount % 5000 == 0: if self.itemcount % 5000 == 0:
print ("... " + str(self.itemcount) + " items processed (" + str(self.wordcount) + " words)") print ("... {} items processed ({} words)".format(str(self.itemcount), str(self.wordcount)))
def CPEExtractor( cpe=None ): def CPEExtractor( cpe=None ):
@ -106,7 +106,7 @@ if __name__ == '__main__':
print(e) print(e)
sys.exit(1) sys.exit(1)
print("Uncompressing " + cpe_path + ".gz ...") print("Uncompressing {}.gz ...".format(cpe_path))
try: try:
with gzip.open(cpe_path + ".gz", 'rb') as cpe_gz: with gzip.open(cpe_path + ".gz", 'rb') as cpe_gz:
with open(cpe_path, 'wb') as cpe_xml: with open(cpe_path, 'wb') as cpe_xml:
@ -117,10 +117,10 @@ if __name__ == '__main__':
sys.exit(1) sys.exit(1)
elif os.path.isfile(cpe_path): elif os.path.isfile(cpe_path):
print("Using existing file " + cpe_path + " ...") print("Using existing file {} ...".format(cpe_path))
if rdb.dbsize() > 0: if rdb.dbsize() > 0:
print("Flushing " + str(rdb.dbsize()) + " keys from the database...") print("Flushing {} keys from the database...".format(str(rdb.dbsize())))
rdb.flushdb() rdb.flushdb()
print("Populating the database (please be patient)...") print("Populating the database (please be patient)...")
@ -128,4 +128,4 @@ if __name__ == '__main__':
Handler = CPEHandler() Handler = CPEHandler()
parser.setContentHandler( Handler ) parser.setContentHandler( Handler )
parser.parse(cpe_path) parser.parse(cpe_path)
print("Done! " + str(rdb.dbsize()) + " keys inserted.") print("Done! {} keys inserted.".format(str(rdb.dbsize())))