mikf/gallery-dl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

cache archive keys generated in __contains__() (#524)

Browse Source 
To avoid writing a different key to the archive than what was checked
against before the file download.
This commit is contained in:
Mike Fährmann 2019-12-20 16:43:08 +01:00
parent bf658fd84b
commit 58391d492d
No known key found for this signature in database
GPG Key ID: 5680CA389D365A88
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
gallery_dl/util.py
View File

@ -748,13 +748,13 @@ class DownloadArchive():
def __contains__(self, kwdict): def __contains__(self, kwdict):
"""Return True if the item described by 'kwdict' exists in archive""" """Return True if the item described by 'kwdict' exists in archive"""
key = self.keygen(kwdict) key = kwdict["_archive_key"] = self.keygen(kwdict)
self.cursor.execute( self.cursor.execute(
"SELECT 1 FROM archive WHERE entry=? LIMIT 1", (key,)) "SELECT 1 FROM archive WHERE entry=? LIMIT 1", (key,))
return self.cursor.fetchone() return self.cursor.fetchone()
def add(self, kwdict): def add(self, kwdict):
"""Add item described by 'kwdict' to archive""" """Add item described by 'kwdict' to archive"""
key = self.keygen(kwdict) key = kwdict.get("_archive_key") or self.keygen(kwdict)
self.cursor.execute( self.cursor.execute(
"INSERT OR IGNORE INTO archive VALUES (?)", (key,)) "INSERT OR IGNORE INTO archive VALUES (?)", (key,))