heckyel/yt-local
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Make accounts.txt more human-readable

Browse Source
This commit is contained in:
James Taylor 2018-12-26 18:45:50 -08:00
parent 9a386145ff
commit dfbd7778d1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
youtube/accounts.py
View File

@ -19,7 +19,7 @@ except FileNotFoundError:
def save_accounts(): def save_accounts():
to_save = {username: account for username, account in accounts.items() if account['save']} to_save = {username: account for username, account in accounts.items() if account['save']}
with open(os.path.join(settings.data_dir, 'accounts.txt'), 'w', encoding='utf-8') as f: with open(os.path.join(settings.data_dir, 'accounts.txt'), 'w', encoding='utf-8') as f:
f.write(json.dumps(to_save)) f.write(json.dumps(to_save, indent=4))
def add_account(username, password, save): def add_account(username, password, save):
cookiejar = http.cookiejar.LWPCookieJar() cookiejar = http.cookiejar.LWPCookieJar()
@ -27,7 +27,7 @@ def add_account(username, password, save):
if successful: if successful:
accounts[username] = { accounts[username] = {
"save":save, "save":save,
"cookies":cookiejar.as_lwp_str(ignore_discard=False, ignore_expires=False), "cookies":cookiejar.as_lwp_str(ignore_discard=False, ignore_expires=False).split('\n'),
} }
if save: if save:
save_accounts() save_accounts()
@ -41,7 +41,7 @@ def cookiejar_from_lwp_str(lwp_str):
return cookiejar return cookiejar
def account_cookiejar(username): def account_cookiejar(username):
return cookiejar_from_lwp_str(accounts[username]['cookies']) return cookiejar_from_lwp_str('\n'.join(accounts[username]['cookies']))
def get_account_login_page(query_string): def get_account_login_page(query_string):
style = ''' style = '''