add_account function

1 files changed, 10 insertions, 0 deletions

diff --git a/youtube/accounts.py b/youtube/accounts.py
index a1865c0..db33496 100644
--- a/youtube/accounts.py
+++ b/youtube/accounts.py
@@ -20,6 +20,16 @@ def save_accounts():
     with open(os.path.join(settings.data_dir, 'accounts.txt'), 'w', encoding='utf-8') as f:
         f.write(json.dumps(to_save))
 
+def add_account(username, password, save):
+    cookie_jar = http.cookiejar.LWPCookieJar()
+    _login(username, password, cookie_jar)
+    accounts.append({
+        "username": username,
+        "save":save,
+        "cookies":cookie_jar.as_lwp_str(ignore_discard=False, ignore_expires=False),
+    })
+
+
 # ---------------------------------
 # Code ported from youtube-dl
 # ---------------------------------