from cpl.database.abc import DbModelDaoABC
from model.user import User


class UserDao(DbModelDaoABC[User]):

    def __init__(self):
        DbModelDaoABC.__init__(self, __name__, User, "users")

        self.attribute(User.name, str)
        self.attribute(User.city_id, int, db_name="CityId")

        self.reference("city", "id", User.city_id, "city")