First version of Grafting a Tree book generator.
parent
30bd684622
commit
628b681940
Binary file not shown.
@ -0,0 +1,48 @@
|
|||||||
|
#!/usr/bin/env/ python
|
||||||
|
from sys import stdout
|
||||||
|
from flask import Flask, Response, render_template, request
|
||||||
|
from pagedjs import make_pdf
|
||||||
|
from graft_a_tree import graft_a_tree
|
||||||
|
from io import StringIO
|
||||||
|
import sys
|
||||||
|
import textwrap
|
||||||
|
|
||||||
|
BASEURL = ''
|
||||||
|
MIN_GROW_YEARS = 0
|
||||||
|
MAX_GROW_YEARS = 25
|
||||||
|
|
||||||
|
app = Flask(__name__)
|
||||||
|
|
||||||
|
def wrap (text, width):
|
||||||
|
return'\n'.join([(' \\\n').join(textwrap.wrap(line, width=width - 2)) for line in text.splitlines()])
|
||||||
|
|
||||||
|
with open('graft_a_tree.py', 'r') as script:
|
||||||
|
source_code = script.read()
|
||||||
|
|
||||||
|
@app.route('{}/'.format(BASEURL))
|
||||||
|
def index():
|
||||||
|
return render_template('index.html', BASEURL=BASEURL)
|
||||||
|
|
||||||
|
@app.route('{}/graft'.format(BASEURL), methods=['POST'])
|
||||||
|
def graft():
|
||||||
|
years = max(MIN_GROW_YEARS, min(MAX_GROW_YEARS, int(request.form['years'])))
|
||||||
|
|
||||||
|
with StringIO() as script_stdout:
|
||||||
|
default_stdout = sys.stdout
|
||||||
|
sys.stdout = script_stdout
|
||||||
|
graft_a_tree(years)
|
||||||
|
content = script_stdout.getvalue()
|
||||||
|
sys.stdout = default_stdout
|
||||||
|
|
||||||
|
html = render_template('book.html', content=content, source_code=wrap(source_code, 100).replace(' ', ' ').replace('\n', '<br />'))
|
||||||
|
# print(html)
|
||||||
|
print('html rendered')
|
||||||
|
pdf = make_pdf(html)
|
||||||
|
print('made pdf')
|
||||||
|
r = Response(pdf, mimetype='application/pdf')
|
||||||
|
|
||||||
|
r.headers.extend({
|
||||||
|
'Content-Disposition': 'attachment; filename="grafted-tree.pdf"'
|
||||||
|
})
|
||||||
|
|
||||||
|
return r
|
||||||
@ -0,0 +1,3 @@
|
|||||||
|
nltk
|
||||||
|
wikipedia
|
||||||
|
flask
|
||||||
@ -0,0 +1,3 @@
|
|||||||
|
export FLASK_APP=app.py
|
||||||
|
export FLASK_ENV=development
|
||||||
|
flask run
|
||||||
@ -0,0 +1,53 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="en">
|
||||||
|
<head>
|
||||||
|
<meta charset="UTF-8">
|
||||||
|
<meta http-equiv="X-UA-Compatible" content="IE=edge">
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||||
|
<title>Document</title>
|
||||||
|
<style>
|
||||||
|
body, html {
|
||||||
|
font-size: 120%;
|
||||||
|
}
|
||||||
|
|
||||||
|
h1 {
|
||||||
|
page-break-after: right;
|
||||||
|
}
|
||||||
|
|
||||||
|
.tree{
|
||||||
|
page-break-before: always;
|
||||||
|
page: tree;
|
||||||
|
}
|
||||||
|
|
||||||
|
code {
|
||||||
|
page-break-before: right;
|
||||||
|
font-size: 57.5%;
|
||||||
|
line-height: 1.2;
|
||||||
|
}
|
||||||
|
|
||||||
|
.tree.first {
|
||||||
|
page-break-before: right;
|
||||||
|
page: initial;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
.tree[data-year] {
|
||||||
|
string-set: year attr(data-year);
|
||||||
|
}
|
||||||
|
|
||||||
|
@page {
|
||||||
|
size: a5 portrait;
|
||||||
|
}
|
||||||
|
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<h1>Grafted tree</h1>
|
||||||
|
|
||||||
|
{% autoescape false %}
|
||||||
|
{{ content }}
|
||||||
|
|
||||||
|
<code>{{ source_code }}</code>
|
||||||
|
{% endautoescape %}
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
@ -0,0 +1,97 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="en">
|
||||||
|
<head>
|
||||||
|
<meta charset="UTF-8">
|
||||||
|
<meta http-equiv="X-UA-Compatible" content="IE=edge">
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||||
|
<title>Document</title>
|
||||||
|
<style>
|
||||||
|
[data-message] {
|
||||||
|
display: none;
|
||||||
|
font-size: var(--font-size-smaller);
|
||||||
|
line-height: var(--line-height--smaller);
|
||||||
|
font-style: italic;
|
||||||
|
margin: var(--line-height--smaller) 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
[data-message][data-active] {
|
||||||
|
display: block;
|
||||||
|
}
|
||||||
|
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<h1>Grafting a tree</h1>
|
||||||
|
<p>
|
||||||
|
<form action="{{ BASEURL }}/graft" method="POST">
|
||||||
|
Graft a tree for <input type="number" name="years" value="3" max="25" min="0" size="4" /> years.
|
||||||
|
<button type="submit">Graft</button>
|
||||||
|
</form>
|
||||||
|
</p>
|
||||||
|
<p data-message="working">
|
||||||
|
Grafting your tree. This might take a while.
|
||||||
|
</p>
|
||||||
|
<p data-message="error">
|
||||||
|
Something went wrong.
|
||||||
|
</p>
|
||||||
|
<script>
|
||||||
|
function getFilename (headers, fallback) {
|
||||||
|
if (headers.has('Content-Disposition')) {
|
||||||
|
header = headers.get('Content-Disposition')
|
||||||
|
matches = header.match(/filename="(.+)"/)
|
||||||
|
|
||||||
|
if (matches.length == 2) {
|
||||||
|
return matches[1]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return fallback;
|
||||||
|
}
|
||||||
|
|
||||||
|
(function () {
|
||||||
|
if (fetch) {
|
||||||
|
var form = document.querySelector('form'),
|
||||||
|
button = form.querySelector('button'),
|
||||||
|
messageWorking = document.querySelector('[data-message="working"]'),
|
||||||
|
messageError = document.querySelector('[data-message="error"]');
|
||||||
|
|
||||||
|
form.addEventListener('submit', function (e) {
|
||||||
|
e.preventDefault();
|
||||||
|
button.disabled = true;
|
||||||
|
delete messageError.dataset.active;
|
||||||
|
messageWorking.dataset.active = true;
|
||||||
|
const data = new FormData(form);
|
||||||
|
|
||||||
|
fetch(form.action, {
|
||||||
|
method: "POST",
|
||||||
|
body: data
|
||||||
|
}).then(function (r) {
|
||||||
|
console.log(r);
|
||||||
|
r.blob().then(function (blob) {
|
||||||
|
var filename = getFilename(r.headers, 'Grafted-tree.pdf');
|
||||||
|
const a = document.createElement('a');
|
||||||
|
a.setAttribute('href', URL.createObjectURL(blob));
|
||||||
|
a.setAttribute('download', filename);
|
||||||
|
|
||||||
|
if (document.createEvent) {
|
||||||
|
const event = document.createEvent('MouseEvents');
|
||||||
|
event.initEvent('click', true, true);
|
||||||
|
a.dispatchEvent(event);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
a.click();
|
||||||
|
}
|
||||||
|
delete messageWorking.dataset.active;
|
||||||
|
button.disabled = false;
|
||||||
|
});
|
||||||
|
}).catch(function() {
|
||||||
|
delete messageWorking.dataset.active;
|
||||||
|
messageError.dataset.active = true;
|
||||||
|
button.disabled = false;
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}
|
||||||
|
})();
|
||||||
|
</script>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
Loading…
Reference in New Issue