示例#1
0
class ItemComment(db.Model, BaseModel):
    __tablename__ = "item_comments"

    item = db.relationship(Item)
    user = db.relationship(User)

    item_comment_id = db.Column(db.Integer, primary_key=True)
    item_id = db.Column(db.Integer,
                        db.ForeignKey("items.item_id"),
                        nullable=False)
    author_id = db.Column(db.Integer,
                          db.ForeignKey("users.user_id"),
                          nullable=False)
    comment_text = db.Column(db.Text)
    create_date = db.Column(db.DateTime, default=datetime.utcnow)

    def __init__(self, item_id, user_id, comment_text):
        self.item_id = item_id
        self.author_id = user_id
        self.comment_text = comment_text

    @classmethod
    def find_by_id(cls, item_comment_id, user_id):
        return cls.query.filter(cls.item_comment_id == item_comment_id,
                                cls.author_id == user_id).first()
示例#2
0
class Status(db.Model):
    __tablename__ = "status"

    status_id = db.Column(db.Integer, primary_key=True)
    status_code = db.Column(db.String(30))
    description = db.Column(db.Text)
    create_date = db.Column(db.DateTime, default=datetime.utcnow)
    create_user = db.Column(db.String(80))

    projects = db.relationship("Project", backref="status_group", lazy=True)
示例#3
0
class Item(db.Model, WorkLevel):
    __tablename__ = "items"

    user = db.relationship(User)
    status = db.relationship(Status)
    task = db.relationship(Task)

    item_id = db.Column(db.Integer, primary_key=True)
    task_id = db.Column(db.Integer,
                        db.ForeignKey("tasks.task_id"),
                        nullable=False)
    author_id = db.Column(db.Integer,
                          db.ForeignKey("users.user_id"),
                          nullable=False)
    status_id = db.Column(db.Integer,
                          db.ForeignKey("status.status_id"),
                          nullable=False)
    assigned_id = db.Column(db.Integer)
    title = db.Column(db.String(200))
    description = db.Column(db.Text)
    create_date = db.Column(db.DateTime, default=datetime.utcnow)

    comments = db.relationship("ItemComment", backref="comments", lazy=True)

    def __init__(self, title, description, task_id, user_id):
        self.title = title
        self.description = description
        self.task_id = task_id
        self.author_id = user_id
        # new items always start as 'Proposed'
        self.status_id = 1

    def delete(self):
        self.delete_comments(self.comments)
        self.delete_from_db()
示例#4
0
class Project(db.Model, WorkLevel):
    __tablename__ = "projects"

    user = db.relationship(User)
    status = db.relationship(Status)

    project_id = db.Column(db.Integer, primary_key=True)
    author_id = db.Column(db.Integer,
                          db.ForeignKey("users.user_id"),
                          nullable=False)
    assigned_id = db.Column(db.Integer)
    status_id = db.Column(db.Integer,
                          db.ForeignKey("status.status_id"),
                          nullable=False)
    title = db.Column(db.String(200))
    description = db.Column(db.Text)
    create_date = db.Column(db.DateTime, default=datetime.utcnow)

    tasks = db.relationship("Task", backref="tasks", lazy=True)
    comments = db.relationship("ProjectComment", backref="comments", lazy=True)

    def __init__(self, title, description, user_id):
        self.title = title
        self.description = description
        self.author_id = user_id
        # new projects always start as 'Proposed'
        self.status_id = 1

    def delete(self):
        self.delete_comments(self.comments)
        tasks = self.tasks

        for task in tasks:
            task.delete()

        self.delete_from_db()
示例#5
0
class User(db.Model, UserMixin, BaseModel):
    __tablename__ = "users"

    user_id = db.Column(db.Integer, primary_key=True)
    user_name = db.Column(db.String(80))
    password = db.Column(db.String(100))
    email = db.Column(db.String(100))
    create_date = db.Column(db.DateTime, default=datetime.utcnow)
    create_user = db.Column(db.String(80))

    projects = db.relationship("Project", backref="author", lazy=True)

    def __init__(self, user_name, password, email):
        self.user_name = user_name
        self.password = generate_password_hash(password)
        self.email = email

    def check_password(self, password):
        return check_password_hash(self.password, password)

    def get_id(self):
        return self.user_id