You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
91 lines
2.5 KiB
Python
91 lines
2.5 KiB
Python
3 years ago
|
from flask import (Blueprint, flash, g, redirect,
|
||
|
|
render_template, request, session, url_for)
|
||
|
|
|
||
|
|
from exquisite_branch.db import get_db
|
||
|
|
from werkzeug.exceptions import abort
|
||
|
|
|
||
|
|
from shortuuid import uuid
|
||
|
|
|
||
|
|
bp = Blueprint('draw', __name__, url_prefix='/draw')
|
||
|
|
|
||
|
|
|
||
|
|
@bp.route('/<parent>', methods=('GET', 'POST'))
|
||
|
|
def draw(parent=None):
|
||
|
|
db = get_db()
|
||
|
|
|
||
|
|
if request.method == 'POST':
|
||
|
|
content = request.form.get('content')
|
||
|
|
branch = request.form.get('branch')
|
||
|
|
if request.is_json:
|
||
|
|
data = request.get_json()
|
||
|
|
content = data['content']
|
||
|
|
branch = data['branch']
|
||
|
|
|
||
|
|
|
||
|
|
db.execute(
|
||
|
|
'INSERT INTO branches (content, parent, branch) VALUES (?, ?, ?)',
|
||
|
|
(content, parent, branch,)
|
||
|
|
)
|
||
|
|
db.commit()
|
||
|
|
print(url_for('share.share', branch=f"{branch}"))
|
||
|
|
return redirect(url_for('share.share', branch=branch))
|
||
|
|
|
||
|
|
branch = uuid()
|
||
|
|
|
||
|
|
previous = db.execute(
|
||
|
|
"SELECT content, branch, parent FROM branches"
|
||
|
|
" WHERE branch = ?",
|
||
|
|
(parent,)
|
||
|
|
).fetchone()
|
||
|
|
|
||
|
|
if previous is None:
|
||
|
|
abort(404, f"Previous with id {parent} doesn't exist")
|
||
|
|
|
||
|
|
return render_template('draw.html', parent=parent, content=previous['content'], branch=branch)
|
||
|
|
|
||
|
|
|
||
|
|
@bp.route('/last', methods=('GET', 'POST'))
|
||
|
|
def last():
|
||
|
|
|
||
|
|
branch = uuid()
|
||
|
|
db = get_db()
|
||
|
|
previous = db.execute(
|
||
|
|
'SELECT * FROM branches ORDER BY id DESC LIMIT 1'
|
||
|
|
).fetchone()
|
||
|
|
|
||
|
|
parent = previous['branch']
|
||
|
|
|
||
|
|
if request.method == 'POST':
|
||
|
|
content = request.form['content']
|
||
|
|
branch = request.form['branch']
|
||
|
|
|
||
|
|
db.execute(
|
||
|
|
'INSERT INTO branches (content, parent, branch) VALUES (?, ?, ?)',
|
||
|
|
(content, parent, branch,)
|
||
|
|
)
|
||
|
|
db.commit()
|
||
|
|
print(url_for('share.share', branch=f"{branch}"))
|
||
|
|
return redirect(url_for('share.share', branch=branch))
|
||
|
|
|
||
|
|
return render_template('draw.html', parent=parent, content=previous['content'], branch=branch)
|
||
|
|
|
||
|
|
|
||
|
|
@bp.route('/', methods=('GET', 'POST'))
|
||
|
|
def new():
|
||
|
|
db = get_db()
|
||
|
|
branch = uuid()
|
||
|
|
parent = 'NEW'
|
||
|
|
|
||
|
|
if request.method == 'POST':
|
||
|
|
content = request.form['content']
|
||
|
|
branch = request.form['branch']
|
||
|
|
|
||
|
|
db.execute(
|
||
|
|
'INSERT INTO branches (content, parent, branch) VALUES (?, ?, ?)',
|
||
|
|
(content, parent, branch,)
|
||
|
|
)
|
||
|
|
db.commit()
|
||
|
|
return redirect(url_for('share.share', branch=branch))
|
||
|
|
|
||
|
|
return render_template('draw.html', parent=parent, branch=branch)
|