resolve circular import error

Author: viperadnan-git

sonicbit/__init__.py

@@ -1,10 +1,9 @@
 import logging
 
-from sonicbit.handlers.token_file import TokenFileHandler
 from sonicbit.sonicbit import SonicBit
 
 __version__ = "0.1.1"
-__all__ = ["SonicBit", "TokenFileHandler"]
+__all__ = ["SonicBit"]
 
 logging.basicConfig(
     level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"

sonicbit/handlers/token_file.py

@@ -1,18 +1,18 @@
 import json
 import os
 
-from sonicbit.types import AuthResponse
+from sonicbit.types.auth_response import AuthResponse
 
 
 class TokenHandler:
     def __init__(self):
         pass
 
     def write(self, email: str, auth: AuthResponse) -> None:
-        print(f"{email} sesion is {auth.sesion}")
+        print(f"{email}'s token is {auth.token}")
 
     def read(self, email: str) -> str | None:
-        return input(f"Enter {email} sesion: ")
+        return input(f"Enter {email}'s token: ")
 
 
 class TokenFileHandler(TokenHandler):

sonicbit/modules/__init__.py

@@ -0,0 +1,15 @@
+from .base import SonicBitBase
+from .auth import Auth
+from .file import File
+from .remote_download import RemoteDownload
+from .torrent import Torrent
+from .user import User
+
+__all__ = [
+    "SonicBitBase",
+    "Auth",
+    "File",
+    "RemoteDownload",
+    "Torrent",
+    "User",
+]

sonicbit/modules/auth.py

@@ -3,7 +3,7 @@
 from requests import Session, request
 
 from sonicbit.constants import Constants
-from sonicbit.handlers.token_file import TokenHandler, TokenFileHandler
+from sonicbit.handlers.token_file import TokenHandler
 from sonicbit.modules.base import SonicBitBase
 from sonicbit.types import AuthResponse
 
@@ -15,8 +15,8 @@ def __init__(
         self,
         email: str,
         password: str,
-        token: str = None,
-        token_handler: TokenHandler = TokenFileHandler(),
+        token: str | None,
+        token_handler: TokenHandler,
     ):
         logger.debug("Initializing Auth")
         self.session = Session()

sonicbit/sonicbit.py

@@ -3,8 +3,8 @@
 from sonicbit.modules.remote_download import RemoteDownload
 from sonicbit.modules.torrent import Torrent
 from sonicbit.modules.user import User
-
+from sonicbit.handlers.token_file import TokenHandler, TokenFileHandler
 
 class SonicBit(Auth, User, File, Torrent, RemoteDownload):
-    def __init__(self, email: str, password: str, token: str = None):
-        super().__init__(email, password, token)
+    def __init__(self, email: str, password: str, token: str = None, token_handler: TokenHandler = TokenFileHandler()):
+        super().__init__(email, password, token, token_handler)