added support for coomer.party

AlphaSlayer1964 · Dec 13, 2021 · 92c236b · 92c236b
1 parent e2e91eb
commit 92c236b
Show file tree

Hide file tree

Showing 4 changed files with 422 additions and 20 deletions.
diff --git a/.gitignore b/.gitignore
@@ -7,3 +7,4 @@ run.bat
 kemono-dl.py.backup
 kemono-dl_html.py.backup
 broken_hashes.log
+coomer.party_cookies.txt
diff --git a/src/arguments.py b/src/arguments.py
@@ -26,11 +26,27 @@ def get_args():
                     metavar="FILE", type=str, default=[],
                     help="File containing URLs to download, one URL per line.")
 
-    ap.add_argument("--favorite-users",
+    # ap.add_argument("--favorite-users",
+    #                 action='store_true', default=False,
+    #                 help="Downloads all favorite users. (Requires cookies while logged in)")
+
+    # ap.add_argument("--favorite-posts",
+    #                 action='store_true', default=False,
+    #                 help="Downloads all favorites posts. (Requires cookies while logged in)")
+
+    ap.add_argument("--kemono-favorite-users",
+                    action='store_true', default=False,
+                    help="Downloads all favorite users. (Requires cookies while logged in)")
+
+    ap.add_argument("--kemono-favorite-posts",
+                    action='store_true', default=False,
+                    help="Downloads all favorites posts. (Requires cookies while logged in)")
+
+    ap.add_argument("--coomer-favorite-users",
                     action='store_true', default=False,
                     help="Downloads all favorite users. (Requires cookies while logged in)")
 
-    ap.add_argument("--favorite-posts",
+    ap.add_argument("--coomer-favorite-posts",
                     action='store_true', default=False,
                     help="Downloads all favorites posts. (Requires cookies while logged in)")
 
@@ -148,13 +164,31 @@ def get_args():
         print('[Error] Only use one: --archive or --update')
         quit()
 
+    # if args['cookies']:
+    #     try:
+    #         args['cookies'] = MozillaCookieJar(args['cookies'])
+    #         args['cookies'].load()
+    #     except (LoadError, FileNotFoundError) as e:
+    #         print(e)
+    #         quit()
+
     if args['cookies']:
-        try:
-            args['cookies'] = MozillaCookieJar(args['cookies'])
-            args['cookies'].load()
-        except (LoadError, FileNotFoundError) as e:
-            print(e)
-            quit()
+        cookie_files = args['cookies'].split(',')
+        if len(cookie_files) == 1:
+            try:
+                args['cookies'] = MozillaCookieJar(args['cookies'])
+                args['cookies'].load()
+            except (LoadError, FileNotFoundError) as e:
+                print(e)
+                quit()
+        elif len(cookie_files) == 2:
+            try:
+                args['cookies'] = MozillaCookieJar()
+                args['cookies'].load(cookie_files[0])
+                args['cookies'].load(cookie_files[1])
+            except (LoadError, FileNotFoundError) as e:
+                print(e)
+                quit()
 
     if args['output']:
         if not os.path.exists(args['output']):