From 49c35b1b5886b9b1f64593c8d0fdc751ee008957 Mon Sep 17 00:00:00 2001 From: John Xina Date: Mon, 24 Jul 2023 15:00:19 +0800 Subject: [PATCH] Several fixes and enhancement --- app.py | 10 +++++----- proxify.py | 5 ++--- shared.py | 4 ++++ 3 files changed, 11 insertions(+), 8 deletions(-) diff --git a/app.py b/app.py index 7e8f32b..31b6bf7 100644 --- a/app.py +++ b/app.py @@ -28,9 +28,9 @@ def append_with_leading_clean(orig, content): # Return the corresponding user name for an id. async def cache_name_from_id(c, i): - if not cache.get(i): - r = await c.get_user_info(i, require=aiotieba.enums.ReqUInfo.USER_NAME) - cache.set(i, r) + if not cache.get(str(i)): + r = await c.get_user_info(i, require=aiotieba.enums.ReqUInfo.NICK_NAME) + cache.set(str(i), r) # Normalize unicode characters to ASCII form. def normalize_utf8(s): @@ -100,7 +100,7 @@ async def _jinja2_filter_translate(frags, reply_id=0): htmlfmt = '' if reply_id: - htmlfmt += f'@{ cache.get(reply_id) } ' + htmlfmt += f'@{ cache.get(str(reply_id)) } ' for i in range(len(frags)): frag = frags[i] @@ -158,7 +158,7 @@ async def thread_view(tid): for floor in thread_info: for comment in floor.comments: available_users.append(comment.author_id) - cache.set(comment.author_id, comment.user.user_name) + cache.set(str(comment.author_id), comment.user.show_name) all_users = {} for floor in thread_info: diff --git a/proxify.py b/proxify.py index 0aff2b3..e4f705f 100644 --- a/proxify.py +++ b/proxify.py @@ -108,9 +108,8 @@ class Asgiproxify(): task = asyncio.create_task(reverse_proxy_task()) ev = await receive() - if ev['type'] == 'http.disconnect': - task.cancel() - return + task.cancel() + return async def __call__(self, scope, receive, send): if scope['type'] != 'http': diff --git a/shared.py b/shared.py index 6b5bbac..93c8a08 100644 --- a/shared.py +++ b/shared.py @@ -27,5 +27,9 @@ app.config['TESTING'] = False ###################################################################### +app.config['CACHE_DEFAULT_TIMEOUT'] = 60 + +#app.config['CACHE_TYPE'] = 'RedisCache' app.config['CACHE_TYPE'] = 'SimpleCache' + cache = Cache(app)