12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169 |
- # ©️ Dan Gazizullin, 2021-2023
- # This file is a part of Hikka Userbot
- # 🌐 https://github.com/hikariatama/Hikka
- # You can redistribute it and/or modify it under the terms of the GNU AGPLv3
- # 🔑 https://www.gnu.org/licenses/agpl-3.0.html
- import ast
- import asyncio
- import contextlib
- import copy
- import importlib
- import importlib.machinery
- import importlib.util
- import inspect
- import logging
- import os
- import re
- import sys
- import time
- import typing
- from dataclasses import dataclass, field
- from importlib.abc import SourceLoader
- import requests
- from hikkatl.hints import EntityLike
- from hikkatl.tl.functions.account import UpdateNotifySettingsRequest
- from hikkatl.tl.types import (
- Channel,
- ChannelFull,
- InputPeerNotifySettings,
- Message,
- UserFull,
- )
- from . import version
- from ._reference_finder import replace_all_refs
- from .inline.types import (
- BotInlineCall,
- BotInlineMessage,
- BotMessage,
- InlineCall,
- InlineMessage,
- InlineQuery,
- InlineUnit,
- )
- from .pointers import PointerDict, PointerList
- __all__ = [
- "JSONSerializable",
- "HikkaReplyMarkup",
- "ListLike",
- "Command",
- "StringLoader",
- "Module",
- "get_commands",
- "get_inline_handlers",
- "get_callback_handlers",
- "BotInlineCall",
- "BotMessage",
- "InlineCall",
- "InlineMessage",
- "InlineQuery",
- "InlineUnit",
- "BotInlineMessage",
- "PointerDict",
- "PointerList",
- ]
- logger = logging.getLogger(__name__)
- JSONSerializable = typing.Union[str, int, float, bool, list, dict, None]
- HikkaReplyMarkup = typing.Union[typing.List[typing.List[dict]], typing.List[dict], dict]
- ListLike = typing.Union[list, set, tuple]
- Command = typing.Callable[..., typing.Awaitable[typing.Any]]
- class StringLoader(SourceLoader):
- """Load a python module/file from a string"""
- def __init__(self, data: str, origin: str):
- self.data = data.encode("utf-8") if isinstance(data, str) else data
- self.origin = origin
- def get_source(self, _=None) -> str:
- return self.data.decode("utf-8")
- def get_code(self, fullname: str) -> bytes:
- return (
- compile(source, self.origin, "exec", dont_inherit=True)
- if (source := self.get_data(fullname))
- else None
- )
- def get_filename(self, *args, **kwargs) -> str:
- return self.origin
- def get_data(self, *args, **kwargs) -> bytes:
- return self.data
- class Module:
- strings = {"name": "Unknown"}
- """There is no help for this module"""
- def config_complete(self):
- """Called when module.config is populated"""
- async def client_ready(self):
- """Called after client is ready (after config_loaded)"""
- def internal_init(self):
- """Called after the class is initialized in order to pass the client and db. Do not call it yourself"""
- self.db = self.allmodules.db
- self._db = self.allmodules.db
- self.client = self.allmodules.client
- self._client = self.allmodules.client
- self.lookup = self.allmodules.lookup
- self.get_prefix = self.allmodules.get_prefix
- self.inline = self.allmodules.inline
- self.allclients = self.allmodules.allclients
- self.tg_id = self._client.tg_id
- self._tg_id = self._client.tg_id
- async def on_unload(self):
- """Called after unloading / reloading module"""
- async def on_dlmod(self):
- """
- Called after the module is first time loaded with .dlmod or .loadmod
- Possible use-cases:
- - Send reaction to author's channel message
- - Create asset folder
- - ...
- ⚠️ Note, that any error there will not interrupt module load, and will just
- send a message to logs with verbosity INFO and exception traceback
- """
- async def invoke(
- self,
- command: str,
- args: typing.Optional[str] = None,
- peer: typing.Optional[EntityLike] = None,
- message: typing.Optional[Message] = None,
- edit: bool = False,
- ) -> Message:
- """
- Invoke another command
- :param command: Command to invoke
- :param args: Arguments to pass to command
- :param peer: Peer to send the command to. If not specified, will send to the current chat
- :param edit: Whether to edit the message
- :returns Message:
- """
- if command not in self.allmodules.commands:
- raise ValueError(f"Command {command} not found")
- if not message and not peer:
- raise ValueError("Either peer or message must be specified")
- cmd = f"{self.get_prefix()}{command} {args or ''}".strip()
- message = (
- (await self._client.send_message(peer, cmd))
- if peer
- else (await (message.edit if edit else message.respond)(cmd))
- )
- await self.allmodules.commands[command](message)
- return message
- @property
- def commands(self) -> typing.Dict[str, Command]:
- """List of commands that module supports"""
- return get_commands(self)
- @property
- def hikka_commands(self) -> typing.Dict[str, Command]:
- """List of commands that module supports"""
- return get_commands(self)
- @property
- def inline_handlers(self) -> typing.Dict[str, Command]:
- """List of inline handlers that module supports"""
- return get_inline_handlers(self)
- @property
- def hikka_inline_handlers(self) -> typing.Dict[str, Command]:
- """List of inline handlers that module supports"""
- return get_inline_handlers(self)
- @property
- def callback_handlers(self) -> typing.Dict[str, Command]:
- """List of callback handlers that module supports"""
- return get_callback_handlers(self)
- @property
- def hikka_callback_handlers(self) -> typing.Dict[str, Command]:
- """List of callback handlers that module supports"""
- return get_callback_handlers(self)
- @property
- def watchers(self) -> typing.Dict[str, Command]:
- """List of watchers that module supports"""
- return get_watchers(self)
- @property
- def hikka_watchers(self) -> typing.Dict[str, Command]:
- """List of watchers that module supports"""
- return get_watchers(self)
- @commands.setter
- def commands(self, _):
- pass
- @hikka_commands.setter
- def hikka_commands(self, _):
- pass
- @inline_handlers.setter
- def inline_handlers(self, _):
- pass
- @hikka_inline_handlers.setter
- def hikka_inline_handlers(self, _):
- pass
- @callback_handlers.setter
- def callback_handlers(self, _):
- pass
- @hikka_callback_handlers.setter
- def hikka_callback_handlers(self, _):
- pass
- @watchers.setter
- def watchers(self, _):
- pass
- @hikka_watchers.setter
- def hikka_watchers(self, _):
- pass
- async def animate(
- self,
- message: typing.Union[Message, InlineMessage],
- frames: typing.List[str],
- interval: typing.Union[float, int],
- *,
- inline: bool = False,
- ) -> None:
- """
- Animate message
- :param message: Message to animate
- :param frames: A List of strings which are the frames of animation
- :param interval: Animation delay
- :param inline: Whether to use inline bot for animation
- :returns message:
- Please, note that if you set `inline=True`, first frame will be shown with an empty
- button due to the limitations of Telegram API
- """
- from . import utils
- with contextlib.suppress(AttributeError):
- _hikka_client_id_logging_tag = copy.copy(self.client.tg_id) # noqa: F841
- if interval < 0.1:
- logger.warning(
- "Resetting animation interval to 0.1s, because it may get you in"
- " floodwaits"
- )
- interval = 0.1
- for frame in frames:
- if isinstance(message, Message):
- if inline:
- message = await self.inline.form(
- message=message,
- text=frame,
- reply_markup={"text": "\u0020\u2800", "data": "empty"},
- )
- else:
- message = await utils.answer(message, frame)
- elif isinstance(message, InlineMessage) and inline:
- await message.edit(frame)
- await asyncio.sleep(interval)
- return message
- def get(
- self,
- key: str,
- default: typing.Optional[JSONSerializable] = None,
- ) -> JSONSerializable:
- return self._db.get(self.__class__.__name__, key, default)
- def set(self, key: str, value: JSONSerializable) -> bool:
- self._db.set(self.__class__.__name__, key, value)
- def pointer(
- self,
- key: str,
- default: typing.Optional[JSONSerializable] = None,
- item_type: typing.Optional[typing.Any] = None,
- ) -> typing.Union[JSONSerializable, PointerList, PointerDict]:
- return self._db.pointer(self.__class__.__name__, key, default, item_type)
- async def _approve(
- self,
- call: InlineCall,
- channel: EntityLike,
- event: asyncio.Event,
- ):
- from . import utils
- local_event = asyncio.Event()
- self.__approve += [(channel, local_event)] # skipcq: PTC-W0037
- await local_event.wait()
- event.status = local_event.status
- event.set()
- await call.edit(
- (
- "💫 <b>Joined <a"
- f' href="https://t.me/{channel.username}">{utils.escape_html(channel.title)}</a></b>'
- ),
- gif="https://static.hikari.gay/0d32cbaa959e755ac8eef610f01ba0bd.gif",
- )
- async def _decline(
- self,
- call: InlineCall,
- channel: EntityLike,
- event: asyncio.Event,
- ):
- from . import utils
- self._db.set(
- "hikka.main",
- "declined_joins",
- list(set(self._db.get("hikka.main", "declined_joins", []) + [channel.id])),
- )
- event.status = False
- event.set()
- await call.edit(
- (
- "✖️ <b>Declined joining <a"
- f' href="https://t.me/{channel.username}">{utils.escape_html(channel.title)}</a></b>'
- ),
- gif="https://static.hikari.gay/0d32cbaa959e755ac8eef610f01ba0bd.gif",
- )
- async def request_join(
- self,
- peer: EntityLike,
- reason: str,
- assure_joined: typing.Optional[bool] = False,
- ) -> bool:
- """
- Request to join a channel.
- :param peer: The channel to join.
- :param reason: The reason for joining.
- :param assure_joined: If set, module will not be loaded unless the required channel is joined.
- ⚠️ Works only in `client_ready`!
- ⚠️ If user declines to join channel, he will not be asked to
- join again, so unless he joins it manually, module will not be loaded
- ever.
- :return: Status of the request.
- :rtype: bool
- :notice: This method will block module loading until the request is approved or declined.
- """
- from . import utils
- event = asyncio.Event()
- await self.client(
- UpdateNotifySettingsRequest(
- peer=self.inline.bot_username,
- settings=InputPeerNotifySettings(show_previews=False, silent=False),
- )
- )
- channel = await self.client.get_entity(peer)
- if channel.id in self._db.get("hikka.main", "declined_joins", []):
- if assure_joined:
- raise LoadError(
- f"You need to join @{channel.username} in order to use this module"
- )
- return False
- if not isinstance(channel, Channel):
- raise TypeError("`peer` field must be a channel")
- if getattr(channel, "left", True):
- channel = await self.client.force_get_entity(peer)
- if not getattr(channel, "left", True):
- return True
- await self.inline.bot.send_animation(
- self.tg_id,
- "https://static.hikari.gay/ab3adf144c94a0883bfe489f4eebc520.gif",
- caption=(
- self._client.loader.lookup("translations")
- .strings("requested_join")
- .format(
- self.__class__.__name__,
- channel.username,
- utils.escape_html(channel.title),
- utils.escape_html(reason),
- )
- ),
- reply_markup=self.inline.generate_markup(
- [
- {
- "text": "💫 Approve",
- "callback": self._approve,
- "args": (channel, event),
- },
- {
- "text": "✖️ Decline",
- "callback": self._decline,
- "args": (channel, event),
- },
- ]
- ),
- )
- self.hikka_wait_channel_approve = (
- self.__class__.__name__,
- channel,
- reason,
- )
- await event.wait()
- with contextlib.suppress(AttributeError):
- delattr(self, "hikka_wait_channel_approve")
- if assure_joined and not event.status:
- raise LoadError(
- f"You need to join @{channel.username} in order to use this module"
- )
- return event.status
- async def import_lib(
- self,
- url: str,
- *,
- suspend_on_error: typing.Optional[bool] = False,
- _did_requirements: bool = False,
- ) -> "Library":
- """
- Import library from url and register it in :obj:`Modules`
- :param url: Url to import
- :param suspend_on_error: Will raise :obj:`loader.SelfSuspend` if library can't be loaded
- :return: :obj:`Library`
- :raise: SelfUnload if :attr:`suspend_on_error` is True and error occurred
- :raise: HTTPError if library is not found
- :raise: ImportError if library doesn't have any class which is a subclass of :obj:`loader.Library`
- :raise: ImportError if library name doesn't end with `Lib`
- :raise: RuntimeError if library throws in :method:`init`
- :raise: RuntimeError if library classname exists in :obj:`Modules`.libraries
- """
- from . import utils # Avoiding circular import
- from .loader import USER_INSTALL, VALID_PIP_PACKAGES
- from .translations import Strings
- def _raise(e: Exception):
- if suspend_on_error:
- raise SelfSuspend("Required library is not available or is corrupted.")
- raise e
- if not utils.check_url(url):
- _raise(ValueError("Invalid url for library"))
- code = await utils.run_sync(requests.get, url)
- code.raise_for_status()
- code = code.text
- if re.search(r"# ?scope: ?hikka_min", code):
- ver = tuple(
- map(
- int,
- re.search(r"# ?scope: ?hikka_min ((\d+\.){2}\d+)", code)[1].split(
- "."
- ),
- )
- )
- if version.__version__ < ver:
- _raise(
- RuntimeError(
- f"Library requires Hikka version {'{}.{}.{}'.format(*ver)}+"
- )
- )
- module = f"hikka.libraries.{url.replace('%', '%%').replace('.', '%d')}"
- origin = f"<library {url}>"
- spec = importlib.machinery.ModuleSpec(
- module,
- StringLoader(code, origin),
- origin=origin,
- )
- try:
- instance = importlib.util.module_from_spec(spec)
- sys.modules[module] = instance
- spec.loader.exec_module(instance)
- except ImportError as e:
- logger.info(
- "Library loading failed, attemping dependency installation (%s)",
- e.name,
- )
- # Let's try to reinstall dependencies
- try:
- requirements = list(
- filter(
- lambda x: not x.startswith(("-", "_", ".")),
- map(
- str.strip,
- VALID_PIP_PACKAGES.search(code)[1].split(),
- ),
- )
- )
- except TypeError:
- logger.warning(
- "No valid pip packages specified in code, attemping"
- " installation from error"
- )
- requirements = [e.name]
- logger.debug("Installing requirements: %s", requirements)
- if not requirements or _did_requirements:
- _raise(e)
- pip = await asyncio.create_subprocess_exec(
- sys.executable,
- "-m",
- "pip",
- "install",
- "--upgrade",
- "-q",
- "--disable-pip-version-check",
- "--no-warn-script-location",
- *["--user"] if USER_INSTALL else [],
- *requirements,
- )
- rc = await pip.wait()
- if rc != 0:
- _raise(e)
- importlib.invalidate_caches()
- kwargs = utils.get_kwargs()
- kwargs["_did_requirements"] = True
- return await self._mod_import_lib(**kwargs) # Try again
- lib_obj = next(
- (
- value()
- for value in vars(instance).values()
- if inspect.isclass(value) and issubclass(value, Library)
- ),
- None,
- )
- if not lib_obj:
- _raise(ImportError("Invalid library. No class found"))
- if not lib_obj.__class__.__name__.endswith("Lib"):
- _raise(
- ImportError(
- "Invalid library. Classname {} does not end with 'Lib'".format(
- lib_obj.__class__.__name__
- )
- )
- )
- if (
- all(
- line.replace(" ", "") != "#scope:no_stats" for line in code.splitlines()
- )
- and self._db.get("hikka.main", "stats", True)
- and url is not None
- and utils.check_url(url)
- ):
- with contextlib.suppress(Exception):
- await self.lookup("loader")._send_stats(url)
- lib_obj.source_url = url.strip("/")
- lib_obj.allmodules = self.allmodules
- lib_obj.internal_init()
- for old_lib in self.allmodules.libraries:
- if old_lib.name == lib_obj.name and (
- not isinstance(getattr(old_lib, "version", None), tuple)
- and not isinstance(getattr(lib_obj, "version", None), tuple)
- or old_lib.version >= lib_obj.version
- ):
- logger.debug("Using existing instance of library %s", old_lib.name)
- return old_lib
- if hasattr(lib_obj, "init"):
- if not callable(lib_obj.init):
- _raise(ValueError("Library init() must be callable"))
- try:
- await lib_obj.init()
- except Exception:
- _raise(RuntimeError("Library init() failed"))
- if hasattr(lib_obj, "config"):
- if not isinstance(lib_obj.config, LibraryConfig):
- _raise(
- RuntimeError("Library config must be a `LibraryConfig` instance")
- )
- libcfg = lib_obj.db.get(
- lib_obj.__class__.__name__,
- "__config__",
- {},
- )
- for conf in lib_obj.config:
- with contextlib.suppress(Exception):
- lib_obj.config.set_no_raise(
- conf,
- (
- libcfg[conf]
- if conf in libcfg
- else os.environ.get(f"{lib_obj.__class__.__name__}.{conf}")
- or lib_obj.config.getdef(conf)
- ),
- )
- if hasattr(lib_obj, "strings"):
- lib_obj.strings = Strings(lib_obj, self.translator)
- lib_obj.translator = self.translator
- for old_lib in self.allmodules.libraries:
- if old_lib.name == lib_obj.name:
- if hasattr(old_lib, "on_lib_update") and callable(
- old_lib.on_lib_update
- ):
- await old_lib.on_lib_update(lib_obj)
- replace_all_refs(old_lib, lib_obj)
- logger.debug(
- "Replacing existing instance of library %s with updated object",
- lib_obj.name,
- )
- return lib_obj
- self.allmodules.libraries += [lib_obj]
- return lib_obj
- class DragonModule:
- """Module is running in compatibility mode with Dragon, so it might be unstable"""
- # fmt: off
- strings_ru = {"_cls_doc": "Модуль запущен в режиме совместимости с Dragon, поэтому он может быть нестабильным"}
- strings_de = {"_cls_doc": "Das Modul wird im Dragon-Kompatibilitäts modus ausgeführt, daher kann es instabil sein"}
- strings_tr = {"_cls_doc": "Modül Dragon uyumluluğu modunda çalıştığı için istikrarsız olabilir"}
- strings_uz = {"_cls_doc": "Modul Dragon muvofiqligi rejimida ishlamoqda, shuning uchun u beqaror bo'lishi mumkin"}
- strings_es = {"_cls_doc": "El módulo se ejecuta en modo de compatibilidad con Dragon, por lo que puede ser inestable"}
- strings_kk = {"_cls_doc": "Модуль Dragon қамтамасыз ету режимінде іске қосылған, сондықтан белсенді емес болуы мүмкін"}
- strings_tt = {"_clc_doc": "Модуль Dragon белән ярашучанлык режимда эшли башлады, шуңа күрә ул тотрыксыз була ала"}
- # fmt: on
- def __init__(self):
- self.name = "Unknown"
- self.url = None
- self.commands = {}
- self.watchers = {}
- self.hikka_watchers = {}
- self.inline_handlers = {}
- self.hikka_inline_handlers = {}
- self.callback_handlers = {}
- self.hikka_callback_handlers = {}
- @property
- def hikka_commands(
- self,
- ) -> typing.Dict[str, Command]:
- return self.commands
- @property
- def __origin__(self) -> str:
- return f"<dragon {self.name}>"
- def config_complete(self):
- pass
- async def client_ready(self):
- pass
- async def on_unload(self):
- pass
- async def on_dlmod(self):
- pass
- class Library:
- """All external libraries must have a class-inheritant from this class"""
- def internal_init(self):
- self.name = self.__class__.__name__
- self.db = self.allmodules.db
- self._db = self.allmodules.db
- self.client = self.allmodules.client
- self._client = self.allmodules.client
- self.tg_id = self._client.tg_id
- self._tg_id = self._client.tg_id
- self.lookup = self.allmodules.lookup
- self.get_prefix = self.allmodules.get_prefix
- self.inline = self.allmodules.inline
- self.allclients = self.allmodules.allclients
- def _lib_get(
- self,
- key: str,
- default: typing.Optional[JSONSerializable] = None,
- ) -> JSONSerializable:
- return self._db.get(self.__class__.__name__, key, default)
- def _lib_set(self, key: str, value: JSONSerializable) -> bool:
- self._db.set(self.__class__.__name__, key, value)
- def _lib_pointer(
- self,
- key: str,
- default: typing.Optional[JSONSerializable] = None,
- ) -> typing.Union[JSONSerializable, PointerDict, PointerList]:
- return self._db.pointer(self.__class__.__name__, key, default)
- class LoadError(Exception):
- """Tells user, why your module can't be loaded, if raised in `client_ready`"""
- def __init__(self, error_message: str): # skipcq: PYL-W0231
- self._error = error_message
- def __str__(self) -> str:
- return self._error
- class CoreOverwriteError(LoadError):
- """Is being raised when core module or command is overwritten"""
- def __init__(
- self,
- module: typing.Optional[str] = None,
- command: typing.Optional[str] = None,
- ):
- self.type = "module" if module else "command"
- self.target = module or command
- super().__init__(str(self))
- def __str__(self) -> str:
- return (
- f"{'Module' if self.type == 'module' else 'command'} {self.target} will not"
- " be overwritten, because it's core"
- )
- class CoreUnloadError(Exception):
- """Is being raised when user tries to unload core module"""
- def __init__(self, module: str):
- self.module = module
- super().__init__()
- def __str__(self) -> str:
- return f"Module {self.module} will not be unloaded, because it's core"
- class SelfUnload(Exception):
- """Silently unloads module, if raised in `client_ready`"""
- def __init__(self, error_message: str = ""):
- super().__init__()
- self._error = error_message
- def __str__(self) -> str:
- return self._error
- class SelfSuspend(Exception):
- """
- Silently suspends module, if raised in `client_ready`
- Commands and watcher will not be registered if raised
- Module won't be unloaded from db and will be unfreezed after restart, unless
- the exception is raised again
- """
- def __init__(self, error_message: str = ""):
- super().__init__()
- self._error = error_message
- def __str__(self) -> str:
- return self._error
- class StopLoop(Exception):
- """Stops the loop, in which is raised"""
- class ModuleConfig(dict):
- """Stores config for modules and apparently libraries"""
- def __init__(self, *entries: typing.Union[str, "ConfigValue"]):
- if all(isinstance(entry, ConfigValue) for entry in entries):
- # New config format processing
- self._config = {config.option: config for config in entries}
- else:
- # Legacy config processing
- keys = []
- values = []
- defaults = []
- docstrings = []
- for i, entry in enumerate(entries):
- if i % 3 == 0:
- keys += [entry]
- elif i % 3 == 1:
- values += [entry]
- defaults += [entry]
- else:
- docstrings += [entry]
- self._config = {
- key: ConfigValue(option=key, default=default, doc=doc)
- for key, default, doc in zip(keys, defaults, docstrings)
- }
- super().__init__(
- {option: config.value for option, config in self._config.items()}
- )
- def getdoc(self, key: str, message: typing.Optional[Message] = None) -> str:
- """Get the documentation by key"""
- ret = self._config[key].doc
- if callable(ret):
- try:
- # Compatibility tweak
- # does nothing in Hikka
- ret = ret(message)
- except Exception:
- ret = ret()
- return ret
- def getdef(self, key: str) -> str:
- """Get the default value by key"""
- return self._config[key].default
- def __setitem__(self, key: str, value: typing.Any):
- self._config[key].value = value
- super().__setitem__(key, value)
- def set_no_raise(self, key: str, value: typing.Any):
- self._config[key].set_no_raise(value)
- super().__setitem__(key, value)
- def __getitem__(self, key: str) -> typing.Any:
- try:
- return self._config[key].value
- except KeyError:
- return None
- def reload(self):
- for key in self._config:
- super().__setitem__(key, self._config[key].value)
- def change_validator(
- self,
- key: str,
- validator: typing.Callable[[JSONSerializable], JSONSerializable],
- ):
- self._config[key].validator = validator
- LibraryConfig = ModuleConfig
- class _Placeholder:
- """Placeholder to determine if the default value is going to be set"""
- async def wrap(func: typing.Callable[[], typing.Awaitable]) -> typing.Any:
- with contextlib.suppress(Exception):
- return await func()
- def syncwrap(func: typing.Callable[[], typing.Any]) -> typing.Any:
- with contextlib.suppress(Exception):
- return func()
- @dataclass(repr=True)
- class ConfigValue:
- option: str
- default: typing.Any = None
- doc: typing.Union[typing.Callable[[], str], str] = "No description"
- value: typing.Any = field(default_factory=_Placeholder)
- validator: typing.Optional[
- typing.Callable[[JSONSerializable], JSONSerializable]
- ] = None
- on_change: typing.Optional[
- typing.Union[typing.Callable[[], typing.Awaitable], typing.Callable]
- ] = None
- def __post_init__(self):
- if isinstance(self.value, _Placeholder):
- self.value = self.default
- def set_no_raise(self, value: typing.Any) -> bool:
- """
- Sets the config value w/o ValidationError being raised
- Should not be used uninternally
- """
- return self.__setattr__("value", value, ignore_validation=True)
- def __setattr__(
- self,
- key: str,
- value: typing.Any,
- *,
- ignore_validation: bool = False,
- ):
- if key == "value":
- try:
- value = ast.literal_eval(value)
- except Exception:
- pass
- # Convert value to list if it's tuple just not to mess up
- # with json convertations
- if isinstance(value, (set, tuple)):
- value = list(value)
- if isinstance(value, list):
- value = [
- item.strip() if isinstance(item, str) else item for item in value
- ]
- if self.validator is not None:
- if value is not None:
- from . import validators
- try:
- value = self.validator.validate(value)
- except validators.ValidationError as e:
- if not ignore_validation:
- raise e
- logger.debug(
- "Config value was broken (%s), so it was reset to %s",
- value,
- self.default,
- )
- value = self.default
- else:
- defaults = {
- "String": "",
- "Integer": 0,
- "Boolean": False,
- "Series": [],
- "Float": 0.0,
- }
- if self.validator.internal_id in defaults:
- logger.debug(
- "Config value was None, so it was reset to %s",
- defaults[self.validator.internal_id],
- )
- value = defaults[self.validator.internal_id]
- # This attribute will tell the `Loader` to save this value in db
- self._save_marker = True
- object.__setattr__(self, key, value)
- if key == "value" and not ignore_validation and callable(self.on_change):
- if inspect.iscoroutinefunction(self.on_change):
- asyncio.ensure_future(wrap(self.on_change))
- else:
- syncwrap(self.on_change)
- def _get_members(
- mod: Module,
- ending: str,
- attribute: typing.Optional[str] = None,
- strict: bool = False,
- ) -> dict:
- """Get method of module, which end with ending"""
- return {
- (
- method_name.rsplit(ending, maxsplit=1)[0]
- if (method_name == ending if strict else method_name.endswith(ending))
- else method_name
- ).lower(): getattr(mod, method_name)
- for method_name in dir(mod)
- if not isinstance(getattr(type(mod), method_name, None), property)
- and callable(getattr(mod, method_name))
- and (
- (method_name == ending if strict else method_name.endswith(ending))
- or attribute
- and getattr(getattr(mod, method_name), attribute, False)
- )
- }
- class CacheRecordEntity:
- def __init__(
- self,
- hashable_entity: "Hashable", # type: ignore # noqa: F821
- resolved_entity: EntityLike,
- exp: int,
- ):
- self.entity = copy.deepcopy(resolved_entity)
- self._hashable_entity = copy.deepcopy(hashable_entity)
- self._exp = round(time.time() + exp)
- self.ts = time.time()
- @property
- def expired(self) -> bool:
- return self._exp < time.time()
- def __eq__(self, record: "CacheRecordEntity") -> bool:
- return hash(record) == hash(self)
- def __hash__(self) -> int:
- return hash(self._hashable_entity)
- def __str__(self) -> str:
- return f"CacheRecordEntity of {self.entity}"
- def __repr__(self) -> str:
- return (
- f"CacheRecordEntity(entity={type(self.entity).__name__}(...),"
- f" exp={self._exp})"
- )
- class CacheRecordPerms:
- def __init__(
- self,
- hashable_entity: "Hashable", # type: ignore # noqa: F821
- hashable_user: "Hashable", # type: ignore # noqa: F821
- resolved_perms: EntityLike,
- exp: int,
- ):
- self.perms = copy.deepcopy(resolved_perms)
- self._hashable_entity = copy.deepcopy(hashable_entity)
- self._hashable_user = copy.deepcopy(hashable_user)
- self._exp = round(time.time() + exp)
- self.ts = time.time()
- @property
- def expired(self) -> bool:
- return self._exp < time.time()
- def __eq__(self, record: "CacheRecordPerms") -> bool:
- return hash(record) == hash(self)
- def __hash__(self) -> int:
- return hash((self._hashable_entity, self._hashable_user))
- def __str__(self) -> str:
- return f"CacheRecordPerms of {self.perms}"
- def __repr__(self) -> str:
- return (
- f"CacheRecordPerms(perms={type(self.perms).__name__}(...), exp={self._exp})"
- )
- class CacheRecordFullChannel:
- def __init__(self, channel_id: int, full_channel: ChannelFull, exp: int):
- self.channel_id = channel_id
- self.full_channel = full_channel
- self._exp = round(time.time() + exp)
- self.ts = time.time()
- @property
- def expired(self) -> bool:
- return self._exp < time.time()
- def __eq__(self, record: "CacheRecordFullChannel") -> bool:
- return hash(record) == hash(self)
- def __hash__(self) -> int:
- return hash((self._hashable_entity, self._hashable_user))
- def __str__(self) -> str:
- return f"CacheRecordFullChannel of {self.channel_id}"
- def __repr__(self) -> str:
- return (
- f"CacheRecordFullChannel(channel_id={self.channel_id}(...),"
- f" exp={self._exp})"
- )
- class CacheRecordFullUser:
- def __init__(self, user_id: int, full_user: UserFull, exp: int):
- self.user_id = user_id
- self.full_user = full_user
- self._exp = round(time.time() + exp)
- self.ts = time.time()
- @property
- def expired(self) -> bool:
- return self._exp < time.time()
- def __eq__(self, record: "CacheRecordFullUser") -> bool:
- return hash(record) == hash(self)
- def __hash__(self) -> int:
- return hash((self._hashable_entity, self._hashable_user))
- def __str__(self) -> str:
- return f"CacheRecordFullUser of {self.user_id}"
- def __repr__(self) -> str:
- return f"CacheRecordFullUser(channel_id={self.user_id}(...), exp={self._exp})"
- def get_commands(mod: Module) -> dict:
- """Introspect the module to get its commands"""
- return _get_members(mod, "cmd", "is_command")
- def get_inline_handlers(mod: Module) -> dict:
- """Introspect the module to get its inline handlers"""
- return _get_members(mod, "_inline_handler", "is_inline_handler")
- def get_callback_handlers(mod: Module) -> dict:
- """Introspect the module to get its callback handlers"""
- return _get_members(mod, "_callback_handler", "is_callback_handler")
- def get_watchers(mod: Module) -> dict:
- """Introspect the module to get its watchers"""
- return _get_members(
- mod,
- "watcher",
- "is_watcher",
- strict=True,
- )
|