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

[cookies:firefox] support using domain + container filters together

Browse Source
This commit is contained in:
Mike Fährmann
2024-09-16 14:47:23 +02:00
parent a9dc7a6f98
commit 73f833d08a
2 changed files with 17 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
gallery_dl/cookies.py
View File

@@ -50,21 +50,27 @@ def load_cookies_firefox(cookiejar, profile=None, container=None, domain=None):
sql = ("SELECT name, value, host, path, isSecure, expiry "
"FROM moz_cookies")
parameters = ()
conditions = []
parameters = []
if container_id is False:
sql += " WHERE NOT INSTR(originAttributes,'userContextId=')"
conditions.append("NOT INSTR(originAttributes,'userContextId=')")
elif container_id:
sql += " WHERE originAttributes LIKE ? OR originAttributes LIKE ?"
conditions.append(
"originAttributes LIKE ? OR originAttributes LIKE ?")
uid = "%userContextId={}".format(container_id)
parameters = (uid, uid + "&%")
elif domain:
parameters += (uid, uid + "&%")
if domain:
if domain[0] == ".":
sql += " WHERE host == ? OR host LIKE ?"
parameters = (domain[1:], "%" + domain)
conditions.append("host == ? OR host LIKE ?")
parameters += (domain[1:], "%" + domain)
else:
sql += " WHERE host == ? OR host == ?"
parameters = (domain, "." + domain)
conditions.append("host == ? OR host == ?")
parameters += (domain, "." + domain)
if conditions:
sql = "{} WHERE ( {} )".format(sql, " ) AND ( ".join(conditions))
set_cookie = cookiejar.set_cookie
for name, value, domain, path, secure, expires in db.execute(