PublicMirrors/you-get
1
0
Fork 0
mirror of https://github.com/soimort/you-get.git synced 2025-02-09 19:52:32 +03:00
Code Issues Actions Packages Projects Releases 1 Wiki Activity

[common] load_cookies(): copy cookies.sqlite to a temporary file in case database is locked (e.g., Firefox is running)

Browse Source
This commit is contained in:
Mort Yao 2019-03-03 18:28:58 +01:00
parent 2f9263c01f
commit 077b604748
No known key found for this signature in database
GPG Key ID: 07DA00CB78203251
1 changed files with 22 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/you_get/common.py
View File

@ -1276,9 +1276,8 @@ def download_main(download, download_playlist, urls, playlist, **kwargs):
def load_cookies(cookiefile): def load_cookies(cookiefile):
from http.cookiejar import Cookie
global cookies global cookies
try: if cookiefile.endswith('.txt'):
# MozillaCookieJar treats prefix '#HttpOnly_' as comments incorrectly! # MozillaCookieJar treats prefix '#HttpOnly_' as comments incorrectly!
# do not use its load() # do not use its load()
# see also: # see also:
@ -1287,6 +1286,7 @@ def load_cookies(cookiefile):
# - https://curl.haxx.se/libcurl/c/CURLOPT_COOKIELIST.html#EXAMPLE # - https://curl.haxx.se/libcurl/c/CURLOPT_COOKIELIST.html#EXAMPLE
#cookies = cookiejar.MozillaCookieJar(cookiefile) #cookies = cookiejar.MozillaCookieJar(cookiefile)
#cookies.load() #cookies.load()
from http.cookiejar import Cookie
cookies = cookiejar.MozillaCookieJar() cookies = cookiejar.MozillaCookieJar()
now = time.time() now = time.time()
ignore_discard, ignore_expires = False, False ignore_discard, ignore_expires = False, False
@ -1338,12 +1338,15 @@ def load_cookies(cookiefile):
continue continue
cookies.set_cookie(c) cookies.set_cookie(c)
except Exception: elif cookiefile.endswith(('.sqlite', '.sqlite3')):
import sqlite3 import sqlite3, shutil, tempfile
temp_dir = tempfile.gettempdir()
temp_cookiefile = os.path.join(temp_dir, 'temp_cookiefile.sqlite')
shutil.copy2(cookiefile, temp_cookiefile)
cookies = cookiejar.MozillaCookieJar() cookies = cookiejar.MozillaCookieJar()
con = sqlite3.connect(cookiefile) con = sqlite3.connect(temp_cookiefile)
cur = con.cursor() cur = con.cursor()
try:
cur.execute("""SELECT host, path, isSecure, expiry, name, value cur.execute("""SELECT host, path, isSecure, expiry, name, value
FROM moz_cookies""") FROM moz_cookies""")
for item in cur.fetchall(): for item in cur.fetchall():
@ -1354,8 +1357,9 @@ def load_cookies(cookiefile):
None, {}, None, {},
) )
cookies.set_cookie(c) cookies.set_cookie(c)
except Exception:
pass else:
log.e('[error] unsupported cookies format')
# TODO: Chromium Cookies # TODO: Chromium Cookies
# SELECT host_key, path, secure, expires_utc, name, encrypted_value # SELECT host_key, path, secure, expires_utc, name, encrypted_value
# FROM cookies # FROM cookies