monitor.py

#!/usr/bin/python3
# ==================================================================
# Python Tutorial: Write a Script to Monitor a Website,
# Send Alert Emails, and Reboot Servers
# ------------------------------------------------------------------
# From: https://www.youtube.com/watch?v=yqm6MBt-yfY
#       Corey Schafer, Published on Mar 4, 2019
# ==================================================================
# --------------- make an environment to run the minitor in
# mkdir  site_monitor
# cd site_monitor
# python3 -m venv monitor_env
# source monitor_env/bin/activate
# --------------- install rerquired packages
# pip3 install requests
# pip3 install linode_api4
# ==================================================================

import os
import smtplib
import requests
# import logging
from linode_api4 import LinodeClient, Instance

EMAIL_ADDRESS = os.environ.get('EMAIL_USER')
EMAIL_PASSWORD = os.environ.get('EMAIL_PASS')
LINODE_TOKEN = os.environ.get('LINODE_TOKEN')

# logging.basicConfig(filename='PATH_TO_DESIRED_LOG_FILE',
#                     level=logging.INFO,
#                     format='%(asctime)s:%(levelname)s:%(message)s')

def notify_user():
    with smtplib.SMTP('smtp.gmail.com', 587) as smtp:
        smtp.ehlo()
        smtp.starttls()
        smtp.ehlo()

        smtp.login(EMAIL_ADDRESS, EMAIL_PASSWORD)

        subject = 'YOUR SITE IS DOWN!'
        body = 'Make sure the server restarted and it is back up'
        msg = f'Subject: {subject}\n\n{body}'

        # logging.info('Sending Email...')
        smtp.sendmail(EMAIL_ADDRESS, 'INSERT_RECEIVER_ADDRESS', msg)


def reboot_server():
    client = LinodeClient(LINODE_TOKEN)
    my_server = client.load(Instance, 376715)
    my_server.reboot()
    # logging.info('Attempting to reboot server...')


try:
    r = requests.get('https://example.com', timeout=5)

    if r.status_code != 200:
        # logging.info('Website is DOWN!')
        notify_user()
        reboot_server()
    else:
        # logging.info('Website is UP')
except Exception as e:
    # logging.info('Website is DOWN!')
    notify_user()
reboot_server()