common update

repositories/user.py

@@ -0,0 +1,47 @@
+from typing import Optional
+from infra.db import postgres
+
+class UserRepository:
+    def __init__(self):
+        self.exp_table = postgres.get_table('user_exp')
+        self.voice_table = postgres.get_table('voice_time')
+
+    async def get_exp(self, user_id: int) -> int:
+        """Get user's experience points from database"""
+        try:
+            response = self.exp_table.select('exp_points').eq('user_id', user_id).execute()
+            if response.data:
+                return response.data[0]['exp_points']
+            return 0
+        except Exception as e:
+            print(f"Error getting user exp: {e}")
+            return 0
+
+    async def update_exp(self, user_id: int, exp_points: int) -> bool:
+        """Update user's experience points in database"""
+        try:
+            response = self.exp_table.upsert({'user_id': user_id, 'exp_points': exp_points}).execute()
+            return len(response.data) > 0
+        except Exception as e:
+            print(f"Error updating user exp: {e}")
+            return False
+
+    async def get_voice_time(self, user_id: int) -> Optional[str]:
+        """Get user's voice time from database"""
+        try:
+            response = self.voice_table.select('last_join_time').eq('user_id', user_id).execute()
+            if response.data:
+                return response.data[0]['last_join_time']
+            return None
+        except Exception as e:
+            print(f"Error getting user voice time: {e}")
+            return None
+
+    async def update_voice_time(self, user_id: int, join_time: str) -> bool:
+        """Update user's voice time in database"""
+        try:
+            response = self.voice_table.upsert({'user_id': user_id, 'last_join_time': join_time}).execute()
+            return len(response.data) > 0
+        except Exception as e:
+            print(f"Error updating user voice time: {e}")
+            return False