Add files via upload

src/archive_emails.py

@@ -0,0 +1,152 @@
+import win32com.client
+import sys
+import datetime
+
+
+def list_outlook_mailboxes():
+    """Returns a list of available mailboxes in Outlook."""
+    outlook = win32com.client.Dispatch("Outlook.Application")
+    namespace = outlook.GetNamespace("MAPI")
+    mailboxes = [folder.Name for folder in namespace.Folders]
+    return mailboxes
+
+def find_online_archive(mailboxes, selected_mailbox):
+    """Finds the corresponding Online Archive for the selected mailbox."""
+    archive_name = f"Online Archive - {selected_mailbox}"
+    for mailbox in mailboxes:
+        if mailbox == archive_name:
+            print(f"Mapped Online Archive: {mailbox}")
+            return mailbox
+    print("No Online Archive found for the selected mailbox.")
+    return None
+
+def list_main_folders(mailbox_name):
+    """Lists main folders of the selected mailbox."""
+    outlook = win32com.client.Dispatch("Outlook.Application")
+    namespace = outlook.GetNamespace("MAPI")
+    
+    try:
+        mailbox = namespace.Folders[mailbox_name]
+    except Exception:
+        print(f"Cannot find mailbox: {mailbox_name}. Check the name in Outlook.")
+        return []
+    
+    return [folder.Name for folder in mailbox.Folders]
+
+def main_archive():
+    total_emails_moved = 0
+    total_size_moved_kb = 0
+    """Main function to select a mailbox and folders for archiving."""
+    mailboxes = list_outlook_mailboxes()
+    if not mailboxes:
+        print("No mailboxes found in Outlook.")
+        sys.exit(1)
+
+    print("Available mailboxes:")
+    for idx, mailbox in enumerate(mailboxes, 1):
+        print(f"{idx}. {mailbox}")
+
+    try:
+        choice = int(input("Select mailbox number: ")) - 1
+        if 0 <= choice < len(mailboxes):
+            mailbox_name = mailboxes[choice]
+        else:
+            raise ValueError
+    except ValueError:
+        print("Invalid mailbox selection.")
+        sys.exit(1)
+
+    archive_mailbox = find_online_archive(mailboxes, mailbox_name)
+    if not archive_mailbox:
+        print("No corresponding Online Archive found. Exiting.")
+        sys.exit(1)
+
+    # ✅ Sprawdźmy, czy wybrana skrzynka jest poprawna
+    print(f"Selected mailbox: {mailbox_name}")
+
+    main_folders = list_main_folders(mailbox_name)
+    if not main_folders:
+        print("No main folders found in the selected mailbox.")
+        sys.exit(1)
+
+    print(f"Main folders in '{mailbox_name}':")
+    for idx, folder in enumerate(main_folders, 1):
+        print(f"{idx}. {folder}")
+
+    selected_folders = input("Select folders by number (comma-separated, e.g., 1,3,6): ")
+    try:
+        selected_indices = [int(i.strip()) - 1 for i in selected_folders.split(",")]
+        selected_folder_names = [main_folders[i] for i in selected_indices if 0 <= i < len(main_folders)]
+        print(f"Selected folders: {', '.join(selected_folder_names)}")
+
+        run_type = input("Do you want to perform a dry-run or actually move emails? (dry-run/move): ").strip().lower()
+        if run_type not in ['dry-run', 'move']:
+            print("Invalid selection. Please restart and choose 'dry-run' or 'move'.")
+            sys.exit(1)
+
+        outlook = win32com.client.Dispatch("Outlook.Application")
+        namespace = outlook.GetNamespace("MAPI")
+        mailbox = namespace.Folders[mailbox_name]
+
+        for folder_name in selected_folder_names:
+            try:
+                print(f"Accessing folder: {folder_name}")
+                folder = mailbox.Folders[folder_name]
+
+                try:
+                    archive_folder = namespace.Folders[archive_mailbox].Folders[folder_name]
+                except Exception:
+                    print(f"Archive folder '{folder_name}' does not exist. Attempting to create it...")
+                    try:
+                        archive_folder = namespace.Folders[archive_mailbox].Folders.Add(folder_name)
+                    except Exception as e:
+                        print(f"Failed to create archive folder '{folder_name}'. Skipping. Error: {e}")
+                        continue
+
+                emails_moved, size_moved_kb = process_folder(folder, archive_folder, run_type)
+                total_emails_moved += emails_moved
+                total_size_moved_kb += size_moved_kb
+            except Exception as e:
+               print(f"Error accessing folder {folder_name}: {e}")
+
+        print("Dry-run complete. No emails were moved." if run_type == 'dry-run' else "Email move completed.")
+        print(f"Total emails processed: {total_emails_moved}")
+        print(f"Total size processed: {total_size_moved_kb / 1024:.2f} MB")
+    except (ValueError, IndexError):
+        print("Invalid folder selection.")
+        sys.exit(1)
+
+def process_folder(folder, archive_folder, run_type, depth=0):
+    old_email_count = 0
+    total_size_kb = 0
+    """Processes a folder and its subfolders, counting and optionally moving emails."""
+    old_email_count = 0
+    total_size_kb = 0
+
+    try:
+        # print(f"{'  ' * depth}Processing folder: {folder.Name}")
+        for item in folder.Items:
+            if hasattr(item, 'ReceivedTime') and item.ReceivedTime < (datetime.datetime.now(datetime.timezone.utc) - datetime.timedelta(days=365)):
+                old_email_count += 1
+                total_size_kb += item.Size / 1024
+                if run_type == 'move':
+                    item.Move(archive_folder)
+    except Exception as e:
+        print(f"Error processing folder {folder.Name}: {e}")
+
+    # ✅ PRZENIESIONE POZA EXCEPT → teraz zawsze przetwarza podfoldery
+    if old_email_count > 0 :
+        print(f"{'  ' * depth}Folder '{folder.Name}': {old_email_count} emails ({total_size_kb / 1024:.2f} MB) would be moved.")
+
+    for subfolder in folder.Folders:
+        sub_emails, sub_size = process_folder(subfolder, archive_folder, run_type, depth + 1)
+        old_email_count += sub_emails
+        total_size_kb += sub_size
+    return old_email_count, total_size_kb
+
+    process_folder(subfolder, archive_folder, run_type, depth + 1)
+
+
+
+if __name__ == "__main__":
+    main_archive()