Fix crash when text attr is not avialable in SHOW_TOTAL_CODE_TIME

anmol098 · Nov 21, 2023 · cf8510c · cf8510c
1 parent 02827d6
commit cf8510c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sources/main.py b/sources/main.py
@@ -160,8 +160,8 @@ async def get_stats() -> str:
 
     if EM.SHOW_TOTAL_CODE_TIME:
         DBM.i("Adding total code time info...")
-        data = await DM.get_remote_json("waka_all")
-        stats += f"![Code Time](http://img.shields.io/badge/{quote('Code Time')}-{quote(str(data['data']['text']))}-blue)\n\n"
+        code_time = data["data"]["text"] if hasattr(data["data"], "text") else data["message"]
+        stats += f"![Code Time](http://img.shields.io/badge/{quote('Code Time')}-{quote(str(code_time))}-blue)\n\n"
 
     if EM.SHOW_PROFILE_VIEWS:
         DBM.i("Adding profile views info...")