33 lines
630 B
Python
33 lines
630 B
Python
from reimagine import rdb, conn
|
|
|
|
|
|
def get_one(cur):
|
|
res = cur.chunks[0]
|
|
return res[0] if len(res) else None
|
|
|
|
|
|
class BaseModel():
|
|
__db__ = None
|
|
__table__ = None
|
|
_rdb = rdb
|
|
_conn = conn
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def filter(cls, f, limit=None):
|
|
q = cls._rdb.db(cls.__db__).table(cls.__table__).filter(f)
|
|
if limit:
|
|
q.limit(int(limit))
|
|
return q.run(cls._conn)
|
|
|
|
|
|
class Site(BaseModel):
|
|
__db__ = 'lastdb'
|
|
__table__ = 'sites'
|
|
|
|
@classmethod
|
|
def get_by_name(cls, name):
|
|
return get_one(cls.filter({'name': name}, limit=1))
|