From 49c35b1b5886b9b1f64593c8d0fdc751ee008957 Mon Sep 17 00:00:00 2001
From: John Xina <bingchilling@riseup.net>
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'<a href="/home/main?id={reply_id}">@{ cache.get(reply_id) }</a> '
+ htmlfmt += f'<a href="/home/main?id={reply_id}">@{ cache.get(str(reply_id)) }</a> '
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)
--
GitLab