diff options
| -rw-r--r-- | youtube/accounts.py | 3 | ||||
| -rw-r--r-- | youtube/youtube.py | 13 |
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' |