class Sub_block(db.Model):
id = db.Column(db.Integer, primary_key=True)
sub = db.Column(db.String(30), db.ForeignKey('sub.name'), nullable=False)
username = db.Column(db.String(20),
db.ForeignKey('iuser.username'),
nullable=False)
def __repr__(self):
return '<Sub_block %r>' % self.id
class Api_key(db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(20),
db.ForeignKey('iuser.username'),
nullable=False)
key = db.Column(db.String(20000), nullable=False)
def __repr__(self):
return '<Api_key %r>' % self.id
class Moderator(db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(20),
db.ForeignKey('iuser.username'),
nullable=False)
sub = db.Column(db.String(30), db.ForeignKey('sub.name'), nullable=False)
rank = db.Column(db.Integer, default=0, nullable=False, autoincrement=True)
def __repr__(self):
return '<Moderator %r>' % self.id
class Ban(db.Model):
id = db.Column(db.Integer, primary_key=True)
sub = db.Column(db.String(30), db.ForeignKey('sub.name'), nullable=False)
username = db.Column(db.String(20),
db.ForeignKey('iuser.username'),
nullable=False)
anonymous = db.Column(db.Boolean, default=False, nullable=False)
def __repr__(self):
return '<Ban %r>' % self.id
class Password_reset(db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(20),
db.ForeignKey('iuser.username'),
nullable=False)
rankey = db.Column(db.String(50), unique=False, nullable=False)
valid = db.Column(db.Boolean, default=True, nullable=False)
expires = db.Column(db.DateTime)
def __repr__(self):
return '<Password_reset %r>' % self.id
class Sub(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(30), unique=True, nullable=False)
created_by = db.Column(db.String(20),
db.ForeignKey('iuser.username'),
nullable=False)
created_by_id = db.Column(db.Integer,
db.ForeignKey('iuser.id'),
nullable=False)
created = db.Column(db.DateTime, default=datetime.now, nullable=False)
rules = db.Column(db.String(20000), nullable=True, default=None)
title = db.Column(db.String(1000), nullable=True, default=None)
nsfw = db.Column(db.Boolean, default=False, nullable=False)
css = db.Column(db.String(20000), default=None)
muted = db.Column(db.Boolean, default=False)
def get_comments(self, deleted=False, count=False):
if count == True:
return db.session.query(Comment).filter_by(
sub_name=self.name, deleted=deleted).count()
return db.session.query(Comment).filter_by(sub_name=self.name,
deleted=deleted)
def get_posts(self, deleted=False, count=False):
if count == True:
return db.session.query(Post).filter_by(sub=self.name,
deleted=deleted).count()
return db.session.query(Post).filter_by(sub=self.name, deleted=deleted)
def get_total_users(self, deleted=False, count=False):
posts = self.get_posts(deleted=False, count=False)
comments = self.get_comments(deleted=False, count=False)
users = []
for i in db.session.query(Post.author_id).filter_by(
sub=self.name, deleted=deleted).all():
if i not in users:
users.append(i)
for i in db.session.query(Comment.author_id).filter_by(
sub_name=self.name, deleted=deleted).all():
if i not in users:
users.append(i)
return users
def __repr__(self):
return '<Sub %r>' % self.name
class Mod_action(db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(20), unique=False, nullable=False)
action = db.Column(db.String(20), unique=False, nullable=False)
url = db.Column(db.String(2000), unique=False, nullable=False)
created = db.Column(db.DateTime, default=datetime.now, nullable=False)
sub = db.Column(db.String(30), db.ForeignKey('sub.name'), nullable=False)
def __repr__(self):
return '<Mod_action %r>' % self.id
class Vote(db.Model):
id = db.Column(db.Integer, primary_key=True)
post_id = db.Column(db.Integer, db.ForeignKey('post.id'), nullable=True)
comment_id = db.Column(db.Integer,
db.ForeignKey('comment.id'),
nullable=True)
vote = db.Column(db.Integer, nullable=False)
user_id = db.Column(db.Integer, db.ForeignKey('iuser.id'), nullable=False)
created = db.Column(db.DateTime, default=datetime.utcnow, nullable=False)
def __repr__(self):
return '<Vote %r>' % self.id
class Hidden(db.Model):
id = db.Column(db.Integer, primary_key=True)
post_id = db.Column(db.Integer, db.ForeignKey('post.id'), nullable=True)
comment_id = db.Column(db.Integer,
db.ForeignKey('comment.id'),
nullable=True)
username = db.Column(db.String(20),
db.ForeignKey('iuser.username'),
nullable=False)
other_user = db.Column(db.Integer,
db.ForeignKey('iuser.id'),
nullable=True)
anonymous = db.Column(db.Boolean, default=False)
def __repr__(self):
return '<Hidden %r>' % self.id
class Post(db.Model):
id = db.Column(db.Integer, primary_key=True)
url = db.Column(db.String(2000), nullable=True)
self_text = db.Column(db.String(20000), nullable=True)
post_type = db.Column(db.String(15), nullable=False)
title = db.Column(db.String(400), nullable=False)
ups = db.Column(db.Integer, default=0, nullable=False)
downs = db.Column(db.Integer, default=0, nullable=False)
inurl_title = db.Column(db.String(75), nullable=False)
author = db.Column(db.String(20),
db.ForeignKey('iuser.username'),
nullable=False)
author_id = db.Column(db.Integer,
db.ForeignKey('iuser.id'),
nullable=False)
author_type = db.Column(db.String(20), default='user', nullable=False)
sub = db.Column(db.String(30), db.ForeignKey('sub.name'), nullable=False)
created = db.Column(db.DateTime, default=datetime.now, nullable=False)
deleted = db.Column(db.Boolean, default=False, nullable=False)
permalink = db.Column(db.String(2000), nullable=True)
stickied = db.Column(db.Boolean, default=False, nullable=False)
locked = db.Column(db.Boolean, default=False, nullable=False)
anonymous = db.Column(db.Boolean, default=False, nullable=False)
edited = db.Column(db.Boolean, default=False, nullable=False)
locked = db.Column(db.Boolean, default=False, nullable=False)
nsfw = db.Column(db.Boolean, default=False, nullable=False)
remote_image_url = db.Column(db.String(2000), default=None, nullable=True)
override = db.Column(db.Boolean, default=False, nullable=False)
def get_permalink(self):
return config.URL + str(urllib.parse.urlparse(self.permalink).path)
def has_voted(self, user_id):
return db.session.query(Vote).filter_by(post_id=self.id,
user_id=user_id).first()
def get_score(self):
return (self.ups - self.downs)
def get_type(self):
return 'post'
def get_votes(self):
return db.session.query(Vote).filter_by(post_id=self.id).all()
def __repr__(self):
return '<Post %r>' % self.id
class Message(db.Model):
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(400), unique=False, nullable=False)
text = db.Column(db.String(40000), unique=False, nullable=False)
read = db.Column(db.Boolean, default=False, nullable=False)
sent_to = db.Column(db.String(20),
db.ForeignKey('iuser.username'),
nullable=False)
sender = db.Column(db.String(20),
db.ForeignKey('iuser.username'),
default=None,
nullable=True)
created = db.Column(db.DateTime, default=datetime.now, nullable=False)
in_reply_to = db.Column(db.String(400), default=None, nullable=True)
anonymous = db.Column(db.Boolean, default=False, nullable=False)
sender_type = db.Column(db.String(20), default='user', nullable=False)
encrypted = db.Column(db.Boolean, default=False, nullable=False)
encrypted_key_id = db.Column(db.Integer,
db.ForeignKey('pgp.id'),
default=None,
nullable=True)
def __repr__(self):
return '<Message %r>' % self.id
class Iuser(db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(20), unique=True, nullable=False)
email = db.Column(db.String(75), nullable=True)
password = db.Column(db.String(255), nullable=False)
created = db.Column(db.DateTime, default=datetime.now, nullable=False)
admin = db.Column(db.Boolean, default=False, nullable=False)
banned = db.Column(db.Boolean, default=False, nullable=False)
anonymous = db.Column(db.Boolean, default=False, nullable=False)
darkmode = db.Column(db.Boolean, default=False, nullable=False)
hide_sub_style = db.Column(db.Boolean, default=False, nullable=False)
pgp = db.Column(db.Boolean, default=False, nullable=False)
def get_recent_comments(self, limit=15, deleted=False):
coms = db.session.query(Comment).filter_by(
author_id=self.id, deleted=deleted).order_by(
Comment.created.desc()).limit(limit).all()
return coms
def get_recent_posts(self, limit=15, deleted=False):
posts = db.session.query(Post).filter_by(
author_id=self.id,
deleted=deleted).order_by(Post.created.desc()).limit(limit).all()
return posts
def __repr__(self):
return '<Iuser %r>' % self.username
class Comment(db.Model):
id = db.Column(db.Integer, primary_key=True)
post_id = db.Column(db.Integer, db.ForeignKey('post.id'), nullable=False)
text = db.Column(db.String(20000), nullable=False)
ups = db.Column(db.Integer, default=0, nullable=False)
downs = db.Column(db.Integer, default=0, nullable=False)
author = db.Column(db.String(20),
db.ForeignKey('iuser.username'),
nullable=False)
author_id = db.Column(db.Integer,
db.ForeignKey('iuser.id'),
nullable=False)
author_type = db.Column(db.String(20), default='user', nullable=False)
parent_id = db.Column(db.Integer,
db.ForeignKey('comment.id'),
nullable=True)
level = db.Column(db.Integer, default=0, nullable=False)
created = db.Column(db.DateTime, default=datetime.now, nullable=False)
sub_name = db.Column(db.String(30),
db.ForeignKey('sub.name'),
nullable=False)
deleted = db.Column(db.Boolean, default=False, unique=False)
permalink = db.Column(db.String(2000), nullable=True)
anonymous = db.Column(db.Boolean, default=False, nullable=False)
edited = db.Column(db.Boolean, default=False, nullable=False)
override = db.Column(db.Boolean, default=False, nullable=False)
def get_permalink(self):
return config.URL + str(urllib.parse.urlparse(self.permalink).path)
def get_children(self, deleted=False):
if deleted == False:
return db.session.query(Comment).filter_by(parent_id=self.id,
deleted=deleted)
return db.session.query(Comment).filter_by(parent_id=self.id)
def child_count(self, deleted=False):
if deleted == False:
return db.session.query(Comment).filter_by(parent_id=self.id,
deleted=False).count()
return db.session.query(Comment).filter_by(parent_id=self.id).count()
def get_votes(self):
return db.session.query(Vote).filter_by(comment_id=self.id).all()
def get_score(self):
return int(self.ups) - int(self.downs)
def get_type(self):
return 'comment'
def __repr__(self):
return '<Comment %r>' % self.id
