upd: base_url | domain usage

Author: iamlostshe

README.md

@@ -22,18 +22,20 @@ from mailtm import Email
 
 async def main() -> None:
     """Start msg cycle."""
+
     def listener(message: dict) -> None:
         print("\nSubject: " + message["subject"])
         print("Content: " + message["text"] if message["text"] else message["html"])
 
-    # Get Domains
+    # Get base url and domains
     test = Email()
     await test.init()
-    print("\nDomain: " + test.domain)
+    print("\nBase url:", test.base_url)
+    print("Domain:", test.domain)
 
     # Make new email address
     await test.register()
-    print("\nEmail Adress: " + str(test.address))
+    print("\nEmail Adress:", test.address)
 
     # Start listening
     await test.start(listener)
@@ -42,7 +44,6 @@ async def main() -> None:
 
 if __name__ == "__main__":
     asyncio.run(main())
-
 ```
 
 # Documentation

src/mailtm/consts.py

@@ -1,6 +1,6 @@
 """Some constant values."""
 
-BASE_URL = "https://api.mail.tm/"
+BASE_URLS = ["https://api.mail.tm/", "https://api.mail.gw/"]
 
 USERNAME_ALPHABET = "abcdefghijklmnopqrstuvwxyz0123456789"
 PASSWORD_ALPHABET = (

src/mailtm/email.py

@@ -4,7 +4,7 @@
 
 from aiohttp import ClientSession
 
-from .consts import BASE_URL, PASSWORD_ALPHABET, USERNAME_ALPHABET
+from .consts import BASE_URLS, PASSWORD_ALPHABET, USERNAME_ALPHABET
 from .message import Listen
 
 
@@ -27,21 +27,23 @@ def password_gen(
 class Email(Listen):
     """The class module for interacting with EMail."""
 
-    async def init(self) -> None:
+    async def init(self, base_url: str | None = random.choice(BASE_URLS)) -> None:  # noqa: S311
         """Init temporary Email class."""
-        self.session = ClientSession(headers={"Content-Type": "application/json"})
+        self.base_url = base_url
+        self.session = ClientSession(
+            base_url=self.base_url,
+            headers={"Content-Type": "application/json"},
+        )
         await self.domains()
 
     async def domains(self) -> None:
         """Init Email domains."""
-        url = f"{BASE_URL}domains"
+        url = "domains"
 
         async with self.session.get(url) as response:
-            data = await response.json()
+            data = (await response.json())["hydra:member"]
 
-        for domain in data["hydra:member"]:
-            if domain["isActive"]:
-                self.domain = domain["domain"]
+        self.domain = random.choice([d for d in data if d["isActive"]])["domain"]  # noqa: S311
 
     async def register(
         self,
@@ -55,7 +57,7 @@ async def register(
 
         self.address = f"{username}@{self.domain}"
 
-        url = f"{BASE_URL}accounts"
+        url = "accounts"
         payload = {
             "address": self.address,
             "password": password,
@@ -66,7 +68,7 @@ async def register(
 
     async def get_token(self, password: str) -> None:
         """Get token."""
-        url = f"{BASE_URL}token"
+        url = "token"
         payload = {
             "address": self.address,
             "password": password,

src/mailtm/message.py

@@ -2,8 +2,6 @@
 
 import asyncio
 
-from .consts import BASE_URL
-
 
 class Listen:
     """Listener class."""
@@ -14,7 +12,7 @@ def __init__(self) -> None:
 
     async def message_list(self) -> list:
         """Get a list of messages."""
-        url = f"{BASE_URL}messages"
+        url = "messages"
 
         async with self.session.get(url) as response:
             data = await response.json()
@@ -27,7 +25,7 @@ async def message_list(self) -> list:
 
     async def message(self, idx: str) -> dict:
         """Get the text of the message."""
-        url = f"{BASE_URL}messages/{idx}"
+        url = f"messages/{idx}"
         async with self.session.get(url) as response:
             return await response.json()