migrate from supabase to postgres

repositories/home_debt.py

@@ -1,57 +1,69 @@
 from typing import List, Optional
-from models.home_debt import DiscordHomeDebt
+from sqlalchemy import select
+from sqlalchemy.orm import selectinload
+
+from models.home_debt import HomeDebt
 from infra.db import postgres
 
 class HomeDebtRepository:
     def __init__(self):
-        self.table = postgres.get_table('discord_home_debt')
+        self.Session = postgres.get_sessionmaker()
 
-    async def get(self, discord_user_id: int) -> Optional[DiscordHomeDebt]:
+    async def get(self, discord_user_id: int) -> Optional[HomeDebt]:
         """Lấy thông tin thành viên"""
         try:
-            response = self.table.select('*').eq('user_id', discord_user_id).execute()
-            if response.data:
-                return DiscordHomeDebt(**response.data[0])
-            return None
+            async with self.Session() as session:
+                stmt = select(HomeDebt).where(HomeDebt.user_id == discord_user_id)
+                result = await session.execute(stmt)
+                return result.scalar_one_or_none()
         except Exception as e:
             print(f"Error getting user: {e}")
             return None
         
-    async def get_other(self, discord_user_id: int) -> Optional[DiscordHomeDebt]:
+    async def get_other(self, discord_user_id: int) -> Optional[HomeDebt]:
         """Lấy thông tin thành viên khác"""
         try:
-            response = self.table.select('*').neq('user_id', discord_user_id).execute()
-            if response.data:
-                return DiscordHomeDebt(**response.data[0])
-            return None
+            async with self.Session() as session:
+                stmt = select(HomeDebt).where(HomeDebt.user_id != discord_user_id).limit(1)
+                result = await session.execute(stmt)
+                return result.scalar_one_or_none()
         except Exception as e:
             print(f"Error getting other member: {e}")
             return None
 
-    async def create_home_debt(self, user_id: int, value: int) -> Optional[DiscordHomeDebt]:
+    async def create_home_debt(self, user_id: int, value: int) -> Optional[HomeDebt]:
         """Tạo mới khoản nợ"""
         try:
-            home_debt = DiscordHomeDebt(user_id=user_id, value=value)
-            response = self.table.insert(home_debt.dict(exclude_none=True)).execute()
-            return DiscordHomeDebt(**response.data[0])
+            async with self.Session() as session:
+                home_debt = HomeDebt(user_id=user_id, value=value)
+                session.add(home_debt)
+                await session.commit()
+                await session.refresh(home_debt)
+                return home_debt
         except Exception as e:
             print(f"Error creating home debt: {e}")
             return None
 
-    async def update_home_debt(self, home_debt: DiscordHomeDebt) -> Optional[DiscordHomeDebt]:
+    async def update_home_debt(self, home_debt: HomeDebt) -> Optional[HomeDebt]:
         """Cập nhật khoản nợ"""
         try:
-            response = self.table.update(home_debt.to_dict()).eq('user_id', home_debt.user_id).execute()
-            return DiscordHomeDebt(**response.data[0])
+            async with self.Session() as session:
+                # Merge object vào session để cập nhật
+                merged_debt = await session.merge(home_debt)
+                await session.commit()
+                await session.refresh(merged_debt)
+                return merged_debt
         except Exception as e:
             print(f"Error updating home debt: {e}")
             return None
 
-    async def get_all(self) -> List[DiscordHomeDebt]:
+    async def get_all(self) -> List[HomeDebt]:
         """Lấy tất cả khoản nợ"""
         try:
-            response = self.table.select('*').execute()
-            return [DiscordHomeDebt(**home_debt) for home_debt in response.data]
+            async with self.Session() as session:
+                stmt = select(HomeDebt)
+                result = await session.execute(stmt)
+                return result.scalars().all()
         except Exception as e:
             print(f"Error getting all home debts: {e}")
             return []