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))