class PostTags(db.Model):
__tablename__ = 'post_tags'
id = db.Column(db.Integer(), primary_key=True)
post_id = db.Column(db.Integer(),
db.ForeignKey('posts.id', ondelete='CASCADE'))
tag_id = db.Column(db.Integer(),
db.ForeignKey('theTags.id', ondelete='CASCADE'))
def __repr__(self):
return f"Role('{self.user_id}', '{self.post_id}' )"
class CommentLikes(db.Model):
__tablename__ = 'comment_likes'
id = db.Column(db.Integer(), primary_key=True)
user_id = db.Column(db.Integer(),
db.ForeignKey('users.id', ondelete='CASCADE'))
comment_id = db.Column(db.Integer(),
db.ForeignKey('comments.id', ondelete='CASCADE'))
def __repr__(self):
return f"Role('{self.user_id}', '{self.comment_id}' )"
class UserRoles(db.Model):
__tablename__ = 'user_roles'
id = db.Column(db.Integer(), primary_key=True)
user_id = db.Column(db.Integer(),
db.ForeignKey('users.id', ondelete='CASCADE'))
role_id = db.Column(db.Integer(),
db.ForeignKey('roles.id', ondelete='CASCADE'))
def __repr__(self):
return f"Role('{self.user_id}', '{self.role_id}' )"
class Discussion(db.Model):
__tablename__ = 'discussions'
id = db.Column(db.Integer, primary_key=True)
content = db.Column(db.Text, nullable=False)
uid = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
cid = db.Column(db.Integer, db.ForeignKey('comments.id'), nullable=False)
date_posted = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow)
def __repr__(self):
return f"Discussion('{self.id}', '{self.content}')"
class Comment(db.Model):
__tablename__ = 'comments'
id = db.Column(db.Integer, primary_key=True)
content = db.Column(db.Text, nullable=False)
uid = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
pid = db.Column(db.Integer, db.ForeignKey('posts.id'), nullable=False)
date_posted = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow)
discussions = db.relationship('Discussion',
backref='comment_discussion',
lazy=True)
reports = db.relationship('Report', backref='comment_report', lazy=True)
def __repr__(self):
return f"Comment('{self.id}', '{self.content}')"
class Link(db.Model):
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(100), nullable=False)
address = db.Column(db.Text, nullable=False)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
def __repr__(self):
return f"Link('{self.title}', '{self.address}'')"
class Link(db.Model):
id = db.Column(db.Integer, primary_key=True)
text = db.Column(db.String(80))
url = db.Column(db.String(255))
project_id = db.Column(db.Integer, db.ForeignKey('project.id'))
def __repr__(self):
return '{} [{}]'.format(self.text, self.url)
class Note(db.Model):
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(100), unique=True, nullable=False)
date_posted = db.Column(db.DateTime, nullable=False, default=datetime.datetime.utcnow)
content = db.Column(db.Text, nullable=False)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
def __repr__(self):
return f"Note('{self.title}', '{self.date_posted}'')"
class Message(db.Model):
id = db.Column(db.Integer, primary_key=True)
sender_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
recipient_id = db.Column(db.Integer, nullable=False)
subject = db.Column(db.String(120), nullable=False, default="Subject")
body = db.Column(db.Text, nullable=False)
time_sent = db.Column(db.DateTime, nullable=False, default=datetime.datetime.utcnow)
unread = db.Column(db.Boolean, nullable=False, default=True)
def __repr__(self):
return f"""Message(Sender: {User.query.get(self.sender_id).username}
class Report(db.Model):
__tablename__ = 'reports'
id = db.Column(db.Integer, primary_key=True)
complain = db.Column(db.Text, nullable=False)
cid = db.Column(db.Integer, db.ForeignKey('comments.id'), nullable=False)
date_posted = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow)
def __repr__(self):
return f"Report('{self.id}', '{self.complain}')"
class Position(db.Model):
__tablename__ = 'positions'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.Text, nullable=False, default="Empty")
ticker = db.Column(db.Text, nullable=False, default="Empty")
percent = db.Column(db.Float, nullable=False, default="0.0")
currency = db.Column(db.Text, nullable=False, default='NA')
port_id = db.Column(db.Integer,
db.ForeignKey('portfolios.id'),
nullable=False)
def __repr__(self):
return f"Position('{self.name}', Size: '{self.percent}')"
class Project(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(80))
url = db.Column(db.String(255))
description = db.Column(db.Text)
date = db.Column(db.Date, default=date.today())
display = db.Column(db.Boolean, default=True)
links = db.relationship('Link', backref='project', lazy='select')
tags = db.relationship('Tag',
secondary=tags,
backref=db.backref('projects', lazy='dynamic'))
tag_id = db.Column(db.Integer, db.ForeignKey('tag.id'))
def __repr__(self):
return '{}'.format(self.name)
class Post(SearchableMixin, db.Model):
__tablename__ = 'posts'
__searchable__ = ['content']
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(100), nullable=False)
date_posted = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow)
abstract = db.Column(db.Text, nullable=False)
content = db.Column(db.Text, nullable=False)
region = db.Column(db.Integer, default=0)
user_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
comments = db.relationship('Comment', backref='post_comment', lazy=True)
who_liked = db.relationship('User', backref='post_liked', lazy=True)
def __repr__(self):
return f"Post('{self.title}', '{self.date_posted}' )"
class Portfolio(db.Model):
__tablename__ = 'portfolios'
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.Text, nullable=False)
numberPositions = db.Column(db.Integer, nullable=False)
user_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
positions = db.relationship('Position', backref='portfolio', lazy=True)
def __repr__(self):
return f"Portfolio('{self.name}', Size: '{self.numberPositions}')"
@staticmethod
def hasOneCurrency(pf):
currency = ""
for pos in pf.positions:
if currency == "":
currency = pos.currency
else:
if pos.currency != currency:
return False
return True
@staticmethod
def isDefined(pf):
balance = 0
for pos in pf.positions:
balance += pos.percent
# return balance == 100
return (abs(balance - 100) < 0.01)
from datetime import date
from homepage import db
tags = db.Table(
'tags', db.Column('tag_id', db.Integer, db.ForeignKey('tag.id')),
db.Column('project_id', db.Integer, db.ForeignKey('project.id')))
class Project(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(80))
url = db.Column(db.String(255))
description = db.Column(db.Text)
date = db.Column(db.Date, default=date.today())
display = db.Column(db.Boolean, default=True)
links = db.relationship('Link', backref='project', lazy='select')
tags = db.relationship('Tag',
secondary=tags,
backref=db.backref('projects', lazy='dynamic'))
tag_id = db.Column(db.Integer, db.ForeignKey('tag.id'))
def __repr__(self):
return '{}'.format(self.name)
class Link(db.Model):
id = db.Column(db.Integer, primary_key=True)
text = db.Column(db.String(80))
url = db.Column(db.String(255))
project_id = db.Column(db.Integer, db.ForeignKey('project.id'))