class Kgpian(db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String, nullable=False)
rating = db.Column(db.Integer, nullable=False)
max_rating = db.Column(db.Integer, nullable=False)
def __repr__(self):
return self.username
class Tag(db.Model):
id = db.Column(db.Integer, primary_key=True, nullable=False)
name = db.Column(db.String(20), nullable=False)
posts = db.relationship('Post',
backref='tag',
secondary='association_table',
lazy="dynamic")
isCompany = db.Column(db.Boolean, default=False)
def __repr__(self):
return self.name
class Round(db.Model):
id = db.Column(db.Integer, primary_key=True)
round_type = db.Column(db.Enum(RoundType), nullable=False)
company_name = db.Column(db.String(20), nullable=False)
content = db.Column(db.Text, nullable=False)
blog_id = db.Column(db.Integer, db.ForeignKey('blog.id'), nullable=True)
selected = db.Column(db.Boolean, default=0)
joining = db.Column(db.Boolean, default=0)
class Comment(db.Model):
id = db.Column(db.Integer, primary_key=True)
content = db.Column(db.Text, nullable=False)
parent_id = db.Column(db.Integer, db.ForeignKey('comment.id'))
parent = db.relationship(lambda: Comment,
remote_side=id,
backref='children')
author_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
depth = db.Column(db.Integer, default=0)
published_at = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow())
votes = db.relationship('Vote', secondary=CommentVotes, lazy="dynamic")
votes_count = db.Column(db.Integer, default=0)
post_id = db.Column(db.Integer, db.ForeignKey('post.id'), default=None)
# problem_id = db.Column(db.Integer, db.ForeignKey('problem.id'))
stream_id = db.Column(db.String, db.ForeignKey('stream.id'), default=None)
@hybrid_property
def media(self):
if self.post_id:
return self.post
elif self.stream_id:
return self.stream
return None
@hybrid_method
def voteStatus(self, user):
vote = next(
(vote for vote in self.votes.all() if vote.user_id == user.id),
None)
return vote
def __repr__(self):
return f"Comment('{self.content}', by: {self.author.username})"
class User(db.Model, UserMixin):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(60), unique=True, nullable=False)
email = db.Column(db.String(120), unique=True, nullable=False)
password = db.Column(db.String(80), nullable=False)
profile_pic_url = db.Column(
db.String(255), default='//userpic.codeforces.com/no-title.jpg')
name = db.Column(db.String(40), nullable=False)
comments = db.relationship('Comment', backref='author', lazy=True)
posts = db.relationship('Post', backref='author', lazy=True)
votes = db.relationship('Vote', backref='user', lazy=True)
def __repr__(self):
return f"User('{self.username}', '{self.email}')"
class Post(db.Model):
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(100), nullable=False)
content = db.Column(db.Text, nullable=False)
blog = db.relationship('Blog', uselist=False, backref="post", lazy=True)
published_at = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow())
author_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
comments = db.relationship('Comment', backref='post', lazy=True)
tags = db.relationship('Tag',
backref='post',
secondary='association_table',
lazy="dynamic")
votes_count = db.Column(db.Integer, default=0)
votes = db.relationship('Vote', secondary=PostVotes, lazy="dynamic")
def __repr__(self):
return self.title
class Team(db.Model):
id = db.Column(db.Integer, primary_key=True)
member1_id = db.Column(db.Integer,
db.ForeignKey('kgpian.id'),
nullable=False)
member1 = db.relationship('Kgpian', foreign_keys=[member1_id])
member2_id = db.Column(db.Integer,
db.ForeignKey('kgpian.id'),
nullable=False)
member2 = db.relationship('Kgpian', foreign_keys=[member2_id])
member3_id = db.Column(db.Integer,
db.ForeignKey('kgpian.id'),
nullable=False)
member3 = db.relationship('Kgpian', foreign_keys=[member3_id])
year = db.Column(db.DateTime, nullable=False) # Just chose the same year.
regional_site = db.Column(db.Enum(RegionalSite), nullable=False)
class Vote(db.Model):
id = db.Column(db.Integer, primary_key=True)
vote_value = db.Column(db.Integer, default=0)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
vote_on = db.Column(db.Enum(PostOrComment), nullable=False)
class Blog(db.Model):
id = db.Column(db.Integer, primary_key=True)
post_id = db.Column(db.Integer, db.ForeignKey('post.id'), nullable=False)
shortlisting_content = db.Column(db.Text, nullable=False)
interview_content = db.Column(db.Text, nullable=True)
rounds = db.relationship('Round', backref='blog', lazy=True)
class Stream(db.Model):
id = db.Column(db.Integer, primary_key=True)
youtube_code = db.Column(db.String, unique=True)
scheduled_at = db.Column(db.DateTime)
comments = db.relationship('Comment', backref='stream', lazy=True)
stream_type = db.Column(db.Enum(StreamType), nullable=False)
from goc import db, login_manager, admin, STATIC_DIR
from flask_admin.contrib.sqla import ModelView
from datetime import datetime
from flask_login import UserMixin
import enum
from sqlalchemy.ext.hybrid import hybrid_property, hybrid_method
Association = db.Table(
'association_table',
db.Column('post_id', db.Integer, db.ForeignKey('post.id')),
db.Column('tag_id', db.Integer, db.ForeignKey('tag.id')))
CommentVotes = db.Table(
'comment_votes_table',
db.Column('comment_id', db.Integer, db.ForeignKey('comment.id')),
db.Column('vote_id', db.Integer, db.ForeignKey('vote.id')))
PostVotes = db.Table(
'post_votes_table',
db.Column('post_id', db.Integer, db.ForeignKey('post.id')),
db.Column('vote_id', db.Integer, db.ForeignKey('vote.id')))
class Blog(db.Model):
id = db.Column(db.Integer, primary_key=True)
post_id = db.Column(db.Integer, db.ForeignKey('post.id'), nullable=False)
shortlisting_content = db.Column(db.Text, nullable=False)
interview_content = db.Column(db.Text, nullable=True)
rounds = db.relationship('Round', backref='blog', lazy=True)