feat: user check script
This commit is contained in:
parent
c441763d39
commit
d66a589dfc
11
check
Executable file
11
check
Executable 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)
|
||||
|
||||
2
main.py
2
main.py
@ -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__":
|
||||
|
||||
@ -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:
|
||||
|
||||
Loading…
Reference in New Issue
Block a user