support gmx.sg

mail/mail_constants.py

@@ -18,6 +18,7 @@ DOMAIN_GMX_NET = "gmx.net"
 DOMAIN_GMX_AT = "gmx.at"
 DOMAIN_GMX_FR = "gmx.fr"
 DOMAIN_GMX_US = "gmx.us"
+DOMAIN_GMX_SG = "gmx.sg"
 DOMAIN_GMX_CH = "gmx.ch"
 DOMAIN_GMX_PT = "gmx.pt"
 DOMAIN_ONET = "onet.pl"
@@ -89,7 +90,7 @@ def create_imap(login: str):
     elif DOMAIN_FIREMAIL_DE in login:
         # imap = imaplib.IMAP4_SSL(SERVER_FIREMAIL_DE)
         imap = IMAPClient(SERVER_FIREMAIL_DE, use_uid=True)
-    elif DOMAIN_GMX in login or DOMAIN_GMX_FR in login or DOMAIN_GMX_US in login or DOMAIN_GMX_CH in login or DOMAIN_GMX_PT in login:
+    elif DOMAIN_GMX in login or DOMAIN_GMX_FR in login or DOMAIN_GMX_US in login or DOMAIN_GMX_CH in login or DOMAIN_GMX_PT in login or DOMAIN_GMX_SG in login:
         imap = imaplib.IMAP4_SSL(SERVER_GMX)
     elif DOMAIN_SINA in login:
         # imap = imaplib.IMAP4_SSL(IMAP_SERVER_SINA)

utils/cookies_manager.py

@@ -0,0 +1,13 @@
+from http.cookies import SimpleCookie
+
+
+def get_new_cookies_from_response(response):
+    _cookie = SimpleCookie()
+    cookies_to_set = response.headers['set-cookie']
+    _cookie.load(cookies_to_set)
+    new_cookies = {k: v.value for k, v in _cookie.items()}
+    new_coolies_str = ""
+    for key in new_cookies:
+        new_coolies_str = new_coolies_str + key + "=" + new_cookies[key] + ";"
+    print(new_coolies_str)
+    return new_coolies_str