voussoir/etiquette
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Use dotdict to remove all these ugly brackets.

Browse source
This commit is contained in:
voussoir 2021-01-05 13:09:31 -08:00
parent af73bc580f
commit 06f26cd7b7
1 changed files with 14 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
frontends/etiquette_flask/backend/decorators.py
View file

@ -2,6 +2,7 @@ import flask; from flask import request
import functools import functools
import time import time
from voussoirkit import dotdict
from voussoirkit import passwordy from voussoirkit import passwordy
import etiquette import etiquette
@ -31,31 +32,31 @@ def cached_endpoint(max_age):
An example use case would be large-sized data dumps that don't need to be An example use case would be large-sized data dumps that don't need to be
precisely up to date every time. precisely up to date every time.
''' '''
state = { state = dotdict.DotDict({
'max_age': max_age, 'max_age': max_age,
'stored_value': None, 'stored_value': None,
'stored_etag': None, 'stored_etag': None,
'headers': {'ETag': None, 'Cache-Control': f'max-age={max_age}'}, 'headers': {'ETag': None, 'Cache-Control': f'max-age={max_age}'},
'last_run': 0, 'last_run': 0,
} })
def wrapper(function): def wrapper(function):
def get_value(*args, **kwargs): def get_value(*args, **kwargs):
if state['max_age'] and (time.time() - state['last_run']) > state['max_age']: if state.max_age and (time.time() - state.last_run) > state.max_age:
return state['stored_value'] return state.stored_value
value = function(*args, **kwargs) value = function(*args, **kwargs)
if isinstance(value, flask.Response): if isinstance(value, flask.Response):
if value.headers.get('Content-Type'): if value.headers.get('Content-Type'):
state['headers']['Content-Type'] = value.headers.get('Content-Type') state.headers['Content-Type'] = value.headers.get('Content-Type')
value = value.response value = value.response
if value != state['stored_value']: if value != state.stored_value:
state['stored_value'] = value state.stored_value = value
state['stored_etag'] = passwordy.random_hex(20) state.stored_etag = passwordy.random_hex(20)
state['headers']['ETag'] = state['stored_etag'] state.headers['ETag'] = state.stored_etag
state['last_run'] = time.time() state.last_run = time.time()
return value return value
@functools.wraps(function) @functools.wraps(function)
@ -63,10 +64,10 @@ def cached_endpoint(max_age):
value = get_value(*args, **kwargs) value = get_value(*args, **kwargs)
client_etag = request.headers.get('If-None-Match', None) client_etag = request.headers.get('If-None-Match', None)
if client_etag == state['stored_etag']: if client_etag == state.stored_etag:
response = flask.Response(status=304, headers=state['headers']) response = flask.Response(status=304, headers=state.headers)
else: else:
response = flask.Response(value, status=200, headers=state['headers']) response = flask.Response(value, status=200, headers=state.headers)
return response return response
return wrapped return wrapped