diff options
| -rw-r--r-- | health_monitor.py | 78 | ||||
| -rw-r--r-- | sbhs_server/credentials.py.example | 3 | ||||
| -rw-r--r-- | sbhs_server/settings.py | 5 |
3 files changed, 83 insertions, 3 deletions
diff --git a/health_monitor.py b/health_monitor.py new file mode 100644 index 0000000..aa19cb1 --- /dev/null +++ b/health_monitor.py @@ -0,0 +1,78 @@ +import os, serial, smtplib +from time import sleep +from sbhs_server.settings import online_mids, ADMIN_EMAIL +from sbhs_server.helpers import mailer + +MAX_PORTS = 256 +MIN_TEMP = 10 +MAX_TEMP = 70 + +def write_to_port(s, com, arg): + s.write(chr(int(com))) + sleep(0.5) + s.write(chr(int(arg))) + sleep(0.5) + +def read_from_port(s, com): + s.write(chr(int(com))) + sleep(0.5) + if com == 255: + result = ord(s.read(1)) + (0.1 * ord(s.read(1))) + elif com == 252: + result = ord(s.read(1)) + else: + result = -1 + sleep(0.5) + return result + +def create_message(check_mids, defective_ports): + msg = '' + if len(check_mids) != 0: + msg += 'Please check the following machine ids :\n' + for i,mid in enumerate(check_mids): + msg = msg + str(i) + '. ' + str(mid) + '\n' + if len(defective_ports) != 0: + msg = msg + '\nPlease check the following ports :\n' + for port in defective_ports: + msg = msg + str(port) + '\n' + return msg + +def main(): + present_online_mids, offline_mids, defective_ports = [], [], [] + for i in range(0,MAX_PORTS): + path = '/dev/ttyUSB' + str(i) + if os.path.exists(path): + try: + s = serial.Serial(port=path, baudrate=9600, bytesize=8, parity='N', stopbits=1, timeout=2) + if not s.is_open: + s.open() + assert s.is_open + mid = read_from_port(s, 252) # get machine id + assert mid > 0 + write_to_port(s, 253, 100) # set fan speed + write_to_port(s, 254, 0) # set heater value + current_temp = read_from_port(s, 255) # get current temperature + assert current_temp >= MIN_TEMP and current_temp <= MAX_TEMP + present_online_mids.append(mid) + except: + if s.is_open: + if mid > 0: + offline_mids.append(mid) + else: + defective_ports.append(path) + if s.is_open: + s.close() + check_mids = list(set(online_mids).difference(set(present_online_mids))) + msg = create_message(check_mids, defective_ports) + if len(msg) == 0: + msg = "Nothing out of order was detected on the server. :)" + #thread.start_new_thread(mailer.email, (ADMIN_EMAIL, "Health report of SBHS Server", msg)) + mailer.email(ADMIN_EMAIL, "Health report of SBHS Server", msg) + print msg + +if __name__ == "__main__": + main() + + + + diff --git a/sbhs_server/credentials.py.example b/sbhs_server/credentials.py.example index 9f6ec5f..255bbae 100644 --- a/sbhs_server/credentials.py.example +++ b/sbhs_server/credentials.py.example @@ -2,4 +2,5 @@ PROJECT_SECRET_KEY = 'your django project secret key' EMAIL_HOST_USER = 'username' EMAIL_HOST_PASSWORD = 'password' MAP_FILE = 'path/to/file/map_machine_ids.txt' -LOG_FILE = '/path/to/logfile'
\ No newline at end of file +LOG_FILE = '/path/to/logfile' +ADMIN_EMAIL='admin email'
\ No newline at end of file diff --git a/sbhs_server/settings.py b/sbhs_server/settings.py index b7ec8b7..15dc8ae 100644 --- a/sbhs_server/settings.py +++ b/sbhs_server/settings.py @@ -136,6 +136,7 @@ EMAIL_HOST = 'smtp.gmail.com' #'smtp-auth.iitb.ac.in' EMAIL_PORT = 587 #25 EMAIL_HOST_USER = credentials.EMAIL_HOST_USER EMAIL_HOST_PASSWORD = credentials.EMAIL_HOST_PASSWORD +ADMIN_EMAIL=credentials.ADMIN_EMAIL # Static files (CSS, JavaScript, Images) @@ -251,13 +252,13 @@ with open(os.path.join(BASE_DIR, 'map_machine_ids.txt')) as f: for line in f: try: data = line.split("=") - MID_PORT_MAP[int(data[0])]=data[1].strip() brd = sbhs.Sbhs() b = brd.connect(int(data[0])) assert b == True - key = int(brd.getMachineId()) + key = int(brd.getMachineId()) assert key > 0 brd.reset_board() + MID_PORT_MAP[key]=data[1].strip() boards[str(key)] = {"board": brd, "experiment_id": None} except: pass |