class Milestone(db.Model): # delete this
milestone_id = db.Column(db.Integer, primary_key=True, autoincrement=True)
milestone_name = db.Column(db.String(100), nullable=False)
task_id = db.Column(db.Integer, db.ForeignKey('task.task_id'), nullable=True)
def __repr__(self):
return f"Milestone('{self.milestone_id}', '{self.milestone_name}', '{self.milestone_date}', '{self.task_id}')"
class User(db.Model, UserMixin):
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
username = db.Column(db.String(20), unique=True, nullable=False)
email = db.Column(db.String(120), unique=True, nullable=False)
password = db.Column(db.String(60), nullable=False)
# Flask-Login integration
def is_authenticated(self):
return True
def is_active(self):
return True
def is_anonymous(self):
return False
def get_id(self):
return self.id
# Required for administrative interface
def __unicode__(self):
return self.username
def __repr__(self):
return f"User('{self.username}', '{self.email}')"
class Category(db.Model):
category_id = db.Column(db.Integer, primary_key=True, autoincrement=True)
category_name = db.Column(db.String(100), nullable=False)
category_color = db.Column(db.String(100), nullable=False)
category_checked = db.Column(db.Boolean, nullable=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
user = db.relationship('User', backref="categories")
def __repr__(self):
return f"Category('{self.category_name}', '{self.category_color}')"
class Task(db.Model):
"""checks user and category, gets category color. creates task and milestone"""
task_id = db.Column(db.Integer, primary_key=True, autoincrement=True)
task_name = db.Column(db.String(100), nullable=False)
task_start_date = db.Column(db.String(100), nullable=False)
task_end_date = db.Column(db.String(100), nullable=False)
task_milestone_name = db.Column(db.String(100), nullable = True)
task_milestone_date = db.Column(db.String(100), nullable = True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
user = db.relationship('User', backref='tasks')
category_id = db.Column(db.Integer, db.ForeignKey('category.category_id'))
category = db.relationship('Category', backref="tasks")
def __repr__(self):
return f"Task('{self.task_id}', '{self.task_name}', '{self.task_start_date}', '{self.task_end_date}', '{self.category_id}', '{self.task_milestone_name}', '{self.task_milestone_date}')"