def addCookie(self, cookie, expires):
db.execute(
"""
insert into usercookie (userid, cookie, expires)
values (:userid, :cookie, :expires)
""",
userid=self.id,
cookie=cookie,
expires=expires,
)
def getAllUsers(cls):
result = db.execute("""
select u.id, u.name, u.namecss, u.hash, u.created, u.lastlogin, u.status
from user u order by u.id
""")
models = [User(**row) for row in result.fetchall()]
return models
def getOnlineModels(cls):
result = db.execute("""
select u.id, u.name, u.namecss, u.hash, u.created, u.lastlogin, u.status
from user u where u.usertype = 'model' and u.status = 'online'
""")
models = [User(**row) for row in result.fetchall()]
models = sorted(models, key=attrgetter('score', 'name'))
return models
def save(self):
result = db.execute(
"""
update user set namecss=:namecss
where id=:id
""",
**self.__dict__
)
def getByName(cls, name):
result = db.execute("""
select u.id, u.name, u.namecss, u.hash, u.created, u.lastlogin, u.status
from user u where u.name = :name
""", name=name)
row = result.first()
if not row:
return None
return User(**row)
def getAllUsers(cls):
result = db.execute(
"""
select u.id, u.name, u.namecss, u.hash, u.created, u.lastlogin, u.status
from user u order by u.id
"""
)
models = [User(**row) for row in result.fetchall()]
return models
def getByCookie(cls, cookie):
result = db.execute("""
select u.id, u.name, u.namecss, u.hash, u.created, u.lastlogin, u.status
from user u inner join usercookie uc on uc.userid = u.id
where uc.cookie = :cookie and uc.expires > datetime()
""", cookie=cookie)
row = result.first()
if not row:
return None
return User(**row)
def getOnlineModels(cls):
result = db.execute(
"""
select u.id, u.name, u.namecss, u.hash, u.created, u.lastlogin, u.status
from user u where u.usertype = 'model' and u.status = 'online'
"""
)
models = [User(**row) for row in result.fetchall()]
models = sorted(models, key=attrgetter("score", "name"))
return models
def addUser(cls, name, password, namecss=None, usertype=None, status=None):
result = db.execute("""
insert into user (name, hash, namecss, usertype, status, created)
values(:name, :hash, :namecss, :usertype, :status, datetime())
""",
name=name,
hash=bcrypt.hashpw(password, bcrypt.gensalt()),
namecss=namecss,
usertype=usertype,
status=status)
return User.getByName(name)
def getByName(cls, name):
result = db.execute(
"""
select u.id, u.name, u.namecss, u.hash, u.created, u.lastlogin, u.status
from user u where u.name = :name
""",
name=name,
)
row = result.first()
if not row:
return None
return User(**row)
def getByCookie(cls, cookie):
result = db.execute(
"""
select u.id, u.name, u.namecss, u.hash, u.created, u.lastlogin, u.status
from user u inner join usercookie uc on uc.userid = u.id
where uc.cookie = :cookie and uc.expires > datetime()
""",
cookie=cookie,
)
row = result.first()
if not row:
return None
return User(**row)
def addUser(cls, name, password, namecss=None, usertype=None, status=None):
result = db.execute(
"""
insert into user (name, hash, namecss, usertype, status, created)
values(:name, :hash, :namecss, :usertype, :status, datetime())
""",
name=name,
hash=bcrypt.hashpw(password, bcrypt.gensalt()),
namecss=namecss,
usertype=usertype,
status=status,
)
return User.getByName(name)
def addCookie(self, cookie, expires):
db.execute("""
insert into usercookie (userid, cookie, expires)
values (:userid, :cookie, :expires)
""", userid=self.id, cookie=cookie, expires=expires)
def save(self):
result = db.execute("""
update user set namecss=:namecss
where id=:id
""", **self.__dict__)
