113 lines
4.3 KiB
Python
113 lines
4.3 KiB
Python
from typing import List, Optional
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import select, update, insert
|
|
from sqlalchemy.dialects.postgresql import insert as pg_insert
|
|
|
|
from models.score import Score
|
|
from infra.db import postgres
|
|
|
|
class ScoreRepository:
|
|
def __init__(self):
|
|
self.Session = postgres.get_sessionmaker()
|
|
|
|
async def get(self, user_id: int) -> Optional[Score]:
|
|
"""Lấy thông tin thành viên"""
|
|
try:
|
|
async with self.Session() as session:
|
|
stmt = select(Score).where(Score.user_id == 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 create(self, user_id: int, point: int) -> Optional[Score]:
|
|
"""Tạo thông tin thành viên"""
|
|
try:
|
|
async with self.Session() as session:
|
|
currency = Score(user_id=user_id, point=point)
|
|
session.add(currency)
|
|
await session.commit()
|
|
await session.refresh(currency)
|
|
return currency
|
|
except Exception as e:
|
|
print(f"Error creating user: {e}")
|
|
return None
|
|
|
|
async def update(self, user_id: int, point: int) -> Optional[Score]:
|
|
"""Cập nhật thông tin thành viên"""
|
|
try:
|
|
async with self.Session() as session:
|
|
stmt = (
|
|
update(Score)
|
|
.where(Score.user_id == user_id)
|
|
.values(point=point)
|
|
.returning(Score)
|
|
)
|
|
result = await session.execute(stmt)
|
|
await session.commit()
|
|
return result.scalar_one_or_none()
|
|
except Exception as e:
|
|
print(f"Error updating user: {e}")
|
|
return None
|
|
|
|
async def get_all(self) -> List[Score]:
|
|
"""Lấy tất cả thông tin thành viên"""
|
|
try:
|
|
async with self.Session() as session:
|
|
stmt = select(Score).order_by(Score.point.desc())
|
|
result = await session.execute(stmt)
|
|
return result.scalars().all()
|
|
except Exception as e:
|
|
print(f"Error getting all users: {e}")
|
|
return []
|
|
|
|
async def get_all_with_point(self) -> List[Score]:
|
|
"""Lấy tất cả thông tin thành viên và số dư"""
|
|
try:
|
|
async with self.Session() as session:
|
|
stmt = select(Score)
|
|
result = await session.execute(stmt)
|
|
return result.scalars().all()
|
|
except Exception as e:
|
|
print(f"Error getting all users with point: {e}")
|
|
return []
|
|
|
|
# Get all with sort by point
|
|
async def get_all_with_sort_by_point(self) -> List[Score]:
|
|
"""Lấy tất cả thông tin thành viên và sắp xếp theo số dư"""
|
|
try:
|
|
async with self.Session() as session:
|
|
stmt = select(Score).order_by(Score.point.desc())
|
|
result = await session.execute(stmt)
|
|
return result.scalars().all()
|
|
except Exception as e:
|
|
print(f"Error getting all users with sort by point: {e}")
|
|
return []
|
|
|
|
async def upsert_or_increment_point(self, user_id: str, user_name: str, amount: int) -> Optional[int]:
|
|
try:
|
|
async with self.Session() as session:
|
|
# Sử dụng PostgreSQL UPSERT (ON CONFLICT)
|
|
stmt = pg_insert(Score).values(
|
|
user_id=int(user_id),
|
|
user_name=user_name,
|
|
point=amount
|
|
)
|
|
stmt = stmt.on_conflict_do_update(
|
|
index_elements=['user_id'],
|
|
set_=dict(
|
|
point=Score.point + amount,
|
|
user_name=stmt.excluded.user_name
|
|
)
|
|
).returning(Score.point)
|
|
|
|
result = await session.execute(stmt)
|
|
await session.commit()
|
|
row = result.first()
|
|
return row[0] if row else None
|
|
except Exception as e:
|
|
print(f"Error upserting point for user {user_id}: {e}")
|
|
return None
|