Merge pull request #83 from n1ret/main

Fix auth_session referenced before assignment ToTelethon

setup.py

@@ -1,5 +1,5 @@
 import pathlib
-from setuptools import find_packages, setup
+from setuptools import setup
 import re
 
 README = (pathlib.Path(__file__).parent / "README.md").read_text()
@@ -40,7 +40,7 @@
         "opentele",
     ],
     include_package_data=True,
-    packages=[PACKAGE_NAME],
+    packages=[PACKAGE_NAME, PACKAGE_NAME+'.td', PACKAGE_NAME+'.tl'],
     package_dir={PACKAGE_NAME: SOURCE_DIRECTORY},
     install_requires=requirements,
 )

src/tl/telethon.py

@@ -15,6 +15,7 @@
 from telethon.tl.types.auth import LoginTokenMigrateTo
 import logging
 import warnings
+from typing import Awaitable
 
 
 @extend_override_class
@@ -624,7 +625,9 @@ async def QRLoginToNewClient(
 
                 # for the above reason, we should check if we're already authorized
                 if isinstance(qr_login._resp, types.auth.LoginTokenSuccess):
-                    newClient._on_login(qr_login._resp.authorization.user)
+                    coro = newClient._on_login(qr_login._resp.authorization.user)
+                    if isinstance(coro, Awaitable):
+                        await coro
                     break
 
                 # calculate when will the qr token expire
@@ -683,7 +686,9 @@ async def QRLoginToNewClient(
                     )
 
                     # successful log in
-                    newClient._on_login(result.user)  # type: ignore
+                    coro = newClient._on_login(result.user)  # type: ignore
+                    if isinstance(coro, Awaitable):
+                        await coro
                     break
 
                 except PasswordHashInvalidError as e:
@@ -892,6 +897,8 @@ async def FromTDesktop(
                 raise TypeError(
                     "The given session must be a str or a Session instance."
                 )
+            else:  # session is instance of Session
+                auth_session = session
 
         auth_session.set_dc(endpoint.id, endpoint.ip, endpoint.port)  # type: ignore
         auth_session.auth_key = AuthKey(account.authKey.key)  # type: ignore