chore: add __all__ to where missing

alissonlauffer · alissonlauffer · commit 57ad412ddded · 2023-11-01T20:35:00.000-03:00
also change some type comments into formal type annotations
diff --git a/hydrogram/__init__.py b/hydrogram/__init__.py
@@ -41,3 +41,20 @@ class ContinuePropagation(StopAsyncIteration):
 from .sync import idle, compose
 
 crypto_executor = ThreadPoolExecutor(1, thread_name_prefix="CryptoWorker")
+
+
+__all__ = [
+    "Client",
+    "idle",
+    "compose",
+    "StopTransmission",
+    "StopPropagation",
+    "ContinuePropagation",
+    "crypto_executor",
+    "raw",
+    "types",
+    "filters",
+    "handlers",
+    "emoji",
+    "enums",
+]
diff --git a/hydrogram/connection/__init__.py b/hydrogram/connection/__init__.py
@@ -18,3 +18,6 @@
 #  along with Hydrogram.  If not, see <http://www.gnu.org/licenses/>.
 
 from .connection import Connection
+
+
+__all__ = ["Connection"]
diff --git a/hydrogram/connection/transport/tcp/__init__.py b/hydrogram/connection/transport/tcp/__init__.py
@@ -23,3 +23,13 @@
 from .tcp_full import TCPFull
 from .tcp_intermediate import TCPIntermediate
 from .tcp_intermediate_o import TCPIntermediateO
+
+
+__all__ = [
+    "TCP",
+    "TCPAbridged",
+    "TCPAbridgedO",
+    "TCPFull",
+    "TCPIntermediate",
+    "TCPIntermediateO",
+]
diff --git a/hydrogram/errors/__init__.py b/hydrogram/errors/__init__.py
@@ -18,7 +18,7 @@
 #  along with Hydrogram.  If not, see <http://www.gnu.org/licenses/>.
 
 from .exceptions import *
-from .rpc_error import UnknownError
+from .rpc_error import UnknownError as UnknownError
 
 
 class BadMsgNotification(Exception):
diff --git a/hydrogram/handlers/__init__.py b/hydrogram/handlers/__init__.py
@@ -29,3 +29,19 @@
 from .poll_handler import PollHandler
 from .raw_update_handler import RawUpdateHandler
 from .user_status_handler import UserStatusHandler
+
+
+__all__ = [
+    "CallbackQueryHandler",
+    "ChatJoinRequestHandler",
+    "ChatMemberUpdatedHandler",
+    "ChosenInlineResultHandler",
+    "DeletedMessagesHandler",
+    "DisconnectHandler",
+    "EditedMessageHandler",
+    "InlineQueryHandler",
+    "MessageHandler",
+    "PollHandler",
+    "RawUpdateHandler",
+    "UserStatusHandler",
+]
diff --git a/hydrogram/parser/__init__.py b/hydrogram/parser/__init__.py
@@ -18,3 +18,6 @@
 #  along with Hydrogram.  If not, see <http://www.gnu.org/licenses/>.
 
 from .parser import Parser
+
+
+__all__ = ["Parser"]
diff --git a/hydrogram/raw/__init__.py b/hydrogram/raw/__init__.py
@@ -25,3 +25,6 @@
 for k, v in objects.items():
     path, name = v.rsplit(".", 1)
     objects[k] = getattr(import_module(path), name)
+
+
+__all__ = ["types", "functions", "base", "core", "objects"]
diff --git a/hydrogram/raw/core/__init__.py b/hydrogram/raw/core/__init__.py
@@ -30,3 +30,25 @@
 from .primitives.string import String
 from .primitives.vector import Vector
 from .tl_object import TLObject
+
+
+__all__ = [
+    "FutureSalt",
+    "FutureSalts",
+    "GzipPacked",
+    "List",
+    "Message",
+    "MsgContainer",
+    "Bool",
+    "BoolFalse",
+    "BoolTrue",
+    "Bytes",
+    "Double",
+    "Int",
+    "Long",
+    "Int128",
+    "Int256",
+    "String",
+    "Vector",
+    "TLObject",
+]
diff --git a/hydrogram/raw/core/primitives/__init__.py b/hydrogram/raw/core/primitives/__init__.py
@@ -23,3 +23,18 @@
 from .int import Int, Long, Int128, Int256
 from .string import String
 from .vector import Vector
+
+
+__all__ = [
+    "Bool",
+    "BoolFalse",
+    "BoolTrue",
+    "Bytes",
+    "Double",
+    "Int",
+    "Long",
+    "Int128",
+    "Int256",
+    "String",
+    "Vector",
+]
diff --git a/hydrogram/session/__init__.py b/hydrogram/session/__init__.py
@@ -19,3 +19,6 @@
 
 from .auth import Auth
 from .session import Session
+
+
+__all__ = ["Auth", "Session"]
diff --git a/hydrogram/session/internals/__init__.py b/hydrogram/session/internals/__init__.py
@@ -20,3 +20,6 @@
 from .data_center import DataCenter
 from .msg_factory import MsgFactory
 from .msg_id import MsgId
+
+
+__all__ = ["DataCenter", "MsgFactory", "MsgId"]
diff --git a/hydrogram/storage/__init__.py b/hydrogram/storage/__init__.py
@@ -20,3 +20,6 @@
 from .file_storage import FileStorage
 from .memory_storage import MemoryStorage
 from .storage import Storage
+
+
+__all__ = ["FileStorage", "MemoryStorage", "Storage"]
diff --git a/hydrogram/storage/sqlite_storage.py b/hydrogram/storage/sqlite_storage.py
@@ -91,7 +91,7 @@ class SQLiteStorage(Storage):
     def __init__(self, name: str):
         super().__init__(name)
 
-        self.conn = None  # type: sqlite3.Connection
+        self.conn: sqlite3.Connection = None
 
     def create(self):
         with self.conn:
diff --git a/hydrogram/types/__init__.py b/hydrogram/types/__init__.py
@@ -22,8 +22,8 @@
 from .inline_mode import *
 from .input_media import *
 from .input_message_content import *
-from .list import List
+from .list import List as List
 from .messages_and_media import *
-from .object import Object
+from .object import Object as Object
 from .update import *
 from .user_and_chats import *
diff --git a/hydrogram/types/messages_and_media/message.py b/hydrogram/types/messages_and_media/message.py
@@ -655,7 +655,7 @@ async def _parse(
             forward_signature = None
             forward_date = None
 
-            forward_header = message.fwd_from  # type: raw.types.MessageFwdHeader
+            forward_header: raw.types.MessageFwdHeader = message.fwd_from
 
             if forward_header:
                 forward_date = utils.timestamp_to_datetime(forward_header.date)
