-
Notifications
You must be signed in to change notification settings - Fork 5.3k
Expand file tree
/
Copy pathapp.py
More file actions
87 lines (72 loc) · 2.67 KB
/
app.py
File metadata and controls
87 lines (72 loc) · 2.67 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
import base64
from flask import Flask, redirect, render_template, request, session, url_for
from pygments import highlight
from pygments.formatters import HtmlFormatter
from pygments.lexers import Python3Lexer
from pygments.styles import get_all_styles
from utils import take_screenshot_from_url
app = Flask(__name__)
app.secret_key = (
"AddYourSecretKeyHere" # See the README.md file for instructions
)
PLACEHOLDER_CODE = "print('Hello, World!')"
DEFAULT_STYLE = "monokai"
NO_CODE_FALLBACK = "# No Code Entered"
@app.route("/", methods=["GET"])
def code():
if session.get("code") is None:
session["code"] = PLACEHOLDER_CODE
lines = session["code"].split("\n")
context = {
"message": "Paste Your Python Code 🐍",
"code": session["code"],
"num_lines": len(lines),
"max_chars": len(max(lines, key=len)),
}
return render_template("code_input.html", **context)
@app.route("/save_code", methods=["POST"])
def save_code():
session["code"] = request.form.get("code") or NO_CODE_FALLBACK
return redirect(url_for("code"))
@app.route("/reset_session", methods=["POST"])
def reset_session():
session.clear()
session["code"] = PLACEHOLDER_CODE
return redirect(url_for("code"))
@app.route("/style", methods=["GET"])
def style():
if session.get("style") is None:
session["style"] = DEFAULT_STYLE
formatter = HtmlFormatter(style=session["style"])
context = {
"message": "Select Your Style 🎨",
"all_styles": list(get_all_styles()),
"style": session["style"],
"style_definitions": formatter.get_style_defs(),
"style_bg_color": formatter.style.background_color,
"highlighted_code": highlight(
session["code"], Python3Lexer(), formatter
),
}
return render_template("style_selection.html", **context)
@app.route("/save_style", methods=["POST"])
def save_style():
if request.form.get("style") is not None:
session["style"] = request.form.get("style")
if request.form.get("code") is not None:
session["code"] = request.form.get("code") or NO_CODE_FALLBACK
return redirect(url_for("style"))
@app.route("/image", methods=["GET"])
def image():
session_data = {
"name": app.config["SESSION_COOKIE_NAME"],
"value": request.cookies.get(app.config["SESSION_COOKIE_NAME"]),
"url": request.host_url,
}
target_url = request.host_url + url_for("style")
image_bytes = take_screenshot_from_url(target_url, session_data)
context = {
"message": "Done! 🎉",
"image_b64": base64.b64encode(image_bytes).decode("utf-8"),
}
return render_template("image.html", **context)