from models.user import User
from extensions import db

class UserRepository:

    @staticmethod
    def find_by_username(username: str):
        return User.query.filter_by(username=username).first()

    @staticmethod
    def find_by_id(user_id: int):
        return User.query.get(user_id)

    @staticmethod
    def create(username: str, password: str):
        user = User(username=username, password=password)
        db.session.add(user)
        db.session.commit()
        return user