class Friendship(db.Model):
friend_a_id = db.Column(db.Integer,
db.ForeignKey('user.id'),
primary_key=True)
friend_b_id = db.Column(db.Integer,
db.ForeignKey('user.id'),
primary_key=True)
class Access(db.Model):
__table_args__ = {'extend_existing': True}
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), primary_key=True)
post_id = db.Column(db.Integer, db.ForeignKey('post.id'), primary_key=True)
def __repr__(self):
return '<Access>\n\tuser_id:{}\n\tpost_id:{}'.format(
self.user_id, self.post_id)
class Comment(db.Model):
id = db.Column(db.Integer, primary_key=True)
post_id = db.Column(db.Integer, db.ForeignKey('post.id'))
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
comment_filename = db.Column(db.String(100))
shared = db.Column(db.Boolean, default=False)
def __repr__(self):
return '<Comment {}>\n\tpost_id:{}\n'
def create_filename(self, hashed_comment):
self.comment_filename = secure_filename(
os.path.join(str(self.user_id), hashed_comment))
def get_relative_filename(self):
return self.comment_filename
def get_full_filename(self):
print(UPLOAD_FOLDER)
print(self.get_relative_filename())
return os.path.join(UPLOAD_FOLDER, self.get_relative_filename())
def save_post(self, comment_data):
if not self.comment_filename:
print(self.comment_filename)
dummy, temp_filename = mkstemp()
tempfile = open(temp_filename, 'w')
tempfile.write(json.dumps(comment_data))
tempfile.flush()
self.create_filename(sha256sum(temp_filename))
shutil.move(temp_filename, self.get_full_filename())
else:
print("saving (not first one)")
file = open(self.get_full_filename(), 'w')
file.write(json.dumps(comment_data))
file.flush()
def has_file(self):
return self.comment_filename != None
def delete_file(self):
if self.comment_filename:
tmp_full_filename = self.get_full_filename()
os.remove(tmp_full_filename)
class User(UserMixin, db.Model):
# __tablename__ = 'users'
id = db.Column(db.Integer, primary_key=True)
# User Authentication fields
# email = db.Column(db.String(255), nullable=False, unique=True)
# email_confirmed_at = db.Column(db.DateTime())
username = db.Column(db.String(20), nullable=False, unique=True)
password_hash = db.Column(db.String(255), nullable=False)
posts = db.relationship('Post', backref='author', lazy='dynamic')
friends = db.relationship('User',
secondary=Friendship.__tablename__,
primaryjoin=(Friendship.friend_a_id == id),
secondaryjoin=(Friendship.friend_b_id == id))
access_posts = db.relationship('Post',
secondary=Access.__tablename__,
primaryjoin=(Access.user_id == id),
secondaryjoin=(Access.post_id == Post.id),
backref='accessors')
own_posts = db.relationship('Post',
primaryjoin=(Post.user_id == id),
backref='owner')
def timeline_posts(self, index=0, count=20):
print("access posts: ", self.access_posts)
print("own posts: ", self.own_posts)
return sorted([x.id for x in self.access_posts] +
[x.id for x in self.own_posts])
def __repr__(self):
return '<User {}>'.format(self.username)
def set_password(self, password):
self.password_hash = generate_password_hash(password)
def check_password(self, password):
return check_password_hash(self.password_hash, password)
class Post(db.Model):
# __table__name = 'posts'
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
# owner = db.relationship('User', back_populates="own_posts")
entry_filename = db.Column(db.String(100))
title = db.Column(db.String(100))
# filename = db.Column(db.String())
# accessors = db.relationship('User',
# secondary=Access.__tablename__,
# primarjoin=)
# users_with_access = db.relationship('User',
# secondary=Access.__table__name,
# back_populates)
comments = db.relationship('Comment',
primaryjoin=(Comment.post_id == id),
backref='original_post')
def __repr__(self):
return '<Post {}>\n\tuser_id:{}\n\ttitle:{}'.format(
self.id, self.user_id, self.title)
def create_filename(self, hashed_entry):
self.entry_filename = secure_filename(
os.path.join(str(self.user_id), hashed_entry))
def get_relative_filename(self):
return self.entry_filename
def get_full_filename(self):
print(UPLOAD_FOLDER)
print(self.get_relative_filename())
return os.path.join(UPLOAD_FOLDER, self.get_relative_filename())
def save_post(self, entry_data):
if not self.entry_filename:
print(self.entry_filename)
dummy, temp_filename = mkstemp()
tempfile = open(temp_filename, 'w')
tempfile.write(json.dumps(entry_data))
tempfile.flush()
self.create_filename(sha256sum(temp_filename))
shutil.move(temp_filename, self.get_full_filename())
else:
print("saving (not first one)")
file = open(self.get_full_filename(), 'w')
file.write(json.dumps(entry_data))
file.flush()
def has_file(self):
return self.entry_filename != None
def delete_file(self):
if self.entry_filename:
tmp_full_filename = self.get_full_filename()
os.remove(tmp_full_filename)