Add cookies saved in home cache directory

gdown/download.py

@@ -1,5 +1,6 @@
 from __future__ import print_function
 
+import json
 import os
 import os.path as osp
 import re
@@ -16,6 +17,7 @@
 from .parse_url import parse_url
 
 CHUNK_SIZE = 512 * 1024  # 512KB
+home = osp.expanduser("~")
 
 
 if hasattr(textwrap, "indent"):
@@ -81,6 +83,17 @@ def download(url, output=None, quiet=False, proxy=None, speed=None):
     url_origin = url
     sess = requests.session()
 
+    # Load cookies
+    cache_dir = osp.join(home, ".cache", "gdown")
+    if not osp.exists(cache_dir):
+        os.makedirs(cache_dir)
+    cookies_file = osp.join(cache_dir, "cookies.json")
+    if osp.exists(cookies_file):
+        with open(cookies_file) as f:
+            cookies = json.load(f)
+        for k, v in cookies:
+            sess.cookies[k] = v
+
     if proxy is not None:
         sess.proxies = {"http": proxy, "https": proxy}
         print("Using proxy:", proxy, file=sys.stderr)
@@ -96,6 +109,10 @@ def download(url, output=None, quiet=False, proxy=None, speed=None):
             print(e, file=sys.stderr)
             return
 
+        # Save cookies
+        with open(cookies_file, "w") as f:
+            json.dump(sess.cookies.items(), f, indent=2)
+
         if "Content-Disposition" in res.headers:
             # This is the file
             break