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