coconut-leaf/src/server.py

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 json
import os

import config
import database
import utils

app = Flask(__name__)
calendar_db = database.CalendarDatabase()

# render_static_resources = None

# =============================================database
'''
def get_database():
    db = getattr(g, '_database', None)
    if db is None:
        db = database.CalendarDatabase()
        db.open()
    return db

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

# ============================================= static page route

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

@app.route('/web/home', methods=['GET'])
def web_homeHandle():
    # UpdateStaticResources()
    return render_template("home.html")

@app.route('/web/calendar', methods=['GET'])
def web_calendarHandle():
    # UpdateStaticResources()
    return render_template("calendar.html")

@app.route('/web/todo', methods=['GET'])
def web_todoHandle():
    # UpdateStaticResources()
    return render_template("todo.html")

@app.route('/web/admin', methods=['GET'])
def web_adminHandle():
    # UpdateStaticResources()
    return render_template("admin.html")

@app.route('/web/login', methods=['GET'])
def web_loginHandle():
    # UpdateStaticResources()
    return render_template("login.html")

@app.route('/web/collection', methods=['GET'])
def web_collectionHandle():
    # UpdateStaticResources()
    return render_template("collection.html")

@app.route('/web/eventAdd', methods=['GET'])
def web_eventAddHandle():
    # UpdateStaticResources()
    return render_template("event.html",
        uuidPath=''
    )

@app.route('/web/eventUpdate/<path:uuidPath>', methods=['GET'])
def web_eventUpdateHandle(uuidPath):
    # UpdateStaticResources()
    return render_template("event.html",
        uuidPath = uuidPath
    )

# ============================================= query page route

# ================================ common

@app.route('/api/common/salt', methods=['POST'])
def api_common_saltHandle():
    return SmartDbCaller(calendar_db.common_salt, 
    (('username', str, False), ),
    None)

@app.route('/api/common/login', methods=['POST'])
def api_common_loginHandle():
    # construct client data first
    clientUa = request.user_agent.string
    if request.headers.getlist("X-Forwarded-For"):
        clientIp = request.headers.getlist("X-Forwarded-For")[0]
    else:
        clientIp = request.remote_addr

    return SmartDbCaller(calendar_db.common_login, 
    (('username', str, False), 
    ('password', str, False),
    ('clientUa', str, False),
    ('clientIp', str, False)),
    {
        'clientUa': clientUa,
        'clientIp': clientIp
    })

@app.route('/api/common/webLogin', methods=['POST'])
def api_common_webLoginHandle():
    # construct client data first
    clientUa = request.user_agent.string
    if request.headers.getlist("X-Forwarded-For"):
        clientIp = request.headers.getlist("X-Forwarded-For")[0]
    else:
        clientIp = request.remote_addr

    return SmartDbCaller(calendar_db.common_webLogin, 
    (('username', str, False), 
    ('password', str, False),
    ('clientUa', str, False),
    ('clientIp', str, False)),
    {
        'clientUa': clientUa,
        'clientIp': clientIp
    })

@app.route('/api/common/logout', methods=['POST'])
def api_common_logoutHandle():
    return SmartDbCaller(calendar_db.common_logout, 
    (('token', str, False), ),
    None)

@app.route('/api/common/tokenValid', methods=['POST'])
def api_common_tokenValidHandle():
    return SmartDbCaller(calendar_db.common_tokenValid, 
    (('token', str, False), ),
    None)

# ================================ calendar

@app.route('/api/calendar/getFull', methods=['POST'])
def api_calendar_getFullHandle():
    return SmartDbCaller(calendar_db.calendar_getFull, 
    (('token', str, False),
    ('startDateTime', int, False),
    ('endDateTime', int, False)),
    None)

@app.route('/api/calendar/getList', methods=['POST'])
def api_calendar_getListHandle():
    return SmartDbCaller(calendar_db.calendar_getList, 
    (('token', str, False),
    ('startDateTime', int, False),
    ('endDateTime', int, False)),
    None)

@app.route('/api/calendar/getDetail', methods=['POST'])
def api_calendar_getDetailHandle():
    return SmartDbCaller(calendar_db.calendar_getDetail, 
    (('token', str, False),
    ('uuid', str, False)),
    None)

@app.route('/api/calendar/update', methods=['POST'])
def api_calendar_updateHandle():
    return SmartDbCaller(calendar_db.calendar_update, 
    (('token', str, False),
    ('uuid', str, False),
    ('belongTo', str, True),
    ('title', str, True),
    ('description', str, True),
    ('eventDateTimeStart', int, True),
    ('eventDateTimeEnd', int, True),
    ('loopRules', str, True),
    ('timezoneOffset', int, True),
    ('lastChange', str, False)),
    None)

@app.route('/api/calendar/add', methods=['POST'])
def api_calendar_addHandle():
    return SmartDbCaller(calendar_db.calendar_add, 
    (('token', str, False),
    ('belongTo', str, False),
    ('title', str, False),
    ('description', str, False),
    ('eventDateTimeStart', int, False),
    ('eventDateTimeEnd', int, False),
    ('loopRules', str, False),
    ('timezoneOffset', int, False)),
    None)

@app.route('/api/calendar/delete', methods=['POST'])
def api_calendar_deleteHandle():
    return SmartDbCaller(calendar_db.calendar_delete, 
    (('token', str, False),
    ('uuid', str, False),
    ('lastChange', str, False)),
    None)

# ================================ collection

@app.route('/api/collection/getFullOwn', methods=['POST'])
def api_collection_getFullOwnHandle():
    return SmartDbCaller(calendar_db.collection_getFullOwn, 
    (('token', str, False), ),
    None)

@app.route('/api/collection/getListOwn', methods=['POST'])
def api_collection_getListOwnHandle():
    return SmartDbCaller(calendar_db.collection_getListlOwn, 
    (('token', str, False), ),
    None)

@app.route('/api/collection/getDetailOwn', methods=['POST'])
def api_collection_getDetailOwnHandle():
    return SmartDbCaller(calendar_db.collection_getDetailOwn, 
    (('token', str, False), 
    ('uuid', str, False)),
    None)

@app.route('/api/collection/addOwn', methods=['POST'])
def api_collection_addOwnHandle():
    return SmartDbCaller(calendar_db.collection_addOwn, 
    (('token', str, False), 
    ('name', str, False)),
    None)

@app.route('/api/collection/updateOwn', methods=['POST'])
def api_collection_updateOwnHandle():
    return SmartDbCaller(calendar_db.collection_updateOwn, 
    (('token', str, False), 
    ('uuid', str, False),
    ('name', str, False), 
    ('lastChange', str, False)),
    None)

@app.route('/api/collection/deleteOwn', methods=['POST'])
def api_collection_deleteOwnHandle():
    return SmartDbCaller(calendar_db.collection_deleteOwn, 
    (('token', str, False), 
    ('uuid', str, False),
    ('lastChange', str, False)),
    None)


@app.route('/api/collection/getSharing', methods=['POST'])
def api_collection_getSharingHandle():
    return SmartDbCaller(calendar_db.collection_getSharing, 
    (('token', str, False), 
    ('uuid', str, False)),
    None)

@app.route('/api/collection/deleteSharing', methods=['POST'])
def api_collection_deleteSharingHandle():
    return SmartDbCaller(calendar_db.collection_deleteSharing, 
    (('token', str, False), 
    ('uuid', str, False),
    ('target', str, False),
    ('lastChange', str, False)),
    None)

@app.route('/api/collection/addSharing', methods=['POST'])
def api_collection_addSharingHandle():
    return SmartDbCaller(calendar_db.collection_addSharing, 
    (('token', str, False), 
    ('uuid', str, False),
    ('target', str, False),
    ('lastChange', str, False)),
    None)


@app.route('/api/collection/getShared', methods=['POST'])
def api_collection_getSharedHandle():
    return SmartDbCaller(calendar_db.collection_getShared, 
    (('token', str, False), ),
    None)

# ================================ todo

@app.route('/api/todo/getFull', methods=['POST'])
def api_todo_getFullHandle():
    return SmartDbCaller(calendar_db.todo_getFull, 
    (('token', str, False), ),
    None)

@app.route('/api/todo/getList', methods=['POST'])
def api_todo_getListHandle():
    return SmartDbCaller(calendar_db.todo_getList, 
    (('token', str, False), ),
    None)

@app.route('/api/todo/getDetail', methods=['POST'])
def api_todo_getDetailHandle():
    return SmartDbCaller(calendar_db.todo_getDetail, 
    (('token', str, False), 
    ('uuid', str, False)),
    None)

@app.route('/api/todo/add', methods=['POST'])
def api_todo_addHandle():
    return SmartDbCaller(calendar_db.todo_add, 
    (('token', str, False), ),
    None)

@app.route('/api/todo/update', methods=['POST'])
def api_todo_updateHandle():
    return SmartDbCaller(calendar_db.todo_update, 
    (('token', str, False), 
    ('uuid', str, False), 
    ('data', str, False), 
    ('lastChange', str, False)),
    None)

@app.route('/api/todo/delete', methods=['POST'])
def api_todo_deleteHandle():
    return SmartDbCaller(calendar_db.todo_delete, 
    (('token', str, False), 
    ('uuid', str, False), 
    ('lastChange', str, False)),
    None)

# ================================ admin

@app.route('/api/admin/get', methods=['POST'])
def api_admin_getHandle():
    return SmartDbCaller(calendar_db.admin_get, 
    (('token', str, False), ),
    None)

@app.route('/api/admin/add', methods=['POST'])
def api_admin_addHandle():
    return SmartDbCaller(calendar_db.admin_add, 
    (('token', str, False), 
    ('username', str, False)),
    None)

@app.route('/api/admin/update', methods=['POST'])
def api_admin_updateHandle():
    return SmartDbCaller(calendar_db.admin_update, 
    (('token', str, False), 
    ('username', str, False),
    ('password', str, True),
    ('isAdmin', utils.Str2Bool, True)),
    None)

@app.route('/api/admin/delete', methods=['POST'])
def api_admin_deleteHandle():
    return SmartDbCaller(calendar_db.admin_delete, 
    (('token', str, False), 
    ('username', str, False)),
    None)

# ================================ profile

@app.route('/api/profile/isAdmin', methods=['POST'])
def api_profile_isAdminHandle():
    return SmartDbCaller(calendar_db.profile_isAdmin, 
    (('token', str, False), ),
    None)

@app.route('/api/profile/changePassword', methods=['POST'])
def api_profile_changePasswordHandle():
    return SmartDbCaller(calendar_db.profile_changePassword, 
    (('token', str, False),
    ('password', str, False)),
    None)

@app.route('/api/profile/getToken', methods=['POST'])
def api_profile_getTokenHandle():
    return SmartDbCaller(calendar_db.profile_getToken, 
    (('token', str, False), ),
    None)

@app.route('/api/profile/deleteToken', methods=['POST'])
def api_profile_deleteTokenHandle():
    return SmartDbCaller(calendar_db.profile_deleteToken, 
    (('token', str, False),
    ('deleteToken', str, False)),
    None)

# =============================================main run

'''
def UpdateStaticResources():
    global render_static_resources
    if render_static_resources is not None:
        return

    render_static_resources = {
        'url_js_localStorageAssist': url_for('static', filename='js/localStorageAssist.js'),
        'url_js_i18n': url_for('static', filename='js/i18n.js'),
        'url_js_api': url_for('static', filename='js/api.js'),
        'url_js_headerNav': url_for('static', filename='js/headerNav.js'),

        'url_tmpl_headerNac': url_for('static', filename='tmpl/headerNav.tmpl'),

        'url_js_pageHome': url_for('static', filename='js/page/home.js')
    }
'''

def SmartDbCaller(dbMethod, paramTuple, extraDict):
    result = (False, 'Invalid parameter', None)
    optCount = 0
    paramList = []
    optParamDict = {}
    # for each item, 
    # item[0] is field name. 
    # item[1] is type. 
    # item[2] is whether it is optional field
    realForm = request.form.to_dict()
    if extraDict is not None:
        realForm.update(extraDict)
    for item in paramTuple:
        cache = item[1](realForm.get(item[0], None))
        if item[2]:
            # optional param
            if cache is not None:
                optParamDict[item[0]] = cache
            optCount += 1
        else:
            if cache is None:
                break
            paramList.append(cache)
    else:
        # at least one opt param
        if optCount == 0 or len(optParamDict) != 0:
            result = dbMethod(*paramList, **optParamDict)
    
    return ConstructResponseBody(result)

def ConstructResponseBody(returnedTuple):
    return {
        'success': returnedTuple[0],
        'error': returnedTuple[1],
        'data': returnedTuple[2]
    }

def run():
    calendar_db.open()
    app.run(port=config.CustomConfig['web']['port'])
    calendar_db.close()