main.py

from __future__ import print_function

import os
import json

from utils.detector import is_pdf, is_ppt, is_docx
from utils.convertor import convert_ppt_to_pdf, convert_docx_to_pdf


class CheckMail: 
    def __init__(self, root_dir=os.getcwd()):
        self.root_dir = root_dir

    def convert_to_pdf(self, file_path, folder_path):
        """This will convert the ppt and docx to pdf"""
        file = file_path.split('/')[-1]
        if is_ppt(file_path):
            pdf_file = f'{folder_path}/{file.split(".")[0]}.pdf'
            convert_ppt_to_pdf(file_path, pdf_file)
        if is_docx(file_path):
            pdf_file = f'{folder_path}/{file.split(".")[0]}.pdf'
            convert_docx_to_pdf(file_path, pdf_file)

    def check_pdf(self, file_path, file):
        """This will check if the file is pdf or not and save in a file with path name. If not then it will remove it."""
        if is_pdf(file_path):
            print(f'{file} is a PDF.')
            if os.path.exists(os.path.join(self.root_dir, 'Pdf_file_paths.json')):
                with open(os.path.join(self.root_dir, 'Pdf_file_paths.json'), 'r') as f:
                    try: 
                        file = json.load(f)
                    except Exception as e:
                        print(f"Error: {e}.")
                with open(os.path.join(self.root_dir, 'Pdf_file_paths.json'), 'w') as f:
                    pdf_file_path_array = file['paths']
                    pdf_file_path_array.append(file_path) if file_path not in pdf_file_path_array else None
                    json.dump({'paths': pdf_file_path_array}, f)

            else:
                with open(os.path.join(self.root_dir, 'Pdf_file_paths.json'), 'w') as f:
                    json.dump({'paths': [file_path]}, f)

        else:
            try:
                os.remove(file_path)
                print(f'{file} is not a PDF. So Removed it.')
            except OSError as e:
                print(f"Error: {e.file} - {e.strerror}.")

    def remove_empty_dir(self, dir):
        """This will check if the directory is empty or not. If empty then it will remove it."""
        try:
            if(os.listdir(dir) == []):
                os.rmdir(dir)
        except Exception as e:
            print(f"Error: {e}.")

    def navigate_folder(self, dir):
        """This will navigate through the folder and check if the file is pdf or not. If not then it will convert it to pdf. Else it will remove it."""
        for folders in os.listdir(dir):
            folder_path = os.path.join(dir, folders)
            print('\n')
            print(f'Checking folder: {folder_path}')
            try: 
                for file in os.listdir(folder_path):
                    file_path = os.path.join(folder_path, file)
                    self.convert_to_pdf(file_path, folder_path)
                    self.check_pdf(file_path, file)

            except NotADirectoryError as e:
                print(f"Error: {e}.")
            self.remove_empty_dir(folder_path)

    def check_dir(self, dir):
        """This will check if the directory is present or not. If not then it will create it."""
        try:
            os.listdir(dir)
        except FileNotFoundError as e:
            os.mkdir(dir)

    def run(self, dir='Attachments'):
        self.check_dir(dir)
        self.navigate_folder(dir)
        
if __name__ == '__main__':
    check_mail = CheckMail()
    check_mail.run()