Set theme to default if no theme set
This commit is contained in:
parent
1c7747497a
commit
a544dac974
1 changed files with 9 additions and 5 deletions
|
@ -5,16 +5,20 @@ client = MongoClient()
|
|||
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):
|
||||
username = username.lower()
|
||||
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):
|
||||
return accounts_db.find_one({'username': username})['displayname']
|
||||
|
||||
|
|
Loading…
Add table
Reference in a new issue