mirror of
https://github.com/dancojocaru2000/foxbank.git
synced 2025-02-23 12:29:35 +02:00
25 lines
703 B
Python
25 lines
703 B
Python
|
from functools import wraps
|
||
|
|
|
||
|
|
import db
|
||
|
|
import models
|
||
|
|
|
||
|
|
def get_db(fn):
|
||
|
|
@wraps(fn)
|
||
|
|
def wrapper(*args, **kargs):
|
||
|
|
return fn(db.get(), *args, **kargs)
|
||
|
|
return wrapper
|
||
|
|
|
||
|
|
@get_db
|
||
|
|
def get_user(db: db.get_return, username: str|None = None, user_id: int|None = None) -> models.User | None:
|
||
|
|
cur = db.cursor()
|
||
|
|
if username is not None:
|
||
|
|
cur.execute('select * from users where username=?', (username,))
|
||
|
|
elif user_id is not None:
|
||
|
|
cur.execute('select * from users where id=?', (user_id,))
|
||
|
|
else:
|
||
|
|
raise Exception('Neither username or user_id passed')
|
||
|
|
result = cur.fetchone()
|
||
|
|
if result is None:
|
||
|
|
return None
|
||
|
|
return models.User.from_query(result)
|