1 files changed, 42 insertions, 42 deletions

diff --git a/quora/mail.py b/quora/mail.py
index a830502e..c4707860 100644
--- a/quora/mail.py
+++ b/quora/mail.py
@@ -1,66 +1,66 @@
+from json import loads
+from re import findall
+from time import sleep
+
+from fake_useragent import UserAgent
 from requests import Session
-from time     import sleep
-from re       import search, findall
-from json     import loads
+
 
 class Emailnator:
     def __init__(self) -> None:
         self.client = Session()
-        self.client.get('https://www.emailnator.com/', timeout=6)
+        self.client.get("https://www.emailnator.com/", timeout=6)
         self.cookies = self.client.cookies.get_dict()
 
         self.client.headers = {
-            'authority'      : 'www.emailnator.com',
-            'origin'         : 'https://www.emailnator.com',
-            'referer'        : 'https://www.emailnator.com/',
-            'user-agent'     : 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/101.0.0.0 Safari/537.36 Edg/101.0.1722.39',
-            'x-xsrf-token'   : self.client.cookies.get("XSRF-TOKEN")[:-3]+"=",
+            "authority": "www.emailnator.com",
+            "origin": "https://www.emailnator.com",
+            "referer": "https://www.emailnator.com/",
+            "user-agent": UserAgent().random,
+            "x-xsrf-token": self.client.cookies.get("XSRF-TOKEN")[:-3] + "=",
         }
-        
+
         self.email = None
-        
+
     def get_mail(self):
-        response = self.client.post('https://www.emailnator.com/generate-email',json = {
-            'email': [
-                'domain',
-                'plusGmail',
-                'dotGmail',
-            ]
-        })
-        
+        response = self.client.post(
+            "https://www.emailnator.com/generate-email",
+            json={
+                "email": [
+                    "domain",
+                    "plusGmail",
+                    "dotGmail",
+                ]
+            },
+        )
+
         self.email = loads(response.text)["email"][0]
         return self.email
-    
+
     def get_message(self):
         print("waiting for code...")
-        
+
         while True:
             sleep(2)
-            mail_token = self.client.post('https://www.emailnator.com/message-list', 
-                json   = {'email': self.email})
-            
+            mail_token = self.client.post(
+                "https://www.emailnator.com/message-list", json={"email": self.email}
+            )
+
             mail_token = loads(mail_token.text)["messageData"]
-            
+
             if len(mail_token) == 2:
                 print(mail_token[1]["messageID"])
                 break
-        
-        mail_context =  self.client.post('https://www.emailnator.com/message-list', json = {
-            'email'    : self.email,
-            'messageID': mail_token[1]["messageID"],
-        })
-        
-        return mail_context.text
-
-# mail_client  = Emailnator()
-# mail_adress  = mail_client.get_mail()
 
-# print(mail_adress)
+        mail_context = self.client.post(
+            "https://www.emailnator.com/message-list",
+            json={
+                "email": self.email,
+                "messageID": mail_token[1]["messageID"],
+            },
+        )
 
-# mail_content = mail_client.get_message()
-
-# print(mail_content)
-
-# code = findall(r';">(\d{6,7})</div>', mail_content)[0]
-# print(code)
+        return mail_context.text
 
+    def get_verification_code(self):
+        return findall(r';">(\d{6,7})</div>', self.get_message())[0]