59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
|
from flask import Flask, session, url_for, redirect, request, render_template, abort, flash, Markup, json
|
||
|
|
from datetime import datetime
|
||
|
|
from flask_sqlalchemy import SQLAlchemy
|
||
|
|
import random
|
||
|
|
import os
|
||
|
|
from sqlalchemy.sql.expression import func
|
||
|
|
|
||
|
|
app = Flask(__name__)
|
||
|
|
app.secret_key = "debug-attivo"
|
||
|
|
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///db.sqlite'
|
||
|
|
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
|
||
|
|
db = SQLAlchemy(app)
|
||
|
|
app.config.from_object(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
# Database Classes go beyond this line
|
||
|
|
|
||
|
|
class Message(db.Model):
|
||
|
|
mid = db.Column(db.Integer, primary_key=True)
|
||
|
|
floor = db.Column(db.Integer, nullable=False)
|
||
|
|
content = db.Column(db.String, nullable=False)
|
||
|
|
|
||
|
|
def toJSON(self):
|
||
|
|
return {"Message": {'floor': self.floor,
|
||
|
|
'content': self.content}}
|
||
|
|
|
||
|
|
def __init__(self, content, floor):
|
||
|
|
self.content = content
|
||
|
|
self.floor = floor
|
||
|
|
|
||
|
|
def __repr__(self):
|
||
|
|
return self.content
|
||
|
|
|
||
|
|
|
||
|
|
def give_json_response(floor):
|
||
|
|
messages = Message.query.filter_by(floor=floor).order_by(func.random()).limit(5).all()
|
||
|
|
return json.dumps([message.toJSON() for message in messages])
|
||
|
|
|
||
|
|
|
||
|
|
# Website pages and API functions go beyond this line
|
||
|
|
@app.route("/message/<int:floor>", methods=["GET", "POST"])
|
||
|
|
def page_message(floor):
|
||
|
|
if request.method == "GET":
|
||
|
|
return give_json_response(floor)
|
||
|
|
newmessage = Message(request.form['content'], floor)
|
||
|
|
db.session.add(newmessage)
|
||
|
|
db.session.commit()
|
||
|
|
return "Success."
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
# Se non esiste il database viene creato
|
||
|
|
if not os.path.isfile("db.sqlite"):
|
||
|
|
newmessage = Message("Welcome to the dungeon, mortal.", 1)
|
||
|
|
db.create_all()
|
||
|
|
db.session.add(newmessage)
|
||
|
|
db.session.commit()
|
||
|
|
app.run(debug=True)
|