120 lines
3.5 KiB
Python
120 lines
3.5 KiB
Python
from flask import Flask
|
|
from flask import g
|
|
from flask import render_template
|
|
from flask import url_for
|
|
from flask import request
|
|
|
|
from functools import reduce
|
|
import sqlite3
|
|
import json
|
|
import ServerStruct as ss
|
|
|
|
app = Flask(__name__)
|
|
|
|
def get_db():
|
|
db = getattr(g, '_database', None)
|
|
if db is None:
|
|
db = g._database = sqlite3.connect('decorate.db')
|
|
return db
|
|
|
|
@app.teardown_appcontext
|
|
def close_connection(exception):
|
|
db = getattr(g, '_database', None)
|
|
if db is not None:
|
|
db.close()
|
|
|
|
@app.template_global(name = 'pinDecoder')
|
|
def block_pin_decoder(target):
|
|
return json.loads(target)
|
|
|
|
@app.template_global(name = 'pinDecoder2')
|
|
def block_pin_decoder2(target):
|
|
vab = json.loads(target)
|
|
return [vab['name'], vab['type']]
|
|
|
|
@app.route('/', methods=['GET'])
|
|
def indexHandle():
|
|
cur = get_db().cursor()
|
|
cur.execute("SELECT [graph], [graph_name], [belong_to] FROM graph WHERE [index] != -1 ORDER BY [belong_to], [index] ASC;")
|
|
data = {}
|
|
for i in cur.fetchall():
|
|
if i[2] not in data.keys():
|
|
data[i[2]] = [ss.ScriptItem(i[1], i[0])]
|
|
else:
|
|
data[i[2]].append(ss.ScriptItem(i[1], i[0]))
|
|
|
|
return render_template('index.html', scripts = data)
|
|
|
|
@app.route('/<path:scriptPath>', methods=['GET'])
|
|
def scriptHandle(scriptPath):
|
|
# comput hamburger
|
|
pathSlice = scriptPath.split('/')
|
|
cur = get_db().cursor()
|
|
cur.execute("SELECT [graph], [graph_name] FROM graph WHERE [graph] IN ({});".format(','.join(pathSlice)))
|
|
hamburgerMap = {}
|
|
hamburger = []
|
|
for i in cur.fetchall():
|
|
hamburgerMap[str(i[0])] = i[1]
|
|
currentHamburger = hamburgerMap[pathSlice[-1]]
|
|
|
|
for i in range(len(pathSlice) - 1):
|
|
hamburger.append(ss.HamburgerItem(hamburgerMap[pathSlice[i]], reduce(lambda x,y: x + '/' + y, pathSlice[0:i + 1], '')))
|
|
|
|
# gei w/h
|
|
cur.execute('SELECT [width], [height] FROM graph WHERE [graph] == ?', (pathSlice[-1], ))
|
|
cache = cur.fetchone()
|
|
width = cache[0]
|
|
height = cache[1]
|
|
|
|
# get blocks
|
|
cur.execute('SELECT * FROM block WHERE [belong_to_graph] == ?', (pathSlice[-1], ))
|
|
dbBlocks = cur.fetchall()
|
|
|
|
# get cells
|
|
cur.execute("SELECT * FROM cell WHERE [belong_to_graph] == ?", (pathSlice[-1], ))
|
|
dbCells = cur.fetchall()
|
|
|
|
# get links
|
|
cur.execute("SELECT * FROM link WHERE [belong_to_graph] == ?", (pathSlice[-1], ))
|
|
dbLinks = cur.fetchall()
|
|
|
|
# render
|
|
return render_template('viewer.html',
|
|
currentPath = scriptPath,
|
|
gWidth = width,
|
|
gHeight = height,
|
|
hamburgerHistory = hamburger,
|
|
static_css = url_for('static', filename='site.css'),
|
|
static_js = url_for('static', filename='site.js'),
|
|
hamburgerCurrent = currentHamburger,
|
|
blocks = dbBlocks,
|
|
cells = dbCells,
|
|
links = dbLinks)
|
|
|
|
@app.route('/<path:scriptPath>', methods=['POST'])
|
|
def infoMoveHandle(scriptPath):
|
|
cache = request.form['operation']
|
|
if cache == "info":
|
|
return infoHandle(request.form['target'])
|
|
elif cache == "move":
|
|
return moveHandle(request.form['target'])
|
|
else:
|
|
abort(400)
|
|
|
|
def infoHandle(target):
|
|
cur = get_db().cursor()
|
|
cur.execute("SELECT [field], [data] FROM info WHERE [target] == ?", (target, ))
|
|
data = {}
|
|
for i in cur.fetchall():
|
|
data[i[0]] = i[1]
|
|
|
|
return data
|
|
|
|
def moveHandle(target):
|
|
return {}
|
|
|
|
def run():
|
|
app.run()
|
|
|
|
|