From e8a2ba6de861065662dcf334dc1b9c303c66f033 Mon Sep 17 00:00:00 2001
From: Chanjian Zhu <daya0576@gmail.com>
Date: Tue, 8 Oct 2024 14:31:07 +0800
Subject: [PATCH] Add controller logs

---
 beaverhabits/utils.py | 1 +
 beaverhabits/views.py | 4 +++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/beaverhabits/utils.py b/beaverhabits/utils.py
index dba928c9..80748d0e 100644
--- a/beaverhabits/utils.py
+++ b/beaverhabits/utils.py
@@ -19,6 +19,7 @@ async def fetch_user_timezone() -> None:
 
 
 async def get_or_create_user_timezone() -> str:
+    logger.info("Getting user timezone...")
     if timezone := app.storage.user.get(TIME_ZONE_KEY):
         logger.info(f"User timezone from storage: {timezone}")
         return timezone
diff --git a/beaverhabits/views.py b/beaverhabits/views.py
index b3b7793f..a02964d8 100644
--- a/beaverhabits/views.py
+++ b/beaverhabits/views.py
@@ -1,12 +1,13 @@
 import datetime
 import json
 import random
-import time
 from typing import List
 
 from fastapi import HTTPException
 from nicegui import ui
 
+from beaverhabits.logging import logger
+
 from beaverhabits.app.db import User
 from beaverhabits.storage import get_user_dict_storage, session_storage
 from beaverhabits.storage.dict import DAY_MASK, DictHabitList
@@ -75,6 +76,7 @@ async def get_user_habit(user: User, habit_id: str) -> Habit:
 async def get_or_create_user_habit_list(
     user: User, days: List[datetime.date]
 ) -> HabitList:
+    logger.info("Getting user habit list")
     habit_list = await get_user_habit_list(user)
     if habit_list is not None:
         return habit_list