from __future__ import absolute_import
from flask_sqlalchemy.query import Query as BaseQuery
from dostadmin import db
from dostadmin.mixins import TimestampMixin


class GroupsQuery(BaseQuery):
    def get_details_by_id(self, group_id):
        return self.filter(Groups.id == group_id).first()


class Groups(TimestampMixin, db.Model):
    __tablename__ = "groups"
    query_class = GroupsQuery

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(50))