From c87965478c8eeae2124fd77c1f62cf267ab31b0e Mon Sep 17 00:00:00 2001 From: PAN Lei Date: Tue, 17 Sep 2024 20:44:30 +0200 Subject: [PATCH] support gmx.sg --- mail/mail_constants.py | 3 ++- utils/cookies_manager.py | 13 +++++++++++++ 2 files changed, 15 insertions(+), 1 deletion(-) create mode 100644 utils/cookies_manager.py diff --git a/mail/mail_constants.py b/mail/mail_constants.py index dc84b5c..92ad891 100755 --- a/mail/mail_constants.py +++ b/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) diff --git a/utils/cookies_manager.py b/utils/cookies_manager.py new file mode 100644 index 0000000..40a9d12 --- /dev/null +++ b/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