etiquette/frontends/etiquette_flask/backend/endpoints/admin_endpoints.py

import flask; from flask import request

from voussoirkit import dotdict
from voussoirkit import flasktools
from voussoirkit import timetools

import etiquette

from .. import common

site = common.site
session_manager = common.session_manager

####################################################################################################

@site.route('/admin')
def get_admin():
    if not request.is_localhost:
        flask.abort(403)

    counts = dotdict.DotDict({
        'albums': common.P.get_album_count(),
        'bookmarks': common.P.get_bookmark_count(),
        'photos': common.P.get_photo_count(),
        'tags': common.P.get_tag_count(),
        'users': common.P.get_user_count(),
    })
    return common.render_template(request, 'admin.html', counts=counts)

@site.route('/admin/dbdownload')
def get_dbdump():
    if not request.is_localhost:
        flask.abort(403)

    with common.P.transaction:
        binary = common.P.database_filepath.read('rb')

    now = timetools.now().strftime('%Y-%m-%d_%H-%M-%S')
    download_as = f'etiquette {now}.db'
    outgoing_headers = {
        'Content-Type': 'application/octet-stream',
        'Content-Disposition': f'attachment; filename*=UTF-8\'\'{download_as}',
    }
    return flask.Response(binary, headers=outgoing_headers)

@site.route('/admin/clear_sessions', methods=['POST'])
def post_clear_sessions():
    if not request.is_localhost:
        return flasktools.json_response({}, status=403)

    session_manager.clear()
    return flasktools.json_response({})

@site.route('/admin/reload_config', methods=['POST'])
def post_reload_config():
    if not request.is_localhost:
        return flasktools.json_response({}, status=403)

    common.P.load_config()
    return flasktools.json_response({})

@site.route('/admin/uncache', methods=['POST'])
def post_uncache():
    if not request.is_localhost:
        return flasktools.json_response({}, status=403)

    with common.P.transaction:
        for cache in common.P.caches.values():
            print(cache)
            cache.clear()

    return flasktools.json_response({})