uninstall.py

#!/usr/bin/env python3
# deploy_it uninstall script
# Copyright 2019, Aswin Babu Karuvally

# import serious stuff
import os
import logging
import json
import sys
import subprocess
import pdb
import shutil


def run_cleanup_script(config):
    if not config["cleanup_script"]["enable"]:
        return

    script_file = config["cleanup_script"]["script_file"]
    if not os.path.isfile(script_file):
        logging.critical(script_file + "does not exist, exiting...")
    execute_command(script_file)


def remove_virtualenv(config):
    install_path = config["basics"]["install_path"]
    shutil.rmtree(install_path)
    logging.info("removed virtualenv at " + install_path)


# execute command
def execute_command(command):
    return_code = subprocess.call(command, shell=True)

    if return_code != 0:
        logging.warning(command + " cannot be executed, exiting")
        sys.exit(1)


# remove symlink
def remove_symlink(config):
    if not config["symlink"]["enable"]:
        return

    link_path = os.path.join(config["symlink"]["link_path"])
    execute_command("rm " + link_path)
    logging.info("removed symlink at " + link_path)


# setup logging
def setup_logging():
    format_string = "[%(asctime)s] %(message)s"
    date_format = "%Y-%m-%d %H:%M:%S"

    logging.basicConfig(
        filename = os.path.join("build.log"),
        level = logging.DEBUG,
        format = format_string,
        datefmt = date_format
    )

    # print logs to stderr
    logging.getLogger().addHandler(logging.StreamHandler())


# stop and remove service
def remove_service(config):
    if not config["systemd_service"]["enable"]:
        return

    unit_file = config["systemd_service"]["unit_file"]
    service_name = os.path.basename(unit_file)
    execute_command("systemctl stop " + os.path.basename(service_name))
    execute_command("systemctl disable " + os.path.basename(service_name))
    execute_command("rm /etc/systemd/system/" + service_name)

    logging.info("removed service")


# the main function
def main():
    # check if user is root
    if os.getuid() != 0:
        print("Script cannot be run as normal user, exiting...")
        sys.exit(1)

    # setup logging
    setup_logging()

    # check if config file exists
    if not os.path.isfile("config.json"):
        logging.critical("config.json does not exist, exiting...")
        sys.exit(1)

    with open("config.json") as config_file:
        config = config_file.read()
        config = json.loads(config)

    remove_symlink(config)
    remove_service(config)
    remove_virtualenv(config)
    run_cleanup_script(config)


# call the main function
if __name__ == "__main__":
    main()