diff --git a/commands.py b/commands.py index e6e9029..61b2485 100644 --- a/commands.py +++ b/commands.py @@ -1,4 +1,9 @@ -from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup, LinkPreviewOptions +from telegram import ( + Update, + InlineKeyboardButton, + InlineKeyboardMarkup, + LinkPreviewOptions, +) from telegram.constants import ParseMode from telegram.ext import ContextTypes @@ -12,7 +17,7 @@ async def start_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> N await update.message.reply_text( f"hello, i'm {html_parser.bold(context.bot.first_name)}, an inline image grabber.\n\n" f"to get help, use /help", - parse_mode=ParseMode.HTML + parse_mode=ParseMode.HTML, ) @@ -24,18 +29,14 @@ async def help_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> No f"2. click on the box that has popped up\n" f"3. ???\n" f"4. done!", - parse_mode=ParseMode.HTML + parse_mode=ParseMode.HTML, ) async def about_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None: reply_markup = None if general.sourceurl is not None: - keyboard = [ - [ - InlineKeyboardButton(f"source code", url=general.sourceurl) - ] - ] + keyboard = [[InlineKeyboardButton(f"source code", url=general.sourceurl)]] reply_markup = InlineKeyboardMarkup(keyboard) await update.message.reply_text( @@ -45,30 +46,33 @@ async def about_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> N f"{html_parser.bold("currently configured instance:")}\n" f"{html_parser.italic(app.name)} ({app.hostname})", parse_mode=ParseMode.HTML, - reply_markup=reply_markup + reply_markup=reply_markup, ) async def info_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None: try: post_id = context.args[0] - message = await context.bot.send_message(update.effective_chat.id, - f"{html_parser.bold("Information")}\n" - f"Fetching...", - parse_mode=ParseMode.HTML) + message = await context.bot.send_message( + update.effective_chat.id, + f"{html_parser.bold("Information")}\n" f"Fetching...", + parse_mode=ParseMode.HTML, + ) post_data = get_json(f"posts/{post_id}") if post_data is None: await update.message.reply_text( f"{html_parser.bold("Error")}: That record was not found.", - parse_mode=ParseMode.HTML) + parse_mode=ParseMode.HTML, + ) return uploader_data = get_json(f"users/{post_data['uploader_id']}") # well, we could check the uploader, but why would we do that? keyboard = [ [ - InlineKeyboardButton(f"Open in {app.name}", - url=f"http://{app.hostname}/posts/{post_id}") + InlineKeyboardButton( + f"Open in {app.name}", url=f"http://{app.hostname}/posts/{post_id}" + ) ] ] reply_markup = InlineKeyboardMarkup(keyboard) @@ -76,41 +80,60 @@ async def info_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> No # noinspection PyListCreation m = [] m.append(f"ID: {html_parser.code(post_data['id'])}") - m.append(f"Uploader: {html_parser.hyperlink(uploader_data['name'], + m.append( + f"Uploader: {html_parser.hyperlink(uploader_data['name'], f"http://{app.hostname}/users/{post_data['uploader_id']}")} " - f"{html_parser.hyperlink("»", f"http://{app.hostname}/posts?tags=user:{uploader_data['name']}")}") - created_at = datetime.fromisoformat(post_data['created_at']) - m.append(f"Date: {html_parser.hyperlink( + f"{html_parser.hyperlink("»", f"http://{app.hostname}/posts?tags=user:{uploader_data['name']}")}" + ) + created_at = datetime.fromisoformat(post_data["created_at"]) + m.append( + f"Date: {html_parser.hyperlink( f"{created_at.strftime("%Y-%m-%d %X (%z)")}", - f"http://{app.hostname}/posts?tags=date:{created_at.strftime("%Y-%m-%d")}")}") - if post_data['approver_id'] is not None: + f"http://{app.hostname}/posts?tags=date:{created_at.strftime("%Y-%m-%d")}")}" + ) + if post_data["approver_id"] is not None: approver_data = get_json(f"users/{post_data['approver_id']}") - m.append(f"Approver: {html_parser.hyperlink(approver_data['name'], + m.append( + f"Approver: {html_parser.hyperlink(approver_data['name'], f"http://{app.hostname}/users/{post_data['approver_id']}")} " - f"{html_parser.hyperlink("»", f"http://{app.hostname}/posts?tags=approver:{approver_data['name']}")}") - m.append(f"Size: {html_parser.hyperlink( + f"{html_parser.hyperlink("»", f"http://{app.hostname}/posts?tags=approver:{approver_data['name']}")}" + ) + m.append( + f"Size: {html_parser.hyperlink( 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'])} " - f"({post_data['media_asset']['image_width']}x{post_data['media_asset']['image_height']}) " - f"{html_parser.hyperlink("»", f"http://{app.hostname}/media_assets/{post_data['media_asset']['id']}")}") - m.append(f"Source: {post_data['source'] if post_data['source'] != "" else "🚫"}") + f"({post_data['media_asset']['image_width']}x{post_data['media_asset']['image_height']}) " + f"{html_parser.hyperlink("»", f"http://{app.hostname}/media_assets/{post_data['media_asset']['id']}")}" + ) + m.append( + f"Source: {post_data['source'] if post_data['source'] is not "" else "🚫"}" + ) m.append(f"Rating: {format_rating(post_data['rating'])}") - m.append(f"Score: {html_parser.hyperlink(post_data['score'], + m.append( + f"Score: {html_parser.hyperlink(post_data['score'], f"http://{app.hostname}/post_votes?search[post_id]={post_data['id']}&variant=compact")} " - f"(+{post_data['up_score']} / -{post_data['down_score']})") - m.append(f"Favorites: {html_parser.hyperlink(post_data['fav_count'], - f"http://{app.hostname}/posts/{post_data['id']}/favorites")}") + f"(+{post_data['up_score']} / -{post_data['down_score']})" + ) + m.append( + f"Favorites: {html_parser.hyperlink(post_data['fav_count'], + f"http://{app.hostname}/posts/{post_data['id']}/favorites")}" + ) m.append(f"Status: {format_status(post_data)}") link_preview_options = LinkPreviewOptions(True) - if not post_data['is_banned']: - link_preview_options = LinkPreviewOptions(url=post_data['large_file_url']) + if not post_data["is_banned"]: + link_preview_options = LinkPreviewOptions(url=post_data["large_file_url"]) await context.bot.edit_message_text( f"{html_parser.bold("Information")}\n" + "\n".join(m), - update.effective_chat.id, message.message_id, - parse_mode=ParseMode.HTML, reply_markup=reply_markup, link_preview_options=link_preview_options) + update.effective_chat.id, + message.message_id, + parse_mode=ParseMode.HTML, + reply_markup=reply_markup, + link_preview_options=link_preview_options, + ) except (IndexError, ValueError): await update.message.reply_text( f"{html_parser.bold("Usage")}: {html_parser.code(f"/info <post ID>")}", - parse_mode=ParseMode.HTML) + parse_mode=ParseMode.HTML, + ) diff --git a/config.py b/config.py index c6c1a8b..021c5e9 100644 --- a/config.py +++ b/config.py @@ -12,9 +12,11 @@ class _General: sourceurl: Optional[str] def __init__(self): - self.token = _c.get('General', 'Token') - self.sourceurl = _c.get('General', 'SourceUrl') - if not self.sourceurl.startswith("http://") and not self.sourceurl.startswith("https://"): + self.token = _c.get("General", "Token") + self.sourceurl = _c.get("General", "SourceUrl") + if not self.sourceurl.startswith("http://") and not self.sourceurl.startswith( + "https://" + ): self.sourceurl = None @@ -24,8 +26,8 @@ class _App: hostname: str def __init__(self): - self.name = _c.get('App', 'Name') - self.host = _c.get('App', 'Host') + self.name = _c.get("App", "Name") + self.host = _c.get("App", "Host") # noinspection PyBroadException try: response = requests.get(f"http://{self.host}/status.json") @@ -34,7 +36,7 @@ class _App: except: print("Unable validate App.Host in config.ini.") exit(1) - self.hostname = _c.get('App', 'HostName') + self.hostname = _c.get("App", "HostName") if not self.hostname: self.hostname = self.host diff --git a/extensions.py b/extensions.py index 8ee8db9..6081ab0 100644 --- a/extensions.py +++ b/extensions.py @@ -6,14 +6,14 @@ from config import app def humanize_tags_from_json(value: str, default: str) -> str: - if value != "": - return ", ".join([re.sub('_\\(.*', '', t) for t in value.split()]) + if value is not "": + return ", ".join([re.sub("_\\(.*", "", t) for t in value.split()]) return default def humanize_filesize(value: int) -> str: - for unit in ['B', 'KiB', 'MiB', 'GiB']: - if value < 1024 or unit == 'GiB': + for unit in ["B", "KiB", "MiB", "GiB"]: + if value < 1024 or unit is "GiB": break value /= 1024 return f"{value:.2f} {unit}" @@ -38,10 +38,10 @@ def format_rating(value: str) -> str | None: def format_status(data) -> str: is_active = True - is_pending = data['is_pending'] - is_flagged = data['is_flagged'] - is_deleted = data['is_deleted'] - is_banned = data['is_banned'] + is_pending = data["is_pending"] + is_flagged = data["is_flagged"] + is_deleted = data["is_deleted"] + is_banned = data["is_banned"] if is_pending | is_flagged | is_deleted: is_active = False diff --git a/html_parser.py b/html_parser.py index 65b62c6..841c4b5 100644 --- a/html_parser.py +++ b/html_parser.py @@ -1,14 +1,18 @@ def bold(s: str) -> str: return f"{s}" + def italic(s: str) -> str: return f"{s}" + def strikethrough(s: str) -> str: return f"{s}" + def hyperlink(s: str, href: str) -> str: return f"{s}" + def code(s: str) -> str: - return f"{s}" \ No newline at end of file + return f"{s}" diff --git a/main.py b/main.py index 2cf568d..515ca85 100644 --- a/main.py +++ b/main.py @@ -35,7 +35,7 @@ def get_token() -> str: if os.getenv("BOT_TOKEN") is not None: return os.getenv("BOT_TOKEN") - if general.token != "": + if general.token is not "": return general.token print(