Server-side http logic for adding account

2 files changed, 15 insertions, 1 deletions

diff --git a/youtube/accounts.py b/youtube/accounts.py
index a7461c0..9696899 100644
--- a/youtube/accounts.py
+++ b/youtube/accounts.py
@@ -23,11 +23,12 @@ def save_accounts():
 
 def add_account(username, password, save):
     cookie_jar = http.cookiejar.LWPCookieJar()
-    _login(username, password, cookie_jar)
+    condition = _login(username, password, cookie_jar)
     accounts[username] = {
         "save":save,
         "cookies":cookie_jar.as_lwp_str(ignore_discard=False, ignore_expires=False),
     }
+    return condition
 
 def cookie_jar_from_lwp_str(lwp_str):
     cookie_jar = http.cookiejar.LWPCookieJar()
diff --git a/youtube/youtube.py b/youtube/youtube.py
index a08d7b5..418e707 100644
--- a/youtube/youtube.py
+++ b/youtube/youtube.py
@@ -116,6 +116,19 @@ def youtube(env, start_response):
                 start_response('303 See Other',  (('Location', common.URL_ORIGIN + '/comments?ctoken=' + comments.make_comment_ctoken(video_id, sort=1)),) )
             return ''
 
+        elif path == "/login":
+            if 'save' in fields and fields['save'][0] == "on":
+                save_account = True
+            else:
+                save_account = False
+
+            if accounts.add_account(fields['username'][0], fields['password'][0], save_account ):
+                start_response('200 OK',  () )
+                return b'Account successfully added'
+            else:
+                start_response('200 OK',  () )
+                return b'Failed to add account'
+
         else:
             start_response('404 Not Found', ())
             return b'404 Not Found'