diff --git a/extensions.py b/extensions.py index 26682fe..c6ac510 100644 --- a/extensions.py +++ b/extensions.py @@ -1,7 +1,7 @@ import re -def humanize_tags_from_json(value, default): +def humanize_tags_from_json(value: str, default: str): if value != "": output = str() tags = value.split() @@ -14,7 +14,7 @@ def humanize_tags_from_json(value, default): return default -def format_rating(value): +def format_rating(value: str): match value: case "g": # Negative Squared Latin Capital Letter G diff --git a/inline_query.py b/inline_query.py index 9e6cffc..f2f0cdd 100644 --- a/inline_query.py +++ b/inline_query.py @@ -1,3 +1,4 @@ +from configparser import ConfigParser from uuid import uuid4 import requests @@ -31,7 +32,8 @@ async def inline_query(update: Update, context: ContextTypes.DEFAULT_TYPE) -> No await answer_query(update, query, config, data) -async def answer_query(update, query, config, data): +async def answer_query(update: Update, query: str, + config: ConfigParser, 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") @@ -63,7 +65,7 @@ async def answer_query(update, query, config, data): await update.inline_query.answer(results) -async def invalid_query(update, query): +async def invalid_query(update: Update, query: str): results = [ InlineQueryResultArticle( id=str(uuid4()),