bakatrouble/e621_bot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

request less tags per query

Browse Source
This commit is contained in:
bakatrouble 2023-11-21 17:33:37 +03:00
parent 9e1471c9d2
commit 061115be9e
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
main.py
View File

@ -38,6 +38,8 @@ dp = Dispatcher()
ChatFilter = F.chat.id.in_(set(map(int, os.environ['USERS'].split(',')))) ChatFilter = F.chat.id.in_(set(map(int, os.environ['USERS'].split(','))))
PAGE_SIZE = 20
def format_tags(tags: Iterable[str]): def format_tags(tags: Iterable[str]):
return ' '.join(f'#{re.sub(r"[^0-9a-zA-Z_]", "_", tag)}' for tag in tags) or 'None' return ' '.join(f'#{re.sub(r"[^0-9a-zA-Z_]", "_", tag)}' for tag in tags) or 'None'
@ -133,8 +135,8 @@ async def check_updates():
logging.warning('Lock acquired...') logging.warning('Lock acquired...')
tag_list = [t.decode() for t in await redis.smembers('e621:subs')] tag_list = [t.decode() for t in await redis.smembers('e621:subs')]
random.shuffle(tag_list) random.shuffle(tag_list)
for tl_idx in range(0, len(tag_list), 40): for tl_idx in range(0, len(tag_list), PAGE_SIZE):
tags = ' '.join(f'~{tag}' for tag in tag_list[tl_idx: tl_idx + 40]) tags = ' '.join(f'~{tag}' for tag in tag_list[tl_idx: tl_idx + PAGE_SIZE])
logging.warning(tags) logging.warning(tags)
posts = await e621.get_posts(tags) posts = await e621.get_posts(tags)
if not posts: if not posts: