simple-forum/helper.py

from flask_sqlalchemy import SQLAlchemy
from flask_login import UserMixin
import json

# This file contains all the databases.
def createDb(app):
	app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///data.db'
	app.config['SECRET_KEY'] = 'hard to guess thing'
	db = SQLAlchemy(app)

	# Create the tables
	class User(UserMixin, db.Model):
			id = db.Column(db.Integer, primary_key=True)
			username = db.Column(db.Text, unique=True)
			password = db.Column(db.Text)
			def __init__(self, username, password):
				self.username = username
				self.password = password
	class Topic(db.Model):
	        id = db.Column(db.Integer, primary_key=True)
	        title = db.Column(db.Text)
	        content = db.Column(db.Text)
	        date = db.Column(db.Text)
	        lastActivity = db.Column(db.Text)
	        author = db.Column(db.Text)
	        category = db.Column(db.Text)
	        private = db.Column(db.Boolean)
	        likes = db.Column(db.Text)
	        replies = db.Column(db.Text)
	
	        # Numbers
	        likesNum = db.Column(db.Integer)
	        repliesNum = db.Column(db.Integer)
	        views = db.Column(db.Integer)
		
	        def __init__(self, title, content, date, author, category, private=False):
	            self.title = title
	            self.content = content
	            self.date = date
	            self.author = author
	            self.category = category
	            self.private = private
	            self.likesNum = 0
	            self.repliesNum = 0
	            self.views = 0
	            self.likes = '[]'
	            self.lastActivity = date
	        # Create like and other def for this thing
	        def like(self, username):
	            l = json.loads(self.likes)
	            if username in l:
	                l.remove(username)
	                self.likesNum -= 1
	                self.likes = json.dumps(l)
	            else:
	                l.append(username)
	                self.likesNum += 1
	                self.likes = json.dumps(l)
	        def reply(self, date):
	            self.lastActivity = date
	            self.repliesNum += 1
	class Reply(db.Model):
	        id = db.Column(db.Integer, primary_key=True)
	        content = db.Column(db.Text)
	        date = db.Column(db.Text)
	        author = db.Column(db.Text)
	        inReplyTo = db.Column(db.Integer)
	        likes = db.Column(db.Text)

	        # Adding the numbers
	        likesNum = db.Column(db.Integer)
	        def __init__(self, content, date, author, inReplyTo):
	            self.content = content
	            self.date = date
	            self.author = author
	            self.inReplyTo = inReplyTo
	            self.likesNum = 0
	            self.likes = '[]'
	        def like(self, username):
	            l = json.loads(self.likes)
	            if username in l:
	                l.remove(username)
	                self.likesNum -= 1
	                self.likes = json.dumps(l)
	            else:
	                l.append(username)
	                self.likesNum += 1
	                self.likes = json.dumps(l)
	db.create_all()

	# Return the data
	return {'db': db, 'User': User, 'Topic': Topic, 'Reply': Reply}