From 0202d211168d822665a6071f478017911cfa8d0e Mon Sep 17 00:00:00 2001
From: "Gaige B. Paulsen" <gaige@cluetrust.com>
Date: Mon, 2 Sep 2024 12:09:49 -0400
Subject: [PATCH] fix: use namespace for separator

---
 parsedmarc/__init__.py                | 16 ++++++++++++----
 parsedmarc/mail/imap.py               |  8 ++++++++
 parsedmarc/mail/mailbox_connection.py |  3 +++
 3 files changed, 23 insertions(+), 4 deletions(-)

diff --git a/parsedmarc/__init__.py b/parsedmarc/__init__.py
index cb859656..33a8e71e 100644
--- 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
index 150185e1..4ffa55fd 100644
--- 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
index ecaa2cbb..ba7c2cf7 100644
--- 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