feat: user check script

This commit is contained in:
TING-JUN WANG 2024-04-26 21:22:03 +08:00
parent c441763d39
commit d66a589dfc
3 changed files with 16 additions and 2 deletions

11
check Executable file
View File

@ -0,0 +1,11 @@
#!/usr/bin/env python3
from src.cleaner import Cleaner
from constant import DATE, WHITE_LIST, KEEP_FILES
import os
student_dir = os.environ['HOME']
cleaner = Cleaner(student_dir, DATE, WHITE_LIST, KEEP_FILES)
cleaner.run(test=True, notify_stage=False)

View File

@ -81,7 +81,7 @@ def main(clean_type, test=True, notify_stage=True):
# start clean
for index, student_dir in enumerate(student_list):
print(f'{index+1}/{len(student_list)} {student_dir}')
cleaner = Cleaner(student_dir, "2024-02-01", mailer, WHITE_LIST, KEEP_FILES, MAIL_MESSAGE)
cleaner = Cleaner(student_dir, DATE, WHITE_LIST, KEEP_FILES, mailer, MAIL_MESSAGE)
cleaner.run(test=test, notify_stage=notify_stage)
if __name__ == "__main__":

View File

@ -2,7 +2,7 @@ import os
import subprocess
class Cleaner():
def __init__(self, user_dir, date, mailer, white_list, keep_files, mail_template):
def __init__(self, user_dir, date, white_list, keep_files, mailer=None, mail_template=""):
self.user_dir = user_dir
self.date = date
self.white_list = white_list
@ -56,6 +56,9 @@ class Cleaner():
# subprocess.getoutput(command)
def notify(self, username, test, remove_objs):
if self.mailer == None:
return
if test:
receiver = 'snsd0805@cmlab.csie.ntu.edu.tw'
else: