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
|
# start clean
|
||||||
for index, student_dir in enumerate(student_list):
|
for index, student_dir in enumerate(student_list):
|
||||||
print(f'{index+1}/{len(student_list)} {student_dir}')
|
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)
|
cleaner.run(test=test, notify_stage=notify_stage)
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|||||||
@ -2,7 +2,7 @@ import os
|
|||||||
import subprocess
|
import subprocess
|
||||||
|
|
||||||
class Cleaner():
|
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.user_dir = user_dir
|
||||||
self.date = date
|
self.date = date
|
||||||
self.white_list = white_list
|
self.white_list = white_list
|
||||||
@ -56,6 +56,9 @@ class Cleaner():
|
|||||||
# subprocess.getoutput(command)
|
# subprocess.getoutput(command)
|
||||||
|
|
||||||
def notify(self, username, test, remove_objs):
|
def notify(self, username, test, remove_objs):
|
||||||
|
if self.mailer == None:
|
||||||
|
return
|
||||||
|
|
||||||
if test:
|
if test:
|
||||||
receiver = 'snsd0805@cmlab.csie.ntu.edu.tw'
|
receiver = 'snsd0805@cmlab.csie.ntu.edu.tw'
|
||||||
else:
|
else:
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user