diff options
| author | James Taylor <user234683@users.noreply.github.com> | 2018-12-26 02:21:20 -0800 |
|---|---|---|
| committer | James Taylor <user234683@users.noreply.github.com> | 2018-12-26 02:21:20 -0800 |
| commit | b321b5fc6484c38d861530e6b89405b062e32459 (patch) | |
| tree | 17857ea71377604b887e1242ff596a6817c165d8 | |
| parent | b4aa7cecbc2433a9a734c1896bf24ed57b141884 (diff) | |
| download | yt-local-b321b5fc6484c38d861530e6b89405b062e32459.tar.lz yt-local-b321b5fc6484c38d861530e6b89405b062e32459.tar.xz yt-local-b321b5fc6484c38d861530e6b89405b062e32459.zip | |
Server-side http logic for adding account
| -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' |