from sqlalchemy import String

from api.database import (Model, SurrogatePK, db,
                          Column, relationship)

registration_target_assoc = db.Table(
    "registration_target",
    db.Column("registration", db.Integer(), db.ForeignKey("target.id")),
    db.Column("target", db.Integer(), db.ForeignKey("registration.id"))
)


class Registration(SurrogatePK, Model):
    __tablename__ = "registration"
    name = Column(String(255), unique=True, nullable=False)
    token = Column(String(255), nullable=False)
    targets = relationship(
        "Target", secondary=registration_target_assoc, backref=db.backref("registrations"))

    def __init__(self, **kwargs):
        super(Registration, self).__init__(**kwargs)