slpkg/clean.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-


import os
import shutil


class Clean:
    """Clean all data like man page, log files, PACKAGES.TXT and
    configuration files. This is useful if "slpkg" installed via
    "pip" because pip uninstalls only Python packages and script
    and not data. So if uninstall with "# pip uninstall slpkg" after
    run "# python clean.py" to remove all data and configuration files.
    NOTE: Run this script as root."""
    def __init__(self):
        self.files = [
            "/usr/man/man8/slpkg.8.gz",
            "/etc/bash_completion.d/slpkg.bash-completion",
            "/etc/fish/completions/slpkg.fish"
        ]

        self.dirs = [
            "/etc/slpkg/",
            "/var/log/slpkg/",
            "/var/lib/slpkg/",
            "/tmp/slpkg/"
        ]

    def start(self):
        for f in self.files:

            if os.path.isfile(f):
                print(f"Remove file --> {f}")
                os.remove(f)

        for d in self.dirs:

            if os.path.exists(d):
                print(f"Remove directory --> {d}")
                shutil.rmtree(d)


if __name__ == "__main__":
    clean = Clean()
    clean.start()