79 lines
2.4 KiB
Python
79 lines
2.4 KiB
Python
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(),
|
|
})
|
|
cached = dotdict.DotDict({
|
|
'albums': len(common.P.caches[etiquette.objects.Album]),
|
|
'bookmarks': len(common.P.caches[etiquette.objects.Bookmark]),
|
|
'photos': len(common.P.caches[etiquette.objects.Photo]),
|
|
'tags': len(common.P.caches[etiquette.objects.Tag]),
|
|
'users': len(common.P.caches[etiquette.objects.User]),
|
|
})
|
|
return common.render_template(request, 'admin.html', cached=cached, 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({})
|