shibao/szuru-eink-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add 100 max retries

Browse Source
This commit is contained in:
shibao
2026-03-04 05:54:29 +00:00
parent b5d6fc2152
commit 2748a5a75b
2 changed files with 21 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
szuru-eink.py
View File

@@ -26,7 +26,10 @@ booru = pyszuru.API(szuru_url, username=szuru_user, token=szuru_token)
highest_post = next(booru.search_post("sort:id type:image", page_size=1))
post = None
while post is None:
retries = 0
MAX_RETRIES = 100
while post is None and retries < MAX_RETRIES:
random_id = random.randint(0, highest_post.id_)
try:
temp_post = booru.getPost(random_id)
@@ -42,8 +45,12 @@ while post is None:
print(f"[DEBUG] Found image post with ID: {post.id_}, MIME: {post.mime}")
else:
print(
f"[DEBUG] Skipping post ID: {random_id} (MIME: {temp_post.mime if temp_post else 'None'}, Content: {bool(temp_post.content) if temp_post else 'None'}) - not a valid image."
f"[DEBUG] Skipping post ID: {random_id} (MIME: {temp_post.mime if temp_post else 'None'}, Content: {bool(temp_post.content) if temp_post else 'None'}) - not a valid image. (Retry {retries + 1}/{MAX_RETRIES})"
)
retries += 1
if post is None:
raise RuntimeError(f"Failed to find a valid post after {MAX_RETRIES} retries.")
# download image
image_downloaded = False
@@ -72,7 +79,7 @@ while not image_downloaded:
os.remove("image.jpg")
# Find a new post
post = None
while post is None:
while post is None and retries < MAX_RETRIES:
random_id = random.randint(0, highest_post.id_)
try:
temp_post = booru.getPost(random_id)
@@ -90,8 +97,14 @@ while not image_downloaded:
)
else:
print(
f"[DEBUG] Skipping post ID: {random_id} (MIME: {temp_post.mime if temp_post else 'None'}, Content: {bool(temp_post.content) if temp_post else 'None'}) - not a valid image."
f"[DEBUG] Skipping post ID: {random_id} (MIME: {temp_post.mime if temp_post else 'None'}, Content: {bool(temp_post.content) if temp_post else 'None'}) - not a valid image. (Retry {retries + 1}/{MAX_RETRIES})"
)
retries += 1
if post is None:
raise RuntimeError(
f"Failed to find a valid post after {MAX_RETRIES} retries."
)
# Process and upload
img = Image.open("image.jpg")