def __init__(self, app, dbservice):
self.dbservice = dbservice
self.sched = None
self.url_check = URLCheck()
notificator_email = app.config['NOTIFICATOR_EMAIL']
notificator_user = app.config['NOTIFICATOR_USER']
notificator_password = app.config['NOTIFICATOR_PASSWORD']
notificator_mail_server = app.config['NOTIFICATOR_MAIL_SERVER']
self.notificator_email_to = app.config['NOTIFICATOR_EMAIL_TO']
self.mail_sender = MailSender(notificator_email, notificator_user, notificator_password, notificator_mail_server)
class WebCheckScheduler(object):
def __init__(self, app, dbservice):
self.dbservice = dbservice
self.sched = None
self.url_check = URLCheck()
notificator_email = app.config['NOTIFICATOR_EMAIL']
notificator_user = app.config['NOTIFICATOR_USER']
notificator_password = app.config['NOTIFICATOR_PASSWORD']
notificator_mail_server = app.config['NOTIFICATOR_MAIL_SERVER']
self.notificator_email_to = app.config['NOTIFICATOR_EMAIL_TO']
self.mail_sender = MailSender(notificator_email, notificator_user, notificator_password, notificator_mail_server)
def main_job(self):
websites = self.dbservice.query_db('SELECT * FROM sites')
if len(websites) != 0:
for website in websites:
url = website['url']
webid = website['id']
status_code = self.url_check.get_url_status(url)
last_check = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
self.dbservice.execute_db('UPDATE sites SET last_check=?, status_code=? WHERE id=?', [last_check, status_code, webid])
if status_code != '200':
title = "Notification : " + url
message = "Web site is offline..."
self.mail_sender.send_message(self.notificator_email_to, title, message)
self.dbservice.execute_db('UPDATE dbutil SET db_code=? WHERE id=?', [1, 1])
def change_interval(self, interval_seconds):
self.sched.shutdown()
self.start(interval_seconds)
def start(self, interval_seconds):
self.sched = Scheduler()
self.sched.start()
self.sched.add_interval_job(self.main_job, seconds = interval_seconds)
