auto_upgrade.py

#!/usr/bin/python3

# Import some functions
import smtplib
import sys
import time
from distutils.spawn import find_executable
from email.mime.text import MIMEText
from socket import gethostname
from subprocess import run, PIPE

# Import the variables from settings.py
from ./settings.py import *

# Needs to be a TLS protected SMTP server
SMTP_SERVER = "smtp.gmail.com"
SMTP_PORT = 587
SENDER_EMAIL = sender
SENDER_PASSWORD = senderPassword
RECEIVER_EMAIL = recipient
SUBJECT_TEMPLATE = "Updates available on {hostname}"
MSG_TEMPLATE = """
At {time} on {hostname} there are new updates that should be installed:

{updates}

Please login to {hostname} and run:
\t# pacman -Syu
To upgrade the system.
"""

def add_prefix(prefix, text):
    result = ""
    for line in text.splitlines():
        result += prefix + line + "\n"
    return result

def do_upgrade(timeout=timeout):
    packages = ""
    try:
        run(["pacman", "-Syu", "--noconfirm"], timeout=timeout, universal_newlines=True)
    except TimeoutExpired:
        result = run(["checkupdates"], stdout=PIPE, universal_newlines=True)
        packages += "\nOfficial repositories:\n"
        packages += add_prefix("\t:: ", result.stdout)

    if find_executable("cower"):
        result = run(["cower", "--update", "--color=never"], stdout=PIPE, universal_newlines=True)
        if result.stdout:
            packages += "\nAUR:\n"
            # cower already adds "::" as a prefix
            packages += add_prefix("\t", result.stdout)

    return packages


def do_update_list(timeout=timeout):
    # Clear packages variable
    packages = ""

    # Refresh pacman database and check for updates
    run(["pacman", "-Syy", "--noconfirm"], timeout=timeout, universal_newlines=True)
    result = run(["checkupdates"], stdout=PIPE, universal_newlines=True)
    packages += "\nOfficial repositories:\n"
    packages += add_prefix("\t:: ", result.stdout)

    # Check for aur updates
    if find_executable("pacaur"):
        result = run(["pacaur", "-k"], stdout=PIPE, universal_newlines=True)
        if result.stdout:
            packages += "\npacAUR:\n"
            packages += add_prefix("\t:: ", result.stdout)
    else:
        packages += "\nMissing Package pacAUR\n"

    return packages


def send_email(receiver, sender, password, subject, message):
    # Connect to SMTP server
    server = smtplib.SMTP(SMTP_SERVER, SMTP_PORT)
    server.ehlo()
    server.starttls()
    server.login(sender, password)

    msg = MIMEText(message)
    msg['Subject'] = subject
    msg['From'] = sender
    msg['To'] = receiver

    server.send_message(msg)
    server.quit()

def main():
    available_updates = do_update_list()
    if available_updates:
        print("Updates available, sending e-mail to {}".format(RECEIVER_EMAIL))
        hostname = gethostname()
        send_email(RECEIVER_EMAIL, SENDER_EMAIL, SENDER_PASSWORD,
                   SUBJECT_TEMPLATE.format(hostname=hostname),
                   MSG_TEMPLATE.format(time=time.strftime("%c"),
                                       hostname=hostname,
                                       updates=available_updates)
                   )
    else:
        print("No available package to upgrage, not sending e-mail")

    sys.exit()

if __name__ == "__main__":
    main()