install.py

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

import os
import sys
import subprocess

class install():
    bin = [
        {
            'origin':'src/bin/ansible-autodoc',
            'dest':'ansible-autodoc-dev'
        },
    ]

    libs =[
        {
            'origin':'src/ansibleautodoc',
            'dest':'ansibleautodoc'
        },
    ]
    extra =[ ]


    def __init__(self):

        self.current_file_dir = os.path.dirname(os.path.abspath(__file__))
        self.dest_libs_dir = '/usr/lib/python3/dist-packages'
        self.dest_bin= '/usr/bin'
        self.user_home = os.environ['HOME']
        self.dry = False

        if len(sys.argv) > 2:
            param = sys.argv[2]

            if param == '--dry':
                self.dry = True
            else:
                self.dry = False

        if len(sys.argv) > 1:
            param = sys.argv[1]

            if param == '--help' or param ==  '-h':
                self.print_help()
            elif param == '--install':
                self.install()
            elif param == '--uninstall':
                self.uninstall()
            else:

                self.print_help()
        else:
            self.print_help()

    def print_help(self):
        print('About: use install.py --<install | uninstall> --[dry]')
        print( 'parameters: '+ str(sys.argv))

    def link_mod(self,file,type,action):
        ori = None
        dest = None

        # make a symlink in bin location
        if type == 'bin':
            if 'lib' in file.keys():
                ori = os.path.join(self.dest_libs_dir,  file['lib'])
            elif 'origin' in file.keys():
                ori = os.path.join(self.current_file_dir,  file['origin'])

            dest = os.path.join(self.dest_bin, file['dest'])

        # make a symlink in python 3 libraries location
        if type == 'lib':
            dest = os.path.join(self.dest_libs_dir, file['dest'])
            ori = os.path.join(self.current_file_dir,  file['origin'])

        # make other hardcoded links
        if type == 'extra':
            dest = file['dest']
            ori = os.path.join(self.current_file_dir,  file['origin'])


        if ori is not  None and dest is not None:

            if action == 'add':
                cmd = 'sudo ln -s '+ ori+ ' '+ dest
            elif action == 'rm':
                cmd = 'sudo rm '+dest

            if self.dry is False:
                print(cmd)
                return_code = subprocess.call(cmd, shell=True)
            else:
                print('dry run: '+ cmd)


    def install(self):
        print('Install')
        for i in self.bin:
            self.link_mod(i,'bin','add')
        for j in self.libs:
            self.link_mod(j,'lib','add')
        for k in self.extra:
            self.link_mod(k,'extra','add')

    def uninstall(self):
        print('Uninstall')
        for i in self.bin:
            self.link_mod(i,'bin','rm')
        for j in self.libs:
            self.link_mod(j,'lib','rm')
        for k in self.extra:
            self.link_mod(k,'extra','rm')

if __name__ == '__main__':
    i = install()