26 lines
480 B
Python
26 lines
480 B
Python
from dataclasses import dataclass
|
|
from uuid import UUID
|
|
|
|
from .db import get_db
|
|
|
|
|
|
@dataclass
|
|
class User:
|
|
user_id: UUID
|
|
nickname: str
|
|
|
|
|
|
def get_user(user_id: str) -> User | None:
|
|
try:
|
|
user_id = UUID(user_id)
|
|
except:
|
|
return None
|
|
|
|
db = get_db()
|
|
cur = db.execute("SELECT * FROM users WHERE user_id = ?", [str(user_id)])
|
|
user_row = cur.fetchone()
|
|
if user_row is None:
|
|
return None
|
|
|
|
return User(UUID(user_row[0]), user_row[1])
|