wintergram-userbot 1.2__tar.gz → 1.4__tar.gz

{wintergram_userbot-1.2/wintergram_userbot.egg-info → wintergram_userbot-1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wintergram-userbot
-Version: 1.2
+Version: 1.4
 Summary: WinterGram Userbot for custom MTProto server
 Author: WinterGram Team
 Requires-Python: >=3.8
@@ -122,7 +122,7 @@ history = await client.get_history(chat_id, limit=50)
 ### Свойства события
 
 - `event.id` — ID сообщения
-- `event.text` — текст сообщения (bytes или str)
+- `event.text` — текст сообщения (всегда `str`)
 - `event.date` — время отправки
 - `event.chat_id` — ID чата
 - `event.sender_id` — ID отправителя
@@ -130,6 +130,7 @@ history = await client.get_history(chat_id, limit=50)
 - `event.is_private` — True если личное сообщение
 - `event.is_channel` — True если сообщение из канала
 - `event.from_channel` — True если отправлено каналом
+- `event.message` — внутренний dict сообщения (текст тоже `str`)
 
 ## Требования
 

{wintergram_userbot-1.2 → wintergram_userbot-1.4}/README.md

@@ -111,7 +111,7 @@ history = await client.get_history(chat_id, limit=50)
 ### Свойства события
 
 - `event.id` — ID сообщения
-- `event.text` — текст сообщения (bytes или str)
+- `event.text` — текст сообщения (всегда `str`)
 - `event.date` — время отправки
 - `event.chat_id` — ID чата
 - `event.sender_id` — ID отправителя
@@ -119,6 +119,7 @@ history = await client.get_history(chat_id, limit=50)
 - `event.is_private` — True если личное сообщение
 - `event.is_channel` — True если сообщение из канала
 - `event.from_channel` — True если отправлено каналом
+- `event.message` — внутренний dict сообщения (текст тоже `str`)
 
 ## Требования
 

{wintergram_userbot-1.2 → wintergram_userbot-1.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "wintergram-userbot"
-version = "1.2"
+version = "1.4"
 description = "WinterGram Userbot for custom MTProto server"
 readme = "README.md"
 authors = [{name = "WinterGram Team"}]

{wintergram_userbot-1.2 → wintergram_userbot-1.4}/wintergram_userbot/__init__.py

@@ -4,7 +4,7 @@
 Поддерживаются слои (layers) от 167 до 228 включительно.
 """
 
-__version__ = "1.2"
+__version__ = "1.4"
 
 from .client import WinterGramClient
 from .errors import (

{wintergram_userbot-1.2 → wintergram_userbot-1.4}/wintergram_userbot/api/functions.py

@@ -183,6 +183,16 @@ class Functions:
         """users.getUsers([InputUser...])."""
         return await self.conn.rpc("users.getUsers", {"id": ids})
 
+    async def resolve_username(self, username: str):
+        """contacts.resolveUsername - ищет пользователя по @юзернейму.
+
+        Returns:
+            contacts.ResolvedPeer с полями peer, users, chats
+        """
+        return await self.conn.rpc("contacts.resolveUsername", {
+            "username": (username or "").strip().lstrip("@"),
+        })
+
     # ------------------------------------------------------------------ #
     #  отправка / история
     # ------------------------------------------------------------------ #

{wintergram_userbot-1.2 → wintergram_userbot-1.4}/wintergram_userbot/api/updates.py

@@ -27,6 +27,15 @@ def _as_dict(obj):
     return dict(obj)
 
 
+def _as_text(value):
+    """TL-строки приходят как bytes - приводим к str."""
+    if value is None:
+        return ""
+    if isinstance(value, (bytes, bytearray)):
+        return bytes(value).decode("utf-8", "replace")
+    return str(value)
+
+
 class Event:
     """Базовое событие."""
 
@@ -37,6 +46,27 @@ class Event:
     def raw_dict(self):
         return _as_dict(self.raw)
 
+    @property
+    def message(self):
+        """Внутренний dict сообщения (updateNewMessage.message и т.п.).
+
+        Для updateShort* апдейтов, где сообщение лежит прямо в update,
+        возвращается сам update. Текст нормализован в str.
+        """
+        d = _as_dict(self.raw)
+        inner = d.get("message")
+        # у updateNew* поле message - вложенный dict, у updateShort* - строка
+        if isinstance(inner, (dict, TLObject)):
+            m = _as_dict(inner)
+            if isinstance(m.get("message"), (bytes, bytearray)):
+                m = dict(m)
+                m["message"] = _as_text(m["message"])
+            return m
+        if isinstance(inner, (bytes, bytearray)):
+            d = dict(d)
+            d["message"] = _as_text(inner)
+        return d
+
 
 class NewMessage(Event):
     """Новое сообщение.
@@ -48,7 +78,7 @@ class NewMessage(Event):
                  sender_id=None, out=False, from_channel=False):
         super().__init__(raw)
         self.id = id
-        self.text = text
+        self.text = _as_text(text)
         self.date = date
         self.chat_id = chat_id
         self.sender_id = sender_id

{wintergram_userbot-1.2 → wintergram_userbot-1.4}/wintergram_userbot/client.py

@@ -778,13 +778,59 @@ class WinterGramClient:
             return self.functions.input_peer_self()
         return await self.resolve_peer(peer)
 
+    async def resolve_username(self, username):
+        """Ищет пользователя по юзернейму (с @ или без) через contacts.resolveUsername.
+
+        Args:
+            username: "@nick" или "nick"
+
+        Returns:
+            InputPeer текущей эры
+
+        Raises:
+            WinterGramError: если юзернейм не найден
+        """
+        name = (username or "").strip().lstrip("@")
+        if not name:
+            raise WinterGramError("empty username")
+
+        _, result = await self.functions.resolve_username(name)
+
+        peer = result.get("peer") or {}
+        users = result.get("users") or []
+        chats = result.get("chats") or []
+
+        if "user_id" in peer:
+            target_id = peer.get("user_id")
+            user = next((u for u in users if u.get("id") == target_id), None)
+            if user is None and users:
+                user = users[0]
+            if user is None or user.get("access_hash") is None:
+                raise WinterGramError("no access_hash for @%s" % name)
+            self.storage.store_peer(user["id"], user["access_hash"], "user")
+            return self.functions.input_peer_user(user["id"], user["access_hash"])
+
+        if "chat_id" in peer:
+            return self.functions.input_peer_chat(peer["chat_id"])
+
+        if "channel_id" in peer:
+            target_id = peer.get("channel_id")
+            chat = next((c for c in chats if c.get("id") == target_id), None)
+            access_hash = (chat or {}).get("access_hash", 0)
+            self.storage.store_peer(target_id, access_hash, "channel")
+            return self.functions.input_peer_channel(target_id, access_hash)
+
+        raise WinterGramError("username @%s not found" % name)
+
     async def resolve_peer(self, peer):
         """Пиринг в InputPeer текущей эры (по возможности через кэш)."""
         if isinstance(peer, str):
             low = peer.lower()
             if low in ("me", "self"):
                 return self.functions.input_peer_self()
-            raise RPCError(0, "username resolution not supported; pass id or 'me'")
+            if low.startswith("@") or peer[:1].isalpha() or "_" in peer:
+                return await self.resolve_username(peer)
+            raise RPCError(0, "pass id, 'me' or @username")
         peer_id = int(peer)
         cached = self.storage.get_peer(abs(peer_id)) if self.storage._db else None
         access_hash = cached[0] if cached else 0

{wintergram_userbot-1.2 → wintergram_userbot-1.4}/wintergram_userbot/network/mtproto.py

@@ -67,6 +67,21 @@ ID_PONG = 0x347773C5
 ID_GZIP_PACKED = 0x3072CFA1
 ID_PING = 0x7ABE77EC
 
+# little-endian сигнатура gzip_packed в начале тела ответа
+GZIP_MAGIC_LE = struct.pack("<I", ID_GZIP_PACKED)
+
+
+def inflate(data: bytes) -> bytes:
+    """Распаковывает gzip/zlib/deflate - сервер встречается со всеми тремя."""
+    if not data:
+        raise ValueError("empty packed payload")
+    for wbits in (16 + zlib.MAX_WBITS, zlib.MAX_WBITS, -zlib.MAX_WBITS):
+        try:
+            return zlib.decompress(data, wbits)
+        except zlib.error:
+            continue
+    raise ValueError("cannot inflate %d bytes" % len(data))
+
 # якоря эр (по убыванию) для определения layer
 ERA_ORDER = (228, 202, 198, 181, 167)
 
@@ -309,7 +324,33 @@ class MTProtoConnection:
         desc = self._schema.by_id(ident)
         if desc is None:
             raise TLParseError("unknown constructor 0x%08x" % ident)
-        return decode_constructor(r, desc, self._schema)
+        return self.unpack_object(decode_constructor(r, desc, self._schema))
+
+    @staticmethod
+    def _is_gzip(obj) -> bool:
+        if not isinstance(obj, dict):
+            return False
+        if obj.get("_id") == ID_GZIP_PACKED:
+            return True
+        return str(obj.get("_", "")).lower() == "gzip_packed"
+
+    def unpack_object(self, obj, depth=0):
+        """Распаковывает gzip_packed и декодирует то, что внутри.
+
+        Сервер заворачивает крупные ответы (messages.getDialogs и др.)
+        в gzip_packed: он приходит полем result у rpc_result, поэтому
+        проверка по сигнатуре тела ответа его не ловит.
+        """
+        while depth < 4 and self._is_gzip(obj):
+            packed = obj.get("packed_data") or obj.get("bytes") or b""
+            raw = inflate(packed)
+            rr = TLReader(raw)
+            desc = self._schema.by_id(rr.read_uint())
+            if desc is None:
+                raise TLParseError("unknown constructor inside gzip_packed")
+            obj = decode_constructor(rr, desc, self._schema)
+            depth += 1
+        return obj
 
     def _handle_rpc_result(self, req_msg_id: int, r: TLReader):
         raw = bytes(r.data[r.pos:])
@@ -323,10 +364,10 @@ class MTProtoConnection:
             self._pending.pop(req_msg_id, None)
             return
         body = raw
-        if body[:4] == b"\x30\xcf\xa1\x07":        # gzip_packed little-endian
+        if body[:4] == GZIP_MAGIC_LE:
             pack_r = TLReader(body)
             pack_r.read_uint()
-            body = zlib.decompress(pack_r.read_bytes())
+            body = inflate(pack_r.read_bytes())
         rr = TLReader(body)
         ident = rr.read_uint()
         if ident == ID_RPC_ERROR:
@@ -346,6 +387,7 @@ class MTProtoConnection:
             return
         try:
             obj = decode_constructor(rr, desc, self._schema)
+            obj = self.unpack_object(obj)
         except TLParseError as e:
             fut.set_exception(
                 RPCError(0, "parse failed for 0x%08x: %s (raw=%s)" % (ident, e, body.hex()))

{wintergram_userbot-1.2 → wintergram_userbot-1.4/wintergram_userbot.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wintergram-userbot
-Version: 1.2
+Version: 1.4
 Summary: WinterGram Userbot for custom MTProto server
 Author: WinterGram Team
 Requires-Python: >=3.8
@@ -122,7 +122,7 @@ history = await client.get_history(chat_id, limit=50)
 ### Свойства события
 
 - `event.id` — ID сообщения
-- `event.text` — текст сообщения (bytes или str)
+- `event.text` — текст сообщения (всегда `str`)
 - `event.date` — время отправки
 - `event.chat_id` — ID чата
 - `event.sender_id` — ID отправителя
@@ -130,6 +130,7 @@ history = await client.get_history(chat_id, limit=50)
 - `event.is_private` — True если личное сообщение
 - `event.is_channel` — True если сообщение из канала
 - `event.from_channel` — True если отправлено каналом
+- `event.message` — внутренний dict сообщения (текст тоже `str`)
 
 ## Требования