from __future__ import absolute_import
from dostadmin import db
from dostadmin.mixins import TimestampMixin


class Parent(TimestampMixin, db.Model):
    __tablename__ = "parent"

    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey("users.id"))

    children = db.relationship(
        "Child", backref="parent", primaryjoin="Child.parent_id == Parent.id"
    )

    @classmethod
    def add_parent(cls, user_id):
        parent = Parent(user_id=user_id)
        db.session.add(parent)
        db.session.commit()
        return parent