models.py

from SWIFTDBApp import db


class Partners(db.Model):
    __tablename__ = 'partners'

    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    name = db.Column(db.String(), nullable=False, unique=True)
    country = db.Column(db.String())
    role = db.Column(db.String())
    Deliverables_Rel = db.relationship('Deliverables')
    Tasks_Rel = db.relationship('Tasks')
    Users2Partners_Rel = db.relationship('Users2Partners')

    def __init__(self, name, country, role):
        self.name = name
        self.country = country
        self.role = role

    def __repr__(self):
        return '<name {}>'.format(self.name)


class Work_Packages(db.Model):
    __tablename__ = 'work_packages'

    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    code = db.Column(db.String(), nullable=False, unique=True)
    name = db.Column(db.String(), nullable=False)
    previous_report = db.Column(db.String(()))
    status = db.Column(db.String())
    issues = db.Column(db.String())
    next_deliverable = db.Column(db.String())
    date_edited = db.Column(db.Date())
    Deliverables_Rel = db.relationship('Deliverables')
    Tasks_Rel = db.relationship('Tasks')
    Users2Work_Packages_Rel = db.relationship('Users2Work_Packages')

    def __init__(self, code, name, previous_report, status, issues,
                 next_deliverable, date_edited):
        self.code = code
        self.name = name
        self.previous_report = previous_report
        self.status = status
        self.issues = issues
        self.next_deliverable = next_deliverable
        self.date_edited = date_edited

    def __repr__(self):
        return '<id {}>'.format(self.id)


class Work_Packages_Archive(db.Model):
    __tablename__ = 'work_packages_archive'

    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    date_edited = db.Column(db.Date())
    code = db.Column(db.String(), nullable=False, unique=True)
    status = db.Column(db.String())
    issues = db.Column(db.String())
    next_deliverable = db.Column(db.String())

    def __init__(self, date_edited, code, status, issues,
                 next_deliverable):
        self.date_edited = date_edited
        self.code = code
        self.status = status
        self.issues = issues
        self.next_deliverable = next_deliverable

    def __repr__(self):
        return '<id {}>'.format(self.id)


class Deliverables(db.Model):
    __tablename__ = 'deliverables'

    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    code = db.Column(db.String(), nullable=False, unique=True)
    work_package = db.Column(db.String(), db.ForeignKey('work_packages.code'),
                             nullable=False)
    description = db.Column(db.String(), nullable=False)
    partner = db.Column(db.String(), db.ForeignKey('partners.name'),
                        nullable=False)
    person_responsible = db.Column(db.String())
    month_due = db.Column(db.Date, nullable=False)
    previous_report = db.Column(db.String())
    progress = db.Column(db.String())
    percent = db.Column(db.Integer, nullable=False)
    papers = db.Column(db.String())
    paper_submission_date = db.Column(db.Date())
    date_edited = db.Column(db.Date())

    def __init__(self, code, work_package,  description, partner,
                 person_responsible, month_due, previous_report, progress,
                 percent, papers, paper_submission_date,
                 date_edited):
        self.code = code
        self.work_package = work_package
        self.description = description
        self.partner = partner
        self.person_responsible = person_responsible
        self.month_due = month_due
        self.previous_report = previous_report
        self.progress = progress
        self.percent = percent
        self.papers = papers
        self.paper_submission_date = paper_submission_date
        self.date_edited = date_edited

    def __repr__(self):
        return '<id {}>'.format(self.id)


class Deliverables_Archive(db.Model):
    __tablename__ = 'deliverables_archive'

    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    date_edited = db.Column(db.Date())
    code = db.Column(db.String(), nullable=False, unique=True)
    person_responsible = db.Column(db.String())
    progress = db.Column(db.String())
    percent = db.Column(db.Integer)
    papers = db.Column(db.String())
    paper_submission_date = db.Column(db.Date())

    def __init__(self, date_edited, code, person_responsible,
                 progress, percent, papers, paper_submission_date):
        self.date_edited = date_edited
        self.code = code
        self.person_responsible = person_responsible
        self.progress = progress
        self.percent = percent
        self.papers = papers
        self.paper_submission_date = paper_submission_date

    def __repr__(self):
        return '<id {}>'.format(self.id)


class Users(db.Model):
    __tablename__ = 'users'

    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(), unique=True)
    password = db.Column(db.String())
    Users2Work_Packages_Rel = db.relationship('Users2Work_Packages')
    Users2Partners_Rel = db.relationship('Users2Partners')

    def __init__(self, username, password):
        self.username = username
        self.password = password

    def __repr__(self):
        return '<id {}>'.format(self.id)


class Users2Work_Packages(db.Model):
    __tablename__ = 'users2work_packages'

    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(), db.ForeignKey('users.username'),
                         nullable=False)
    work_package = db.Column(db.String(), db.ForeignKey('work_packages.code'),
                             nullable=False)
    __table_args__ = (db.UniqueConstraint('username', 'work_package',
                                          name='_username_work_package_uc'),)

    def __init__(self, username, work_package):
        self.username = username
        self.work_package = work_package

    def __repr__(self):
        return '<id {}>'.format(self.id)


class Tasks(db.Model):
    __tablename__ = 'tasks'

    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    code = db.Column(db.String(), nullable=False, unique=True)
    work_package = db.Column(db.String(), db.ForeignKey('work_packages.code'),
                             nullable=False)
    description = db.Column(db.String(), nullable=False)
    partner = db.Column(db.String(), db.ForeignKey('partners.name'),
                        nullable=False)
    person_responsible = db.Column(db.String())
    month_due = db.Column(db.Date, nullable=False)
    previous_report = db.Column(db.String())
    progress = db.Column(db.String())
    percent = db.Column(db.Integer, nullable=False)
    papers = db.Column(db.String())
    paper_submission_date = db.Column(db.Date())
    date_edited = db.Column(db.Date())

    def __init__(self, code, work_package, description, partner,
                 person_responsible, month_due, previous_report, progress,
                 percent, papers, paper_submission_date,
                 date_edited):
        self.code = code
        self.work_package = work_package
        self.description = description
        self.partner = partner
        self.person_responsible = person_responsible
        self.month_due = month_due
        self.previous_report = previous_report
        self.progress = progress
        self.percent = percent
        self.papers = papers
        self.paper_submission_date = paper_submission_date
        self.date_edited = date_edited

    def __repr__(self):
        return '<id {}>'.format(self.id)


class Tasks_Archive(db.Model):
    __tablename__ = 'tasks_archive'

    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    date_edited = db.Column(db.Date())
    code = db.Column(db.String(), nullable=False, unique=True)
    person_responsible = db.Column(db.String())
    progress = db.Column(db.String())
    percent = db.Column(db.Integer)
    papers = db.Column(db.String())
    paper_submission_date = db.Column(db.Date())

    def __init__(self, date_edited, code, person_responsible,
                 progress, percent, papers, paper_submission_date):
        self.date_edited = date_edited
        self.code = code
        self.person_responsible = person_responsible
        self.progress = progress
        self.percent = percent
        self.papers = papers
        self.paper_submission_date = paper_submission_date

    def __repr__(self):
        return '<id {}>'.format(self.id)


class Users2Partners(db.Model):
    __tablename__ = 'users2partners'

    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(), db.ForeignKey('users.username'),
                         nullable=False)
    partner = db.Column(db.String(), db.ForeignKey('partners.name'),
                        nullable=False)
    __table_args__ = (db.UniqueConstraint('username', 'partner',
                                          name='_username_partner_uc'),)

    def __init__(self, username, partner):
        self.username = username
        self.partner = partner

    def __repr__(self):
        return '<id {}>'.format(self.id)


class Counts(db.Model):
    __tablename__ = 'counts'

    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    code = db.Column(db.String(), nullable=False, unique=True)
    count = db.Column(db.Integer, nullable=False)

    def __init__(self, code, count):
        self.code = code
        self.count = count

    def __repr__(self):
        return '<id {}>'.format(self.id)