refactor: abuse str.join for more readability (#4)
All checks were successful
Build / Upload to production (push) Successful in 1m44s

Reviewed-on: #4
Co-authored-by: mctaylors <cantsendmails@mctaylors.ru>
Co-committed-by: mctaylors <cantsendmails@mctaylors.ru>
This commit is contained in:
Macintxsh 2025-06-08 08:13:57 +03:00 committed by Forgejo
parent fff77f51f8
commit 6ccf2ace4a
Signed by: Forgejo
GPG key ID: F2B3CB736682B201

View file

@ -15,20 +15,28 @@ from extensions import get_json, format_rating, format_status, humanize_filesize
async def start_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None: async def start_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
await update.message.reply_text( await update.message.reply_text(
f"hello, i'm {html_parser.bold(context.bot.first_name)}, an inline image grabber.\n\n" "\n".join(
f"to get help, use /help", [
f"hello, i'm {html_parser.bold(context.bot.first_name)}, an inline image grabber.\n",
"to get help, use /help",
]
),
parse_mode=ParseMode.HTML, parse_mode=ParseMode.HTML,
) )
async def help_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None: async def help_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
await update.message.reply_text( await update.message.reply_text(
f"{html_parser.bold(f"how to use {context.bot.first_name}")}\n\n" "\n".join(
f"1. open your message box and type:\n" [
f"{html_parser.code(f"@{context.bot.username} &lt;post ID&gt;")}\n" html_parser.bold(f"how to use {context.bot.first_name}\n"),
f"2. click on the box that has popped up\n" "1. open your message box and type:",
f"3. ???\n" html_parser.code(f"@{context.bot.username} &lt;post ID&gt;"),
f"4. done!", "2. click on the box that has popped up",
"3. ???",
"4. done!",
]
),
parse_mode=ParseMode.HTML, parse_mode=ParseMode.HTML,
) )
@ -40,11 +48,15 @@ async def about_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> N
reply_markup = InlineKeyboardMarkup(keyboard) reply_markup = InlineKeyboardMarkup(keyboard)
await update.message.reply_text( await update.message.reply_text(
f"{html_parser.bold(f"about {context.bot.first_name}")}\n" "\n".join(
f"{context.bot.first_name} is an inline image grabber written in python that grabs images from Danbooru " [
f"(or other similar services).\n\n" html_parser.bold(f"about {context.bot.first_name}"),
f"{html_parser.bold("currently configured instance:")}\n" f"{context.bot.first_name} is an inline image grabber written in python that grabs images from "
"Danbooru (or other similar services).\n",
html_parser.bold("currently configured instance:"),
f"{html_parser.italic(app.name)} ({app.hostname})", f"{html_parser.italic(app.name)} ({app.hostname})",
]
),
parse_mode=ParseMode.HTML, parse_mode=ParseMode.HTML,
reply_markup=reply_markup, reply_markup=reply_markup,
) )
@ -55,13 +67,13 @@ async def info_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> No
post_id = context.args[0] post_id = context.args[0]
message = await context.bot.send_message( message = await context.bot.send_message(
update.effective_chat.id, update.effective_chat.id,
f"{html_parser.bold("Information")}\n" f"Fetching...", "\n".join([html_parser.bold("Information"), "Fetching..."]),
parse_mode=ParseMode.HTML, parse_mode=ParseMode.HTML,
) )
post_data = get_json(f"posts/{post_id}") post_data = get_json(f"posts/{post_id}")
if post_data is None: if post_data is None:
await context.bot.edit_message_text( await context.bot.edit_message_text(
f"{html_parser.bold("Error")}: That record was not found.", " ".join([html_parser.bold("Error:"), "That record was not found."]),
update.effective_chat.id, update.effective_chat.id,
message.message_id, message.message_id,
parse_mode=ParseMode.HTML, parse_mode=ParseMode.HTML,
@ -81,52 +93,103 @@ async def info_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> No
reply_markup = InlineKeyboardMarkup(keyboard) reply_markup = InlineKeyboardMarkup(keyboard)
m = [ m = [
f"ID: {html_parser.code(post_data['id'])}", " ".join(["ID:", html_parser.code(post_data["id"])]),
f"Uploader: {html_parser.hyperlink(uploader_data['name'], " ".join(
f"{app.protocol}://{app.hostname}/users/{post_data['uploader_id']}")} " [
f"{html_parser.hyperlink("»", f"{app.protocol}://{app.hostname}/posts?tags=user:{uploader_data['name']}")}", "Uploader:",
html_parser.hyperlink(
uploader_data["name"],
f"{app.protocol}://{app.hostname}/users/{post_data['uploader_id']}",
),
html_parser.hyperlink(
"»",
f"{app.protocol}://{app.hostname}/posts?tags=user:{uploader_data['name']}",
),
]
),
] ]
created_at = datetime.fromisoformat(post_data["created_at"]) created_at = datetime.fromisoformat(post_data["created_at"])
m.append( m.append(
f"Date: {html_parser.hyperlink( " ".join(
f"{created_at.strftime("%Y-%m-%d %X (%z)")}", [
f"{app.protocol}://{app.hostname}/posts?tags=date:{created_at.strftime("%Y-%m-%d")}")}" "Date:",
html_parser.hyperlink(
created_at.strftime("%Y-%m-%d %X (%z)"),
f"{app.protocol}://{app.hostname}/posts?tags=date:{created_at.strftime("%Y-%m-%d")}",
),
]
)
) )
if post_data["approver_id"] is not None: if post_data["approver_id"] is not None:
approver_data = get_json(f"users/{post_data['approver_id']}") approver_data = get_json(f"users/{post_data['approver_id']}")
m.append( m.append(
f"Approver: {html_parser.hyperlink(approver_data['name'], " ".join(
f"{app.protocol}://{app.hostname}/users/{post_data['approver_id']}")} " [
f"{html_parser.hyperlink("»", f"{app.protocol}://{app.hostname}/posts?tags=approver:{approver_data['name']}")}" "Approver:",
html_parser.hyperlink(
approver_data["name"],
f"{app.protocol}://{app.hostname}/users/{post_data['approver_id']}",
),
html_parser.hyperlink(
"»",
f"{app.protocol}://{app.hostname}/posts?tags=approver:{approver_data['name']}",
),
]
)
) )
m.append( m.append(
f"Size: {html_parser.hyperlink( " ".join(
[
"Size:",
html_parser.hyperlink(
f"{humanize_filesize(post_data['media_asset']['file_size'])} .{post_data['media_asset']['file_ext']}", f"{humanize_filesize(post_data['media_asset']['file_size'])} .{post_data['media_asset']['file_ext']}",
"" if post_data['is_banned'] else post_data['file_url'])} " "" if post_data["is_banned"] else post_data["file_url"],
f"({post_data['media_asset']['image_width']}x{post_data['media_asset']['image_height']}) " ),
f"{html_parser.hyperlink("»", f"{app.protocol}://{app.hostname}/media_assets/{post_data['media_asset']['id']}")}" f"({post_data['media_asset']['image_width']}x{post_data['media_asset']['image_height']})",
html_parser.hyperlink(
"»",
f"{app.protocol}://{app.hostname}/media_assets/{post_data['media_asset']['id']}",
),
]
)
) )
m.append( m.append(
f"Source: {post_data['source'] if post_data['source'] != "" else "🚫"}" " ".join(
["Source:", post_data["source"] if post_data["source"] != "" else "🚫"]
) )
m.append(f"Rating: {format_rating(post_data['rating'])}") )
m.append(" ".join(["Rating:", format_rating(post_data["rating"])]))
m.append( m.append(
f"Score: {html_parser.hyperlink(post_data['score'], " ".join(
f"{app.protocol}://{app.hostname}/post_votes?search[post_id]={post_data['id']}&variant=compact")} " [
f"(+{post_data['up_score']} / -{post_data['down_score']})" "Score:",
html_parser.hyperlink(
post_data["score"],
f"{app.protocol}://{app.hostname}/post_votes?search[post_id]={post_data['id']}&variant=compact",
),
f"(+{post_data['up_score']} / -{post_data['down_score']})",
]
)
) )
m.append( m.append(
f"Favorites: {html_parser.hyperlink(post_data['fav_count'], " ".join(
f"{app.protocol}://{app.hostname}/posts/{post_data['id']}/favorites")}" [
"Favorites:",
html_parser.hyperlink(
post_data["fav_count"],
f"{app.protocol}://{app.hostname}/posts/{post_data['id']}/favorites",
),
]
) )
m.append(f"Status: {format_status(post_data)}") )
m.append(" ".join(["Status:", format_status(post_data)]))
link_preview_options = LinkPreviewOptions(True) link_preview_options = LinkPreviewOptions(True)
if not post_data["is_banned"]: if not post_data["is_banned"]:
link_preview_options = LinkPreviewOptions(url=post_data["large_file_url"]) link_preview_options = LinkPreviewOptions(url=post_data["large_file_url"])
await context.bot.edit_message_text( await context.bot.edit_message_text(
f"{html_parser.bold("Information")}\n" + "\n".join(m), "\n".join([html_parser.bold("Information")] + m),
update.effective_chat.id, update.effective_chat.id,
message.message_id, message.message_id,
parse_mode=ParseMode.HTML, parse_mode=ParseMode.HTML,
@ -135,6 +198,8 @@ async def info_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> No
) )
except (IndexError, ValueError): except (IndexError, ValueError):
await update.message.reply_text( await update.message.reply_text(
f"{html_parser.bold("Usage")}: {html_parser.code(f"/info &lt;post ID&gt;")}", " ".join(
[html_parser.bold("Usage:"), html_parser.code("/info &lt;post ID&gt;")]
),
parse_mode=ParseMode.HTML, parse_mode=ParseMode.HTML,
) )