class Role(db.Model):
__tablename__ = 'roles'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(10), nullable=False, unique=True)
def __repr__(self):
return f"Role('{self.id}', '{self.name}' )"
class Tag(db.Model):
__tablename__ = 'theTags'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.Text, nullable=False, unique=True)
def __repr__(self):
return f"Role('{self.id}', '{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'))
def __repr__(self):
return '{} [{}]'.format(self.text, self.url)
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 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 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 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 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 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 Newsletter(db.Model):
__tablename__ = 'newsletters'
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(100), nullable=False)
content = db.Column(db.Text, nullable=False)
date_posted = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow)
sendOut = db.Column(db.Boolean(), nullable=False, default=False)
def __repr__(self):
return f"Comment('{self.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 User(db.Model, UserMixin):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(20), unique=True, nullable=False)
email = db.Column(db.String(120), unique=True, nullable=False)
image_file = db.Column(db.String(20), nullable=False, default='default.jpg')
password = db.Column(db.String(60), nullable=False)
zip_code = db.Column(db.Integer, nullable=False)
notes = db.relationship('Note', backref='author', lazy=True)
links = db.relationship('Link', backref='author', lazy=True)
sent_messages = db.relationship('Message', backref="sender", lazy=True)
def get_reset_token(self, expieres_sec=9000):
s = Serializer(current_app.config['SECRET_KEY'], expieres_sec)
return s.dumps({"user_id": self.id}).decode('utf-8')
@staticmethod
def verify_reset_token(token):
s = Serializer(current_app.config['SECRET_KEY'])
try:
user_id = s.loads(token)['user_id']
except:
return None
return User.query.get(user_id)
def __repr__(self):
return f"User('{self.username}', '{self.email}', '{self.image_file}')"
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 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)
class User(db.Model, UserMixin):
__tablename__ = 'users'
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(20), unique=True, nullable=False)
email = db.Column(db.String(120), unique=True, nullable=False)
image_file = db.Column(db.String(20),
nullable=False,
default='default.jpg')
date_joined = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow)
password = db.Column(db.String(60), nullable=False)
roles = db.relationship('Role', secondary='user_roles')
posts = db.relationship('Post', backref='author', lazy=True)
portfolios = db.relationship('Portfolio', backref='creator', lazy=True)
comments_ = db.relationship('Comment', backref='author_comment', lazy=True)
discussions_ = db.relationship('Discussion',
backref='author_discussion',
lazy=True)
newsletter = db.Column(db.Boolean(), nullable=False, default=False)
activated = db.Column(db.Boolean(), nullable=False, default=False)
banned = db.Column(db.Boolean(), nullable=False, default=False)
def get_reset_token(self, expires_sec=1800):
s = Serializer(current_app.config['SECRET_KEY'], expires_sec)
return s.dumps({'user_id': self.id}).decode('utf-8')
@staticmethod
def verify_reset_token(token):
s = Serializer(current_app.config['SECRET_KEY'])
try:
user_id = s.loads(token)['user_id']
except:
return None
return User.query.get(user_id)
def __repr__(self):
return f"User('{self.id}', '{self.username}', '{self.email}', '{self.image_file}', '{self.roles}' )"
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 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 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}' )"
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'))
class Tag(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(50), unique=True)
def __repr__(self):
return '<{}>'.format(self.name)
