'''

Clean the direction in CML Server (tmp2 or home directory)


Suppose that the CLEAN_PATH may seems like this.
And we will only check the directories & files under the user directory with "depth=1"

CLEAN_PATH
    |
    +---- user1 (user directory)
    |       |
    |       +--- file1 (depth=1, will be check)
    |       |
    |       +--- Project1 (depth=1, will be check)
    |       |       |
    |       |       +--- subset1-1 (depth=2, won't be check)
    |       |
    |       +--- Project2 (depth=1)
    |               |
    |               +--- subset2-1 (depth=2)
    |               |
    |               +--- subset2-2 (depth=2)
    |
    +---- user2 (user directory)
            |
            +--- Project1 (depth=1)
                    |
                    +--- subset1-1 (depth=2)

'''

from src.mailer import Mailer
from src.cleaner import Cleaner
from src.users import *
import os
from constant import GOOGLE_OWNER

GOOGLE_CODE = os.environ['GOOGLE_CODE']


def main(clean_type, test=True, notify_stage=True):
    mailer = Mailer(GOOGLE_OWNER, GOOGLE_CODE)

    if clean_type == 'home':
        student_list = []
        # find all student's path
        for student_type in ['master', 'phd', 'ra', 'extra']:
            func = globals()[f'get_{student_type}']
            student_list += func()

        # start clean
        for index, student_dir in enumerate(student_list):
            print(f'{index+1}/{len(student_list)} {student_dir}')
            cleaner = Cleaner(student_dir, mailer)
            cleaner.run(test=test, notify_stage=notify_stage)

if __name__ == "__main__":
    main('home', test=True, notify_stage=True)