1
0
Fork 0
forked from bton/matekasse
matekasse/Website/__init__.py

432 lines
19 KiB
Python
Raw Normal View History

2023-07-28 22:09:42 +00:00
import queue, time, uuid, json, logging, datetime, os
2023-08-16 21:02:36 +00:00
from flask import Flask, render_template, request, make_response, session, send_file, g
2023-07-28 21:30:45 +00:00
from flask_socketio import SocketIO, join_room, leave_room
from flask_session import Session
from markupsafe import escape
2023-11-08 20:11:45 +00:00
from .db import get_db
2023-11-24 23:05:33 +00:00
from datetime import datetime
2023-11-25 15:59:51 +00:00
finished = None
2023-07-28 21:30:45 +00:00
#flask_config
DATABASE = './Website/mate.db'
2023-07-28 21:30:45 +00:00
2023-11-24 23:05:33 +00:00
#def create_logs(app):
# now = datetime.datetime.now().strftime('%d-%m-%Y-%H-%M-%S')
# logging.basicConfig(filename=f"logs/matekasse-{now}.log",filemode='w', format='%(asctime)s - %(name)s - %(levelname)s - %(message)s' ,encoding='utf-8', level=logging.INFO)
# app.logger = logging.getLogger('db')
#
# app.logger.info("Website is starting")
2023-07-28 21:30:45 +00:00
2023-11-24 23:05:33 +00:00
def log(type=None, userid=None, before=None, after=None):
db = get_db()
c = db.cursor()
c.execute("INSERT or IGNORE INTO transaction_log (timestamp, userid, type, before, after) VALUES (?, ?, ?, ?, ?)", [datetime.now(), userid, type, before, after])
db.commit()
2023-07-28 21:30:45 +00:00
def create_app(test_config=None):
app = Flask(__name__)
key = str(uuid.uuid4().hex)
if test_config is None:
app.config['SESSION_TYPE'] = 'filesystem'
app.config['SECRET_KEY'] = key
app.config['DATABASE'] = DATABASE
else:
app.config.from_mapping(test_config)
try:
os.makedirs(app.instance_path)
except OSError:
pass
2023-11-24 23:05:33 +00:00
#with app.app_context():
# create_logs(app)
2023-07-28 21:30:45 +00:00
Session(app)
socketio = SocketIO(app)
#@app.teardown_appcontext
#def close_connection(exception):
# db = getattr(g, '_database', None)
# if db is not None:
# db.close()
# app.logger.info("Website exited")
2023-07-28 22:09:42 +00:00
2023-07-28 21:30:45 +00:00
#var
2023-08-16 19:36:23 +00:00
user_queue = queue.Queue()
2023-07-28 21:30:45 +00:00
#website
@app.route('/favicon.ico')
def favicon():
2023-08-16 21:02:36 +00:00
return send_file("../static/Logo_CCC.svg.png")
2023-07-28 21:30:45 +00:00
2023-08-16 21:02:36 +00:00
#@app.route('/socket.io.js')
#def socketiojs():
# return url_for('static', filename='socket.io.js')
2023-07-28 21:30:45 +00:00
@app.route("/")
def index():
return """
<a href="/list">user and tag list</a>
<p>The creator of this website accepts no liability for any linguistic or technical errors!</p>
<br style="line-height: 500%;"></br>
<a href="/documentation">Doumentation</a><script src="/socket.io.js" integrity="sha512-q/dWJ3kcmjBLU4Qc47E4A9kTB4m3wuTY7vkFJDTZKjTs8jhyGQnaUrxa0Ytd0ssMZhbNua9hE+E7Qv1j+DyZwA==" crossorigin="anonymous">
</script>
<script type="text/javascript" charset="utf-8">
window.location="/list"
</script>
"""
@app.route("/list")
def list():
db = get_db()
c = db.cursor()
c.execute("SELECT * FROM users")
users = c.fetchall()
text = ""
for i in users:
2023-12-20 18:47:38 +00:00
text = text + f'<form action="/change" method="post"><p style="display: inline;"><a href="list/user?id={i[0]}">{escape(i[1])}</a>: {i[2]/100}€</p><input name="id" type="hidden" value="{i[0]}"> <input name="change" type="number" step="0.01" placeholder="add to balance"></form><br style="line-height: 50%;"></br>'
2023-07-28 21:30:45 +00:00
return '''<!DOCTYPE html>
<html lang="en">
<script src="https://cdnjs.cloudflare.com/ajax/libs/socket.io/4.0.1/socket.io.js" integrity="sha512-q/dWJ3kcmjBLU4Qc47E4A9kTB4m3wuTY7vkFJDTZKjTs8jhyGQnaUrxa0Ytd0ssMZhbNua9hE+E7Qv1j+DyZwA==" crossorigin="anonymous"></script>
<script type="text/javascript" charset="utf-8">
var socket = io();
socket.on("update", function(){
window.location="http://matekasse.server.c3h/list"
});
</script>
2023-09-19 20:23:55 +00:00
<title>Strichliste</title>
2023-07-28 21:30:45 +00:00
<p><a href="/list">user and tag list</a> | <a href="/documentation">Documentation</a></p>
2023-09-20 21:11:39 +00:00
<form action="/list/user" method="get"><input name="user" type="search" placeholder="Search for user"><button>Search</button></form>
2023-11-24 17:26:54 +00:00
<form action="/adduser" method="post"><button type="submit">Add User</button></form>
2023-07-28 21:30:45 +00:00
<br></br>
2023-09-20 17:38:17 +00:00
''' + text + '</html>'
2023-07-28 21:30:45 +00:00
2023-11-24 23:05:33 +00:00
@app.route("/transactionlist")
def transactionlist():
db = get_db()
c = db.cursor()
text = ""
c.execute("SELECT * FROM transaction_log ORDER BY ROWID DESC LIMIT 100")
transactionlist = c.fetchall()
for i in transactionlist:
2023-12-02 17:14:34 +00:00
text = text + f'<form action /api/<p style="display: inline;">{i[0]} userid: {i[1]} {i[2]} {i[3]} to {i[4]}</p>'
2023-11-24 23:05:33 +00:00
return text
2023-07-28 21:30:45 +00:00
@app.route("/list/user", methods=['GET'])
def user_info():
db = get_db()
c = db.cursor()
id = request.args.get("id")
c.execute(f"SELECT * FROM users WHERE id=?", [id])
user_list = c.fetchall()
if user_list != []:
user = user_list[0]
c.execute(f"SELECT * FROM tags WHERE userid={user[0]}")
tags = c.fetchall()
text = ""
for tag in tags:
2023-11-25 18:28:05 +00:00
text = text + f'<p><form action="/removetag" method="post"><label for="removetag">{tag[0]} </label><input name="id" type="hidden" value="{user[0]}"><input name="tagid" type="hidden" value="{tag[0]}"><button id="removetag" type="submit">Remove Tag</button></form> </p>'
2023-07-28 21:30:45 +00:00
return f"""<!DOCTYPE html>
<html lang="en">
<script src="https://cdnjs.cloudflare.com/ajax/libs/socket.io/4.0.1/socket.io.js" integrity="sha512-q/dWJ3kcmjBLU4Qc47E4A9kTB4m3wuTY7vkFJDTZKjTs8jhyGQnaUrxa0Ytd0ssMZhbNua9hE+E7Qv1j+DyZwA==" crossorigin="anonymous"></script>
<script type="text/javascript" charset="utf-8">
var socket = io();
""" + 'socket.on("update", function(){ window.location="http://matekasse.server.c3h/list/user?id=' + id + '"});' + f"""
</script>
2023-09-19 20:23:55 +00:00
<title>{escape(user[1])}</title>
2023-07-28 21:30:45 +00:00
<p><a href="/list">user and tag list</a> | <a href="/documentation">Documentation</a></p>
2023-11-25 17:22:13 +00:00
<p> {escape(user[1])} : {escape(user[2]/100)} <p>
2023-11-24 17:26:54 +00:00
<form action="/addtag" method="post"><input name="id" type="hidden" value="{user[0]}"><button type="submit">Add Tag</button></form>
<form action="/removetag" method="post"><input name="id" type="hidden" value="{user[0]}"><button type="submit">Remove Tag</button></form>
2023-12-20 18:47:38 +00:00
</p><form action="/change" method="post"><input name="id" type="hidden" value="{user[0]}"> <input name="change" type="number" step="0.01" placeholder="change balance"></form>
2023-07-28 21:30:45 +00:00
</p>
<br></br>
<p>Tags:</p>
{text}
<br></br>
2023-11-24 17:26:54 +00:00
<form action="/removeuser" method="post"><input name="id" type="hidden" value="{user[0]}"><button type="submit">Remove User</button></form>
2023-07-28 21:30:45 +00:00
</html>
"""
else:
2023-09-19 20:23:55 +00:00
return render_template("error.html", error_code="043")
2023-07-28 21:30:45 +00:00
2023-11-24 17:26:54 +00:00
@app.route("/adduser", methods=['POST'])
2023-07-28 21:30:45 +00:00
def new_user():
return render_template("adduser.html")
2023-11-24 17:26:54 +00:00
@app.route("/removeuser", methods=['POST'])
2023-07-28 21:30:45 +00:00
def remove_user():
db = get_db()
c = db.cursor()
2023-11-24 17:26:54 +00:00
user_id = request.form["id"]
2023-07-28 21:30:45 +00:00
c.execute(f"SELECT * FROM users WHERE id=?", [user_id])
users = c.fetchall()
if users != []:
user_name = users[0][1]
c.execute(f"DELETE FROM tags WHERE userid=?", [user_id])
app.logger.info(f"Deleted all tags from user ?", [user_id])
c.execute(f"DELETE FROM users WHERE id=?", [user_id])
db.commit()
2023-11-24 23:05:33 +00:00
log(type="removeuser", userid=user_id, before=user_name)
2023-07-28 21:30:45 +00:00
socketio.emit("update", "update")
2023-09-19 20:23:55 +00:00
return f'<title>remove user</title><p><p><a href="/list">user and tag list</a> | <a href="/documentation">Documentation</a></p> <p>Deleted user {escape(user_name)}</p><a href="/list">return to the tags and user list</a></p>'
2023-07-28 21:30:45 +00:00
else:
2023-09-19 20:23:55 +00:00
return render_template("error.html", error_code="043")
2023-07-28 21:30:45 +00:00
2023-11-24 17:26:54 +00:00
@app.route("/adduser/user", methods=['POST'])
2023-07-28 21:30:45 +00:00
def adduser():
db = get_db()
c = db.cursor()
2023-11-24 17:26:54 +00:00
username = request.form["username"]
2023-07-28 21:30:45 +00:00
if username == None:
2023-09-19 20:23:55 +00:00
return render_template("error.html", error_code="418")
2023-07-28 21:30:45 +00:00
c.execute("SELECT * FROM users WHERE username=?", [username])
if c.fetchall() == []:
c.execute("INSERT or IGNORE INTO users (username, balance) VALUES (?, 0)", [username])
db.commit()
socketio.emit("update", "update")
c.execute(f"SELECT * FROM users WHERE username=?", [username])
user = c.fetchone()
2023-11-24 23:05:33 +00:00
log(type="adduser", userid=user[0], after=user[1])
2023-07-28 21:30:45 +00:00
return """<html>
<script src="https://cdnjs.cloudflare.com/ajax/libs/socket.io/4.0.1/socket.io.js" integrity="sha512-q/dWJ3kcmjBLU4Qc47E4A9kTB4m3wuTY7vkFJDTZKjTs8jhyGQnaUrxa0Ytd0ssMZhbNua9hE+E7Qv1j+DyZwA==" crossorigin="anonymous"></script>
<script type="text/javascript" charset="utf-8">
window.location="/list";
</script>
<p>tag was sucsesfully added</p>
</html>
"""
else:
2023-09-19 20:23:55 +00:00
return render_template("error.html", error_code="757")
2023-07-28 21:30:45 +00:00
2023-11-24 17:26:54 +00:00
@app.route("/change", methods=['POST'])
2023-07-28 21:30:45 +00:00
def change():
db = get_db()
c = db.cursor()
try:
2023-11-24 17:26:54 +00:00
user_id = request.form["id"]
2023-11-24 19:38:55 +00:00
change = float(request.form["change"])
2023-07-28 21:30:45 +00:00
except:
2023-09-19 20:23:55 +00:00
return render_template("error.html", error_code="095")
2023-07-28 21:30:45 +00:00
c.execute(f"SELECT * FROM users WHERE id=?", [user_id])
users = c.fetchall()
if users != []:
balance_old = users[0][2]
2023-11-25 17:22:13 +00:00
c.execute(f"UPDATE users SET balance = balance + {change*100} WHERE id={user_id}")
2023-07-28 21:30:45 +00:00
db.commit()
c.execute(f"SELECT * FROM users WHERE id={user_id}")
user = c.fetchone()
2023-11-24 23:05:33 +00:00
log(type="balance", userid=user[0], before=balance_old, after=user[2])
2023-07-28 21:30:45 +00:00
socketio.emit("update", "update")
return """<html>
<script src="https://cdnjs.cloudflare.com/ajax/libs/socket.io/4.0.1/socket.io.js" integrity="sha512-q/dWJ3kcmjBLU4Qc47E4A9kTB4m3wuTY7vkFJDTZKjTs8jhyGQnaUrxa0Ytd0ssMZhbNua9hE+E7Qv1j+DyZwA==" crossorigin="anonymous"></script>
<script type="text/javascript" charset="utf-8">
window.location="/list";
</script>
</html>
"""
else:
2023-09-19 20:23:55 +00:00
return render_template("error.html", error_code="043")
2023-07-28 21:30:45 +00:00
2023-11-24 17:26:54 +00:00
@app.route("/addtag", methods=['POST'])
2023-08-16 19:36:23 +00:00
def get_addtag_request():
2023-07-28 21:30:45 +00:00
try:
2023-11-24 17:26:54 +00:00
user_id = int(request.form["id"])
2023-07-28 21:30:45 +00:00
except: #except im Normalen Code!
2023-09-19 20:23:55 +00:00
return render_template("error.html", error_code="095")
2023-07-28 21:30:45 +00:00
session_id = uuid.uuid4()
session[id] = session_id
2023-08-16 19:36:23 +00:00
user_queue.put([user_id, "add", session_id])
2023-07-28 21:30:45 +00:00
return render_template("addtag.html", user=user_id)
@socketio.on('addtag')
def request_addtag(data):
global finished
global message
join_room(session[id])
2023-08-16 19:36:23 +00:00
if len(user_queue.queue) > 0:
2023-11-24 17:37:10 +00:00
user = user_queue.queue[0]
2023-07-28 21:30:45 +00:00
if user == [data["data"], "add", session[id]]:
socketio.emit("wait", "wait", to=session[id])
i = 0
while finished != [data["data"], "add", session[id]]:
time.sleep(1)
i += 1
if i > 20:
socketio.emit("error", "352", to=session[id])
2023-08-16 19:36:23 +00:00
notimportant = user_queue.get()
2023-07-28 21:30:45 +00:00
break
else:
finished = None
socketio.emit("finished", f"{message}", to=session[id])
else:
socketio.emit("busy", "busy", to=session[id])
else:
socketio.emit("error", "418", to=session[id])
leave_room(session[id])
2023-11-24 17:26:54 +00:00
@app.route("/removetag", methods=['POST'])
2023-08-16 19:36:23 +00:00
def get_removetag_request():
2023-07-28 21:30:45 +00:00
try:
2023-11-24 17:26:54 +00:00
user_id = int(request.form["id"])
2023-07-28 21:30:45 +00:00
except: #except im Normalen Code!
2023-09-19 20:23:55 +00:00
return render_template("error.html", error_code="043")
2023-11-25 18:28:05 +00:00
try:
tag_id = int(request.form["tagid"])
except:
session_id = uuid.uuid4()
session[id] = session_id
user_queue.put([user_id, "remove", session_id])
return render_template("removetag.html", user=user_id)
else:
db = get_db()
c = db.cursor()
c.execute(f"SELECT * FROM tags WHERE (tagid = ? AND userid = ?)", [tag_id, user_id])
if c.fetchall != []:
c.execute(f"DELETE FROM tags WHERE (tagid = ? AND userid = ?)", [tag_id, user_id])
db.commit()
message = f"Removed {tag_id} from user {user_id}"
log(type="removetag", userid=user_id, before=tag_id)
2023-11-25 19:11:15 +00:00
return f"""
<html>
<script>
window.location="/"
</script>
</html>
"""
2023-11-25 18:28:05 +00:00
else:
return render_template("error.html", error_code="054")
2023-07-28 21:30:45 +00:00
@socketio.on('removetag')
def request_removetag(data):
global finished
global message
join_room(session[id])
2023-08-16 19:36:23 +00:00
if len(user_queue.queue) > 0:
2023-11-25 19:11:15 +00:00
queue_item = user_queue.queue[0]
2023-07-28 21:30:45 +00:00
user = queue_item[0]
if queue_item == [data["data"], "remove", session[id]]:
socketio.emit("wait", "wait", to=session[id])
i = 0
while finished != [data["data"], "remove", session[id]]:
time.sleep(1)
i += 1
if i > 20:
socketio.emit("error", "352", to=session[id])
2023-08-16 19:36:23 +00:00
notimportant = user_queue.get()
2023-07-28 21:30:45 +00:00
break
else:
finished = None
socketio.emit("finished", f"{message}", to=session[id])
else:
socketio.emit("busy", "busy", to=session[id])
else:
socketio.emit("error", "418", to=session[id])
leave_room(session[id])
#api
2023-12-02 17:14:34 +00:00
@app.route("/api/balance", methods=['POST'])
2023-07-28 21:30:45 +00:00
def api_change():
db = get_db()
c = db.cursor()
2023-12-02 17:14:34 +00:00
userid = request.form["id"]
2023-07-28 21:30:45 +00:00
c.execute("SELECT * FROM users WHERE id=?", [userid])
user_list = c.fetchall()
if user_list != []:
user = user_list[0]
try:
change = int(request.args.get("change"))
except:
2023-11-25 17:26:47 +00:00
change = -1.5
2023-11-25 17:22:13 +00:00
c.execute(f"UPDATE users SET balance = balance + {change*100} WHERE id={user[0]}")
2023-07-28 21:30:45 +00:00
db.commit()
c.execute(f"SELECT * FROM users WHERE id = {userid}")
user_new = c.fetchone()
2023-11-24 23:05:33 +00:00
log(type="balance", userid=user[0], before=user[2], after=user_new[2])
2023-07-28 21:30:45 +00:00
socketio.emit("update", "update")
return make_response(json.dumps({"mode":"balance", "username":user[1], "balance":user_new[2]}))
else:
return make_response(json.dumps({"mode":"error","error":"043"}))
2023-11-25 17:42:53 +00:00
@app.route("/api/tag_id", methods=['GET', 'POST'])
2023-07-28 21:30:45 +00:00
def get_id():
global finished
global message
db = get_db()
c = db.cursor()
2023-11-25 17:42:53 +00:00
try:
tag_id = request.form["id"]
2024-02-14 19:52:50 +00:00
print(tag_id)
2023-11-25 17:42:53 +00:00
except:
tag_id = request.args.get("id")
2023-07-28 21:30:45 +00:00
c.execute(f"SELECT * FROM tags WHERE tagid=?", [tag_id])
tag_list = c.fetchall()
2023-08-16 19:36:23 +00:00
if user_queue.qsize() > 0:
queue_item = user_queue.get()
2023-07-28 21:30:45 +00:00
user = queue_item[0]
state = queue_item[1]
c.execute(f"SELECT * FROM users WHERE id=?", [user])
2023-11-25 15:59:51 +00:00
user = c.fetchone()
2023-11-24 23:14:51 +00:00
user_id = user[0]
username = user[1]
2023-07-28 21:30:45 +00:00
if state == "add":
c.execute(f"SELECT * FROM tags WHERE tagid={tag_id}")
if c.fetchall() != []:
message = "Error: 170"
finished = queue_item
return make_response(json.dumps({"mode":"error","error":"170"}))
else:
2023-11-25 15:59:51 +00:00
c.execute(f"INSERT OR IGNORE INTO tags (tagid, userid) VALUES ({tag_id}, ?)", [user_id])
2023-11-24 23:13:17 +00:00
db.commit()
2023-11-25 15:59:51 +00:00
message = f"Added {tag_id} to {username}"
2023-11-24 23:05:33 +00:00
log(type="addtag", userid=user_id ,after=tag_id)
2023-07-28 21:30:45 +00:00
finished = queue_item
2023-11-08 20:40:41 +00:00
return make_response(json.dumps({"mode":"message","username":"{}".format(username),"message":"A tag was added"}))
2023-07-28 21:30:45 +00:00
elif state == "remove":
2023-11-25 15:59:51 +00:00
c.execute(f"SELECT * FROM tags WHERE (tagid = {tag_id} AND userid = ?)", [user_id])
2023-07-28 21:30:45 +00:00
tags = c.fetchall()
if tags != []:
2023-11-25 15:59:51 +00:00
c.execute(f"DELETE FROM tags WHERE (tagid = {tag_id} AND userid = ?)", [user_id])
2023-11-24 23:13:17 +00:00
db.commit()
2023-11-25 15:59:51 +00:00
message = f"Removed {tag_id} from {username}"
2023-11-24 23:05:33 +00:00
log(type="removetag", userid=user_id, before=tag_id)
2023-07-28 21:30:45 +00:00
finished = queue_item
2023-11-08 20:40:41 +00:00
return make_response(json.dumps({"mode":"message","username":"{}".format(username),"message":"A tag was removed"}))
2023-07-28 21:30:45 +00:00
else:
message = "054"
finished = queue_item
return make_response(json.dumps({"mode":"error","error":"054"}))
finished = queue_item
socketio.emit("update", "update")
return make_response(json.dumps({"mode":"error","error":"418"}))
elif tag_list != []:
tag = tag_list[0]
c.execute(f"SELECT * FROM users WHERE id={tag[1]}")
user_list = c.fetchall()
if user_list != []:
balance_old = user_list[0][2]
2023-08-16 19:36:23 +00:00
if user_queue.qsize() == 0:
2023-11-25 17:22:13 +00:00
c.execute(f"UPDATE users SET balance = balance - 150 WHERE id={tag[1]}")
2023-07-28 21:30:45 +00:00
db.commit()
c.execute(f"SELECT * FROM users WHERE id={tag[1]}")
user = c.fetchone()
2023-11-24 23:05:33 +00:00
log(type="balance", userid=user[0], before=balance_old, after=user[2])
2023-07-28 21:30:45 +00:00
socketio.emit("update", "update")
2023-11-25 17:26:47 +00:00
return make_response(json.dumps({"mode":"balance", "username":user[1], "balance":user[2]/100}))
2023-07-28 21:30:45 +00:00
else:
return make_response(json.dumps({"mode":"error", "error":"043"}))
socketio.emit("update", "update")
return make_response(json.dumps({"mode":"error","error":"054"}))
#Documentation
@app.route("/documentation")
def documentation():
return render_template("documentation.html")
2024-02-14 19:52:50 +00:00
return {"app":app,"socketio":socketio}