python-modules/wiki.py at main · fiksofficial/python-modules · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
#         ______     ___  ___          _       _
#    ____ | ___ \    |  \/  |         | |     | |
#   / __ \| |_/ /   _| .  . | ___   __| |_   _| | ___
#  / / _` |  __/ | | | |\/| |/ _ \ / _` | | | | |/ _ \
# | | (_| | |  | |_| | |  | | (_) | (_| | |_| | |  __/
#  \ \__,_\_|   \__, \_|  |_/\___/ \__,_|\__,_|_|\___|
#   \____/       __/ |
#               |___/

# На модуль распространяется лицензия "GNU General Public License v3.0"
# https://github.com/all-licenses/GNU-General-Public-License-v3.0

# meta developer: @pymodule
# requires: aiohttp
# meta fhsdesc: tool, tools, wiki, wikipedia, info, wikiinfo

from .. import loader, utils
from ..inline.types import InlineQuery
import aiohttp


@loader.tds
class WikiSearchMod(loader.Module):
    """Search Wikipedia articles"""

    strings = {
        "name": "WikiSearch",
        "no_query": "❗ Please provide a search term.",
        "searching": "🔎 Searching Wikipedia for: <b>{query}</b>",
        "not_found": "🚫 No results found for: <code>{query}</code>",
        "error": "🚫 Error: <code>{error}</code>",
        "article": "<b>{title}</b>\n\n{summary}\n\n🌐 <a href='{url}'>Read more…</a>",
        "inline_title": "📚 {title}",
        "inline_description": "🔍 {summary}",
    }

    strings_ru = {
        "name": "WikiSearch",
        "no_query": "❗ Укажи термин для поиска.",
        "searching": "🔎 Поиск в Википедии по запросу: <b>{query}</b>",
        "not_found": "🚫 Ничего не найдено по запросу: <code>{query}</code>",
        "error": "🚫 Ошибка: <code>{error}</code>",
        "article": "<b>{title}</b>\n\n{summary}\n\n🌐 <a href='{url}'>Читать далее…</a>",
        "inline_title": "📚 {title}",
        "inline_description": "🔍 {summary}",
    }

    def __init__(self):
        self.config = loader.ModuleConfig(
            loader.ConfigValue(
                "lang",
                "en",
                lambda: "Language for Wikipedia search (e.g. en, ru, fr)",
                validator=loader.validators.RegExp(r"^[a-z]{2}$"),
            )
        )

    @loader.command(doc="[term] - Search Wikipedia for a term", ru_doc="[термин] - Поиск статьи в Википедии по запросу")
    async def wiki(self, message):
        query = utils.get_args_raw(message)
        if not query:
            return await utils.answer(message, self.strings("no_query"))

        await utils.answer(message, self.strings("searching").format(query=query))

        article = await self._get_article_async(query)
        if isinstance(article, str):
            return await utils.answer(message, self.strings("error").format(error=article))
        if article is None:
            return await utils.answer(message, self.strings("not_found").format(query=query))

        await utils.answer(
            message,
            self.strings("article").format(
                title=article["title"],
                summary=article["summary"],
                url=article["url"]
            )
        )

    async def _get_article_async(self, query):
        try:
            lang = self.config["lang"]
            search_url = f"https://{lang}.wikipedia.org/w/api.php"
            params = {
                "action": "query",
                "format": "json",
                "list": "search",
                "srsearch": query,
                "srlimit": 1
            }
            headers = {
            "User-Agent": "Mozilla/5.0 (Android 10; Mobile; rv:145.0) Gecko/145.0 Firefox/145.0"
            }
            async with aiohttp.ClientSession(headers=headers) as session:
                async with session.get(search_url, params=params) as res:
                    data = await res.json()
                    results = data.get("query", {}).get("search", [])
                    if not results:
                        return None

                title = results[0]["title"]
                summary_url = f"https://{lang}.wikipedia.org/api/rest_v1/page/summary/{title.replace(' ', '_')}"
                async with session.get(summary_url) as res:
                    data = await res.json()

                return {
                    "title": data.get("title", title),
                    "summary": data.get("extract", "No summary available"),
                    "url": data.get("content_urls", {}).get("desktop", {}).get("page", f"https://{lang}.wikipedia.org/wiki/{title.replace(' ', '_')}")
                }

        except Exception as e:
            return str(e)

    @loader.inline_everyone
    async def wiki_inline_handler(self, query: InlineQuery):
        """[term] - Inline Wikipedia search"""
        if not query.args:
            return await query.e400()

        article = await self._get_article_async(query.args)
        if isinstance(article, str):
            return await query.e500()
        if article is None:
            return await query.e404()

        return [{
            "title": self.strings("inline_title").format(title=article["title"]),
            "description": article["summary"][:100],
            "message": self.strings("article").format(
                title=article["title"],
                summary=article["summary"],
                url=article["url"]
            )
        }]