Last active
May 8, 2024 23:12
-
-
Save crearo/bc23880cf2ba0ec2dc524b66388a0072 to your computer and use it in GitHub Desktop.
A flask sqlalchemy database with foreign key references and integrity constraints
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from flask import Flask, jsonify, request, make_response | |
from flask.ext.httpauth import HTTPBasicAuth | |
from flask_sqlalchemy import SQLAlchemy | |
from flask import render_template, redirect, url_for | |
from sqlalchemy import UniqueConstraint, exc | |
app = Flask(__name__) | |
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///memes.db' | |
app.config['SECRET_KEY'] = 'HALO' | |
db = SQLAlchemy(app) | |
auth = HTTPBasicAuth() | |
STRING_FAIL = 'fail' | |
STRING_SUCCESS = 'success' | |
STRING_UPVOTE = 'upvote' | |
STRING_DOWNVOTE = 'downvote' | |
class Photo(db.Model): | |
__tablename__ = 'photo' | |
id = db.Column(db.Integer, primary_key = True, autoincrement = True) | |
urlPrimary = db.Column(db.String, nullable = False, unique = True) | |
urlSecondary = db.Column(db.String, nullable = True) | |
title = db.Column(db.String, nullable = True) | |
uploadDate = db.Column(db.String, nullable = False) | |
votes = db.relationship('Votes', backref = 'photo', cascade = 'all, delete-orphan', lazy = 'dynamic') | |
def __init__ (self,urlPrimary,urlSecondary,title,uploadDate): | |
self.urlPrimary = urlPrimary | |
self.urlSecondary = urlSecondary | |
self.title = title | |
self.uploadDate = uploadDate | |
def toString(self): | |
up = len([v for v in self.votes if v.voteType == STRING_UPVOTE]) # count total upvotes | |
down = len([v for v in self.votes if v.voteType == STRING_DOWNVOTE]) # count total downvotes | |
return ({'urlPrimary' : self.urlPrimary, 'urlSecondary': self.urlSecondary,'title':self.title, 'uploadDate':self.uploadDate, 'voteCountUp':up, 'voteCountDown' : down}) | |
class Users(db.Model): | |
__tablename__ = 'user' | |
id = db.Column(db.Integer, primary_key = True, autoincrement = True) | |
name = db.Column(db.String, nullable = False) | |
email = db.Column(db.String,nullable = False, unique = True) | |
registeredDate = db.Column(db.String, nullable = False) | |
votes = db.relationship('Votes', backref = 'user', cascade = 'all, delete-orphan', lazy = 'dynamic') | |
def __init__(self,name,email,registeredDate): | |
self.name = name | |
self.email = email | |
self.registeredDate = registeredDate | |
def toString(self): | |
voted = [vote.toString() for vote in self.votes] | |
return ({'name':self.name, 'email':self.email, 'registeredDate':self.registeredDate, 'votes' : voted}) | |
class Votes(db.Model): | |
__tablename__ = 'vote' | |
__table_args__ = (UniqueConstraint('photoID','userID', name = 'unique_photo_user'),) | |
id = db.Column(db.Integer, primary_key = True, autoincrement = True) | |
photoID = db.Column(db.Integer, db.ForeignKey('photo.id'), nullable = False) | |
userID = db.Column(db.Integer, db.ForeignKey('user.id'), nullable = False) | |
voteType = db.Column(db.String, nullable = False) | |
votedDate = db.Column(db.String, nullable = False) | |
def __init__(self,photoID, userID, voteType, votedDate): | |
self.photoID = photoID | |
self.userID = userID | |
self.voteType = voteType | |
self.votedDate = votedDate | |
def toString(self): | |
return ({'photoID':self.photoID, 'userID':self.userID, 'voteType':self.voteType, 'votedDate':self.votedDate}) | |
@app.route('/') | |
def home(): | |
return 'The Memions' | |
@app.route('/photos') | |
def photoAll(): | |
photos = Photo.query.all() | |
p = [photo.toString() for photo in photos] | |
return jsonify(meta = STRING_SUCCESS, photos = p) | |
@app.route('/photos/new', methods = ['POST']) | |
def newPhoto(): | |
jsonData = request.json | |
newPhoto = Photo(jsonData['urlPrimary'], jsonData['urlSecondary'], jsonData['title'], jsonData['uploadDate']) | |
db.session.add(newPhoto) | |
session_commit() | |
@app.route('/photo/<photoid>') | |
def photoIndividual(photoid): | |
photo = Photo.query.filter_by(id = photoid).first() | |
if photo is None : | |
return jsonify(meta = STRING_FAIL) | |
else: | |
return jsonify(meta = STRING_SUCCESS, photo = photo.toString()) | |
@app.route('/users') | |
def users(): | |
users = Users.query.all() | |
u = [user.toString() for user in users] | |
return jsonify(meta = STRING_SUCCESS, users = u) | |
@app.route('/users/new', methods = ['POST']) | |
def newUser(): | |
jsonData = request.json | |
newUser = Users(jsonData['name'], jsonData['email'], jsonData['registeredDate']) | |
db.session.add(newUser) | |
session_commit() | |
@app.route('/user/<userid>') | |
def userIndividual(userid): | |
user = Users.query.filter_by(id = userid).first() | |
if user is None : | |
return jsonify(meta = STRING_FAIL) | |
else: | |
return jsonify(meta = STRING_SUCCESS, user = user.toString()) | |
@app.route('/votes') | |
def votes(): | |
votes = Votes.query.all() | |
v = [vote.toString() for vote in votes] | |
return jsonify(meta = STRING_SUCCESS, votes = v) | |
@app.route('/votes/new', methods = ['POST']) | |
def newVotes(): | |
jsonData = request.json | |
newVote = Votes(jsonData['photoID'], jsonData['userID'], jsonData['voteType'], jsonData['votedDate']) | |
db.session.add(newUser) | |
session_commit() | |
def session_commit(): | |
try : | |
db.session.commit() | |
return jsonify(meta = STRING_SUCCESS) | |
except exc.IntegrityError: | |
print ("IntegrityError while adding new user") | |
db.session.rollback() | |
return jsonify(meta = STRING_FAIL) | |
if __name__ == '__main__' : | |
db.create_all() | |
""" | |
rish = Users(name = 'rish', email ='[email protected]', registeredDate = '123456789') | |
meme = Photo( urlPrimary = 'urlPrimary', urlSecondary = 'urlSecondary', title = 'memeTitle', uploadDate ='1234567898') | |
voteup = Votes(1,1, voteType = STRING_UPVOTE, votedDate = '98765432') | |
db.session.add(rish) | |
db.session.add(meme) | |
db.session.add(voteup) | |
try: | |
db.session.commit() | |
except exc.IntegrityError: | |
print 'IntegrityError while committing brah' | |
db.session.rollback() | |
print Photo.query.get(1).votes.count() | |
""" | |
app.run(debug=True) |
just don't forget you never implemented the authentication against the routes so people learning like me may make a mistake there so can u add @login_required decorator to this and a few small fixes to secure it?
Unfortunately, I haven't worked on this in 4 years. Maybe you can make a gist of your own and link it here?
Great
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
A very basic, working 3 tabled flask app. Catches integrity errors. Also has a toString method to print dict (json) with ease.
Will use this to make future apps.