113 lines
2.6 KiB
Python
Executable File
113 lines
2.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import bottle
|
|
import io
|
|
import qrcode
|
|
import qrcode.image.styledpil as qrstyled
|
|
import sqlite3
|
|
|
|
|
|
application = bottle.default_app()
|
|
connection = sqlite3.connect("db.sqlite")
|
|
cursor = connection.cursor()
|
|
cursor.execute(
|
|
"""
|
|
CREATE TABLE IF NOT EXISTS visitors (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
event TEXT,
|
|
time DATETIME,
|
|
name TEXT,
|
|
address TEXT,
|
|
contact TEXT
|
|
);
|
|
"""
|
|
)
|
|
|
|
|
|
def url(route):
|
|
return "%s://%s/%s" % (
|
|
bottle.request.urlparts.scheme,
|
|
bottle.request.urlparts.netloc,
|
|
route,
|
|
)
|
|
|
|
|
|
@application.get("/")
|
|
@bottle.view("template/new.tpl")
|
|
def new_get():
|
|
pass
|
|
|
|
|
|
@application.post("/")
|
|
@bottle.view("template/new.tpl")
|
|
def new_post():
|
|
event = bottle.request.forms.get("event", "")
|
|
if event:
|
|
bottle.redirect(url("qr/" + event))
|
|
return dict(event=event, feedback=True)
|
|
|
|
|
|
@application.get("/static/<file:path>")
|
|
def static_get(file):
|
|
return bottle.static_file(file, root="static")
|
|
|
|
|
|
@application.get("/qr/<event>")
|
|
@bottle.view("template/qr.tpl")
|
|
def qr_get(event):
|
|
return dict(event=event, url=url("event/" + event))
|
|
|
|
|
|
@application.get("/qr/img/<event>.png")
|
|
def qr_img_get(event):
|
|
qr = qrcode.QRCode(error_correction=qrcode.constants.ERROR_CORRECT_H, box_size=20)
|
|
qr.add_data(url("event/" + event))
|
|
image = qr.make_image(
|
|
image_factory=qrstyled.StyledPilImage,
|
|
embeded_image_path="heizhaus_logo_square.png",
|
|
)
|
|
buffer = io.BytesIO()
|
|
image.save(buffer, format="PNG")
|
|
bottle.response.content_type = "image/png"
|
|
return buffer.getvalue()
|
|
|
|
|
|
@application.get("/event/<event>")
|
|
@bottle.view("template/event.tpl")
|
|
def event_get(event):
|
|
return dict(event=event)
|
|
|
|
|
|
@application.post("/event/<event>")
|
|
@bottle.view("template/event.tpl")
|
|
def event_post(event):
|
|
data = dict(feedback=True, success=True)
|
|
for field in ("event", "name", "address", "contact"):
|
|
data[field] = bottle.request.forms.get(field, "")
|
|
data["success"] = data["success"] and data[field]
|
|
if data["success"]:
|
|
cursor.execute(
|
|
"""
|
|
INSERT INTO visitors (event, time, name, address, contact)
|
|
VALUES (:event, datetime('now'), :name, :address, :contact);
|
|
""",
|
|
data,
|
|
)
|
|
connection.commit()
|
|
return data
|
|
|
|
|
|
@application.get("/cleanup")
|
|
def cleanup():
|
|
cursor.execute(
|
|
"""
|
|
DELETE FROM visitors WHERE time < datetime('now', '-1 month');
|
|
"""
|
|
)
|
|
connection.commit()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
bottle.run(host="0.0.0.0", port=8080)
|
|
else:
|
|
application.catchall = False
|