Initial commit

Signed-off-by: mctaylors <cantsendmails@mctaylors.ru>

inline_query.py

@@ -0,0 +1,78 @@
+from uuid import uuid4
+
+import requests
+from telegram import Update, InlineKeyboardButton, InlineQueryResultArticle, InputTextMessageContent, \
+    InlineKeyboardMarkup
+from telegram.constants import ParseMode
+from telegram.ext import ContextTypes
+
+from extensions import humanize_tags_from_json
+
+
+# noinspection PyUnusedLocal
+async def inline_query(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+    query = update.inline_query.query
+
+    if not query:
+        return
+
+    if not query.isdigit():
+        return
+
+    from main import config
+    response = requests.get(f"https://{config.get('General', 'Domain')}/posts/{query}.json")
+
+    if response.status_code != 200:
+        await invalid_query(update, query)
+        return
+
+    data = response.json()
+
+    await answer_query(update, query, config, data)
+
+
+async def answer_query(update, query, config, data):
+    characters = humanize_tags_from_json(data['tag_string_character'], "no characters")
+    copyrights = humanize_tags_from_json(data['tag_string_copyright'], "unknown copyright")
+    artists = humanize_tags_from_json(data['tag_string_artist'], "unknown artist")
+    keyboard = [
+        [
+            InlineKeyboardButton(f"Open in {config.get('General', 'Name')}",
+                                 url=f"https://{config.get('General', 'Domain')}/posts/{query}"),
+            InlineKeyboardButton("View original", url=data['file_url'])
+        ]
+    ]
+
+    results = [
+        InlineQueryResultArticle(
+            id=str(uuid4()),
+            title=f"ID: {query}",
+            description=f"{characters} ({copyrights}) drawn by {artists}",
+            thumbnail_url=data['preview_file_url'],
+            input_message_content=InputTextMessageContent(
+                f"ID: <code>{query}</code>\n"
+                f"<a href='{data['large_file_url']}'><b>{characters} ({copyrights})</b> "
+                f"drawn by <b>{artists}</b></a>",
+                parse_mode=ParseMode.HTML
+            ),
+            reply_markup=InlineKeyboardMarkup(keyboard)
+        )
+    ]
+
+    await update.inline_query.answer(results)
+
+
+async def invalid_query(update, query):
+    results = [
+        InlineQueryResultArticle(
+            id=str(uuid4()),
+            title=f"ID: {query}",
+            description="not found.",
+            input_message_content=InputTextMessageContent(
+                f"ID: <code>{query}</code>\n<b>requested post does not exist.</b>",
+                parse_mode=ParseMode.HTML
+            )
+        )
+    ]
+
+    await update.inline_query.answer(results)