evgfilim1 / userbot Goto Github PK

This will require wrapping the original message class from Pyrogram into something that will behave almost the same but will support drafts.

Handling requires registering RawUpdateHandler like this:

UpdateDraftMessage will be received when something is saved as a draft:
https://docs.pyrogram.org/telegram/types/update-draft-message#pyrogram.raw.types.UpdateDraftMessage

Implementing this may open new abilities like silent execution without having to edit sources, looped execution, etc.

.rhadd <reply> <key> <text...> — adds a runtime trigger <key> on <text> that sends <reply>

.rhs — list all runtime triggers

.rhdel <key> — removes a runtime trigger

https://github.com/charliermarsh/ruff

.stats — show some stats for the bot and the user

Example output:

⚙ Uptime: 1h2m3s
⭐️ Doesn't have premium
💬 Total chats: 1000
💬 Unread: 2
📥 Archived chats: 50
👤 Private chats: 400
👥 Groups: 400
📢 Channels: 200
🎞 Saved GIFs: 300
📑 Saved stickers: 200
🕒 Archived stickers: 300
...

Take appropriate icons from https://t.me/addemoji/IconsInTg for premium users

w|weather [city] — Gets the hourly weather for the city

If city is not specified, use the default one from config

Weather API with free per-hour forecast: https://www.tomorrow.io/weather-api/
Geocoding API to get coordinates from city name: https://openweathermap.org/api/geocoding-api

• When looking for .help command:

  userbot/userbot/modules/commands.py

  Lines 388 to 403 in 2d862d1

  	if args := command.args:
  	for h in self._handlers:
  	for cmd in h.commands:
  	match cmd:
  	case re.Pattern() as pattern:
  	matches = pattern.fullmatch(args) is not None
  	case str():
  	matches = cmd == args
  	case _:
  	raise AssertionError(f"Unexpected command type: {type(cmd)}")
  	if matches:
  	usage = h.format_usage(full=True)
  	return _("<b>Help for {args}:</b>\n{usage}").format(
  	args=html.escape(args),
  	usage=html.escape(usage),
  	)

• When checking command conflicts while registering:

  userbot/userbot/modules/commands.py

  Lines 423 to 426 in 2d862d1

  	for handler in self._handlers:
  	for cmd in handler.commands:
  	if cmd in commands:
  	raise ValueError(f"Duplicate command: {cmd}")

.clearsvcs [period] — Clear service messages for a period.

period can be a timedelta (e.g. "1d3h") or a time string (e.g. "12:30" or "2022-12-31_23:59").
If period is not specified, assume "1d"

• Extract commit info from .git directory (fixed in a7d37b8)
• Show "deployments" link only when GITHUB_SHA env var is set (fixed in 84e59fc)

.wakastats — Fetch wakatime.com stats for myself

API: https://wakatime.com/developers

Some clients have a bug when they for some reason don't update a message edited by userbot. Adding a small (~0.1–0.25s) delay would help, I guess.

Chat:

Admin log:

.pydoc <query> — Search Python documentation

.tbapidoc <query> — Search Telegram Bot API documentation

Same behavior as in .remind and .remindme

.lang — Set default language for a chat

If not set, en is assumed

Something like this: https://github.com/evgfilim1/spin_everyday_bot/blob/d5cb7917dfa4408832cebf8339a3257d8ea93973/spin_everyday_bot/lang/translation.py#L20-L45

Start here

diff --git a/.github/workflows/lint.yaml b/.github/workflows/lint.yaml
index 833fcde..77cf54c 100644
--- a/.github/workflows/lint.yaml
+++ b/.github/workflows/lint.yaml
@@ -45,6 +45,7 @@ jobs:
         run: |
           isort --check --diff userbot
           black --check --diff userbot
+          mypy userbot
 
       - name: Check message catalog template is up-to-date
         run: python potctl.py --diff
diff --git a/dev-requirements.txt b/dev-requirements.txt
index 390cec5..41f7d7e 100644
--- a/dev-requirements.txt
+++ b/dev-requirements.txt
@@ -1,3 +1,5 @@
 Babel~=2.10.3
 black~=22.10.0
 isort~=5.10.1
+mypy~=0.982
+types-aiofiles~=22.1.0
diff --git a/pyproject.toml b/pyproject.toml
index bcb9333..e1c88f9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -7,3 +7,52 @@ line_length = 100
 py_version = '310'  # (2022-10-28) isort 5.10.1 doesn't support 3.11
 profile = 'black'
 known_first_party = ['userbot']
+
+[tool.mypy]
+# Config file
+warn_unused_configs = true
+
+# Import discovery
+namespace_packages = true
+
+# Platform configuration
+python_version = "3.11"
+
+# Disallow dynamic typing
+#disallow_any_explicit = true
+disallow_any_generics = true
+disallow_subclassing_any = true
+
+# Untyped definitions and calls
+disallow_untyped_calls = true
+disallow_untyped_defs = true
+disallow_incomplete_defs = true
+check_untyped_defs = true
+disallow_untyped_decorators = true
+
+# None and Optional handling
+no_implicit_optional = true
+
+# Configuring warnings
+warn_redundant_casts = true
+warn_unused_ignores = true
+warn_return_any = true
+warn_unreachable = true
+
+# Miscellaneous strictness flags
+local_partial_types = true
+# Pyrogram doesn't explicitly export some attributes, so this has to be disabled
+no_implicit_reexport = false
+strict_equality = true
+strict = true
+
+# Configuring error messages
+show_error_context = true
+show_column_numbers = true
+show_error_codes = true
+pretty = true
+
+# Advanced options
+plugins = [
+]
+warn_incomplete_stub = true