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']