kate/Tweeder
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Set theme to default if no theme set

Browse source
This commit is contained in:
SecretlyTaco 2018-02-25 12:09:44 +00:00
parent 1c7747497a
commit a544dac974
1 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
backend/accounts.py
View file

@ -5,16 +5,20 @@ client = MongoClient()
accounts_db = client.tweeder.accounts accounts_db = client.tweeder.accounts
def get_theme(username):
username = username.lower()
return accounts_db.find_one({"username": username})['theme']
def set_theme(username, value): def set_theme(username, value):
username = username.lower() username = username.lower()
accounts_db.update_one({'username': username}, {"$set": {"theme": value}}) accounts_db.update_one({'username': username}, {"$set": {"theme": value}})
def get_theme(username):
username = username.lower()
if "themes" in accounts_db.find_one({"username": username}).keys():
return accounts_db.find_one({"username": username})['theme']
else:
set_theme(username, "default")
return "default"
def get_display_name(username): def get_display_name(username):
return accounts_db.find_one({'username': username})['displayname'] return accounts_db.find_one({'username': username})['displayname']