SuperScriptMaterializer/SuperScriptViewer/ServerCore.py

import CustomConfig

from flask import Flask
from flask import g
from flask import render_template
from flask import url_for
from flask import request
from flask import abort
from flask import redirect

from functools import reduce
import sqlite3
import json
import ServerStruct as ss

app = Flask(__name__)

# =============================================database
def get_db():
    db = getattr(g, '_database', None)
    if db is None:
        db = g._database = sqlite3.connect(CustomConfig.decorated_db)
    return db

@app.teardown_appcontext
def close_connection(exception):
    db = getattr(g, '_database', None)
    if db is not None:
        db.close()

# =============================================template func
@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']]

# =============================================route
@app.route('/', methods=['GET'])
def nospecHandle():
    return redirect(url_for('indexHandle'))

@app.route('/help', methods=['GET'])
def helpHandle():
    return render_template("help.html")

@app.route('/about', methods=['GET'])
def aboutHandle():
    return render_template("about.html", static_icon = url_for('static', filename='icon.png'))

@app.route('/index', 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('/viewer/<path:scriptPath>', methods=['GET'])
def viewerHandle(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('/viewer/<path:scriptPath>', methods=['POST'])
def actionHandle(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()