fix: use namespace for separator

domainaware · Sep 2, 2024 · 0202d21 · 0202d21
1 parent 17d9599
commit 0202d21
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 4 deletions.
diff --git a/parsedmarc/__init__.py b/parsedmarc/__init__.py
@@ -1416,10 +1416,18 @@ def get_dmarc_reports_from_mailbox(connection: MailboxConnection,
     aggregate_report_msg_uids = []
     forensic_report_msg_uids = []
     smtp_tls_msg_uids = []
-    aggregate_reports_folder = "{0}/Aggregate".format(archive_folder)
-    forensic_reports_folder = "{0}/Forensic".format(archive_folder)
-    smtp_tls_reports_folder = "{0}/SMTP-TLS".format(archive_folder)
-    invalid_reports_folder = "{0}/Invalid".format(archive_folder)
+    aggregate_reports_folder = "{0}{1}Aggregate".format(archive_folder,
+                                                        connection.
+                                                        get_folder_separator())
+    forensic_reports_folder = "{0}{1}Forensic".format(archive_folder,
+                                                      connection.
+                                                      get_folder_separator())
+    smtp_tls_reports_folder = "{0}{1}SMTP-TLS".format(archive_folder,
+                                                      connection.
+                                                      get_folder_separator())
+    invalid_reports_folder = "{0}{1}Invalid".format(archive_folder,
+                                                    connection.
+                                                    get_folder_separator())
 
     if results:
         aggregate_reports = results["aggregate_reports"].copy()

diff --git a/parsedmarc/mail/imap.py b/parsedmarc/mail/imap.py
@@ -26,6 +26,14 @@ def __init__(self,
                                   timeout=timeout,
                                   max_retries=max_retries)
 
+    def get_folder_separator(self):
+        try:
+            namespaces = self._client.namespace()
+            personal = namespaces.personal[0]
+            return personal[1]
+        except (IndexError, NameError):
+            return '/'
+
     def create_folder(self, folder_name: str):
         self._client.create_folder(folder_name)
 

diff --git a/parsedmarc/mail/mailbox_connection.py b/parsedmarc/mail/mailbox_connection.py
@@ -6,6 +6,9 @@ class MailboxConnection(ABC):
     """
     Interface for a mailbox connection
     """
+    def get_folder_separator(self):
+        return "/"
+
     def create_folder(self, folder_name: str):
         raise NotImplementedError