add signup module

Author: viperadnan-git

sonicbit/base.py

@@ -1,7 +1,14 @@
 from requests import Session
 
+from sonicbit.constants import Constants
+
 
 class SonicBitBase:
     """Base class for all SonicBit modules."""
 
     session = Session
+
+
+    @staticmethod
+    def url(path: str) -> str:
+        return f"{Constants.API_BASE_URL}{path}"

sonicbit/modules/__init__.py

@@ -3,11 +3,13 @@
 from .remote_download import RemoteDownload
 from .torrent import Torrent
 from .user import User
+from .signup import Signup
 
 __all__ = [
     "Auth",
     "File",
     "RemoteDownload",
     "Torrent",
     "User",
+    "Signup",
 ]

sonicbit/modules/auth.py

@@ -49,7 +49,3 @@ def login(email: str, password: str) -> AuthResponse:
         )
 
         return AuthResponse.from_response(response)
-
-    @staticmethod
-    def url(path: str) -> str:
-        return f"{Constants.API_BASE_URL}{path}"

sonicbit/modules/signup.py

@@ -0,0 +1,82 @@
+from sonicbit.base import SonicBitBase
+from sonicbit.error.error import SonicbitError
+import requests
+from sonicbit.constants import Constants
+import logging
+
+
+logger = logging.getLogger(__name__)
+
+
+class Signup(SonicBitBase):
+    @staticmethod
+    def signup(name: str, email: str, password: str, otp_callback: callable = None):
+        """Signup to SonicBit."""
+
+        data = {
+            "name": name,
+            "email": email,
+            "password": password,
+        }
+
+        logger.debug(f"Signing up as {email}")
+        response = requests.post(
+            SonicBitBase.url("/user/register"), json=data, headers=Constants.API_HEADERS
+        ).json()
+
+        
+        if response["success"] == True:
+            if otp_callback:
+                otp = otp_callback(email)
+                Signup.submit_otp(otp)
+            return True
+        else:
+            raise SonicbitError(f"Failed to signup: {response.get('msg', response)}")
+
+    @staticmethod
+    def submit_otp(otp: str):
+        """Submit OTP to SonicBit."""
+
+        otp = otp.strip()
+
+        if not otp.isdigit() and len(otp) == 6:
+            raise SonicbitError("OTP must be a 6 digit number")
+
+        data = {"code": otp.strip(), "type": "registration", "platform": "Web_Dash_V4"}
+
+        logger.debug(f"Submitting OTP {otp}")
+        response = requests.post(
+            SonicBitBase.url("/verification/code"),
+            json=data,
+            headers=Constants.API_HEADERS,
+        ).json()
+
+
+        if response["success"] == True:
+            token = response['data']['token']
+            Signup._complete_tutorial(token)
+            return True
+        else:
+            raise SonicbitError(f"Failed to submit OTP: {response.get('msg', response)}")
+
+    @staticmethod
+    def _complete_tutorial(token: str):
+        """Complete signup."""
+
+        data = {"delete": True}
+
+        headers = Constants.API_HEADERS
+        headers['Authorization'] = f"Bearer {token}"
+
+        logger.debug(f"Marking tutorial as completed")
+        response = requests.post(
+            SonicBitBase.url("/user/account/welcome_completed"),
+            json=data,
+            headers=headers
+        ).json()
+
+
+        if response.get('success') == True:
+            return True
+        else:
+            raise SonicbitError(f"Failed to complete signup: {response.get('message', response.get('msg', response))}")

sonicbit/sonicbit.py

@@ -5,9 +5,10 @@
 from sonicbit.modules.remote_download import RemoteDownload
 from sonicbit.modules.torrent import Torrent
 from sonicbit.modules.user import User
+from sonicbit.modules.signup import Signup
 
 
-class SonicBit(Auth, User, File, Torrent, RemoteDownload):
+class SonicBit(Auth, Signup, User, File, Torrent, RemoteDownload):
     def __init__(
         self,
         email: str,