112 lines
3.4 KiB
Python
112 lines
3.4 KiB
Python
'''
|
|
|
|
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)
|
|
|
|
'''
|
|
import os
|
|
import subprocess
|
|
import smtplib
|
|
from email.mime.text import MIMEText
|
|
|
|
CLEAN_PATH = "/home/master/12"
|
|
KEEP_FILES = [".bashrc", ".bash_logout", ".bash_profile", "public-html", ".ssh", ".profile", "examples.desktop", ".bash_history", ".config", ".zshrc", ".vimrc", ".gnupg", ".oh-my-zsh", ".vim", ".viminfo", ".vscode-server", ".gitconfig", ".local"]
|
|
WHITE_LIST = ["mysql", "unix_manager", "unixmanager215", "gangxing", "how083012", "yulunliu", "amie12349", "minliwu", "chenyuyang"]
|
|
|
|
DATE = '2024-02-01'
|
|
|
|
GOOGLE_OWNER = "snsd0805@cmlab.csie.ntu.edu.tw"
|
|
GOOGLE_CODE = os.environ['GOOGLE_CODE']
|
|
|
|
|
|
class Mailer():
|
|
def __init__(self, google_owner, google_code):
|
|
self.google_owner = google_owner
|
|
self.google_code = google_code
|
|
|
|
def send(self, sender, receiver, subject, msg):
|
|
msg = MIMEText(msg, 'html')
|
|
msg['Subject'] = subject
|
|
msg['From'] = sender
|
|
msg['To'] = receiver
|
|
|
|
smtp = smtplib.SMTP('smtp.gmail.com', 587)
|
|
smtp.ehlo()
|
|
smtp.starttls()
|
|
smtp.login(self.google_owner, self.google_code)
|
|
|
|
status = smtp.send_message(msg)
|
|
smtp.quit()
|
|
|
|
if status == {}:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
class Cleaner():
|
|
def __init__(self, root, date, mailer, white_list=WHITE_LIST, keep_files=KEEP_FILES):
|
|
self.root = root
|
|
self.date = date
|
|
self.white_list = white_list
|
|
self.keep_files = keep_files
|
|
self.mailer = mailer
|
|
|
|
def run(self, test=True, send_mail=False):
|
|
for user in sorted(os.listdir(self.root)):
|
|
print(user)
|
|
if user in self.white_list:
|
|
continue
|
|
|
|
for obj in sorted(os.listdir(f'{self.root}/{user}')):
|
|
if obj in self.keep_files:
|
|
continue
|
|
|
|
obj = obj.replace(' ', '\ ')
|
|
|
|
# stat {file or dir}:
|
|
# get Create, modified information
|
|
# grep Modify:
|
|
# get Modified time
|
|
# awk:
|
|
# -F ' ': use space to split the data
|
|
# print $2: column 2 (get date)
|
|
modify_date = subprocess.getoutput(f"stat {self.root}/{user}/{obj} | grep Modify | awk -F ' ' '{{print $2}}'" )
|
|
|
|
needRemove = modify_date < DATE
|
|
print(f' {obj}')
|
|
print(f' {modify_date} {needRemove}')
|
|
|
|
|
|
|
|
|
|
|
|
mailer = Mailer(GOOGLE_OWNER, GOOGLE_CODE)
|
|
mailer.send('unix_manager@cmlab.csie.ntu.edu.tw', 'snsd0805@cmlab.csie.ntu.edu.tw', '[測試郵件] test', 'Test mail')
|
|
cleaner = Cleaner('/home/master/12', "2024/02/01", mailer)
|
|
cleaner.run()
|