snsd0805/CMLAB-GPU-Mailer
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: some exception

Browse Source
This commit is contained in:
Ting-Jun Wang 2024-04-18 14:09:49 +08:00
parent 65beca5067
commit 77b51204af
Signed by: snsd0805
GPG Key ID: 48D331A3D6160354
1 changed files with 31 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
main.py
View File

@ -11,7 +11,7 @@ DEBUG_MODE = False
URL = 'https://www.cmlab.csie.ntu.edu.tw/status-gpu/'
GPU_LIMIT = 2
GOOGLE_CODE = os.environ['GOOGLE_CODE']
MAIL_CD_HOUR = 6
MAIL_CD_HOUR = 12
MAIL_MESSAGE = '''
Hi, {}<br>
<br>
@ -76,7 +76,9 @@ def get_server_gpu_status() -> list:
for box in boxes[:-1]:
soup = BeautifulSoup(box, 'html.parser')
# get server name
server_name = soup.find('span', class_='f7').text.replace(' ', '')
server = soup.find('span', class_='f7')
if server:
server_name = server.text.replace(' ', '')
gpu_names = []
users = []
@ -93,7 +95,7 @@ def get_server_gpu_status() -> list:
for user_obj in user_objs:
if user_obj != None:
username = user_obj.text
if username != 'gdm':
if username != 'gdm' and username != 'root':
user_in_this_gpu.add(username)
# get gpu names
@ -197,7 +199,7 @@ def mail_notify(violators: list) -> None:
msg['Subject'] = f'[網管通知] 提醒您已經超過 CMLab GPU 使用限制！({username})'
msg['From'] = 'unix_manager@cmlab.csie.ntu.edu.tw'
msg['To'] = f'{username}@cmlab.csie.ntu.edu.tw' if not DEBUG_MODE else 'snsd0805@cmlab.csie.ntu.edu.tw'
msg['Cc'] = 'unix_manager@cmlab.csie.ntu.edu.tw' if not DEBUG_MODE else 'snsd0805@cmlab.csie.ntu.edu.tw'
# msg['Cc'] = 'unix_manager@cmlab.csie.ntu.edu.tw' if not DEBUG_MODE else 'snsd0805@cmlab.csie.ntu.edu.tw'
smtp = smtplib.SMTP('smtp.gmail.com', 587)
smtp.ehlo()