From a544dac97430279091a3b7190ec75714ba57dcfb Mon Sep 17 00:00:00 2001 From: SecretlyTaco Date: Sun, 25 Feb 2018 12:09:44 +0000 Subject: [PATCH] Set theme to default if no theme set --- backend/accounts.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/backend/accounts.py b/backend/accounts.py index d0f1530..b15bbb9 100644 --- a/backend/accounts.py +++ b/backend/accounts.py @@ -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']