Null-safety fixes

aquaismissing · Jan 12, 2024 · 2580bd1 · 2580bd1
1 parent 0ef8a52
commit 2580bd1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/collectors/core.py b/collectors/core.py
@@ -166,6 +166,9 @@ async def unique_monthly():
         with open(config.CACHE_FILE_PATH, encoding='utf-8') as f:
             cache = json.load(f)
 
+        if cache.get('monthly_unique_players') is None:
+            cache['monthly_unique_players'] = data
+
         if data != cache.get('monthly_unique_players'):
             await send_alert('monthly_unique_players',
                              (cache['monthly_unique_players'], data))

diff --git a/main.py b/main.py
@@ -387,7 +387,7 @@ async def profile_info(client: BClient, session: UserSession, callback_query: Ca
     with open(config.CACHE_FILE_PATH, encoding='utf-8') as f:
         cache_file = json.load(f)
 
-    if cache_file['webapi'] != 'normal':
+    if cache_file.get('webapi') != 'normal':
         return await send_about_maintenance(client, session, callback_query)
 
     await callback_query.edit_message_text(session.locale.bot_choose_cmd,