95 lines
3.1 KiB
Python
95 lines
3.1 KiB
Python
|
#!/usr/bin/python
|
||
|
|
|
||
|
|
##########
|
||
|
|
#
|
||
|
|
# This script install ansible modules for manage Eltex ESR devices
|
||
|
|
# There are two modules: esr_command and esr_config
|
||
|
|
# For succesfully install you need properly installed Ansible software with version >= 2.7
|
||
|
|
#
|
||
|
|
# Eltex, 2019
|
||
|
|
#
|
||
|
|
##########
|
||
|
|
|
||
|
|
import os
|
||
|
|
import re
|
||
|
|
from distutils.dir_util import copy_tree
|
||
|
|
import shutil
|
||
|
|
|
||
|
|
MODULES_PATH = {
|
||
|
|
"root": "",
|
||
|
|
"doc_fragments": ""
|
||
|
|
}
|
||
|
|
|
||
|
|
def copy_files(files, src_dir, dest_dir):
|
||
|
|
if not os.path.exists(dest_dir):
|
||
|
|
print("Creating directory {}".format(dest_dir))
|
||
|
|
os.mkdir(dest_dir)
|
||
|
|
|
||
|
|
if type(files) is not list:
|
||
|
|
files = [files]
|
||
|
|
|
||
|
|
for file in files:
|
||
|
|
src_file_path = os.path.join(src_dir, file)
|
||
|
|
dest_file_path = os.path.join(dest_dir, file)
|
||
|
|
if os.path.exists(src_file_path):
|
||
|
|
if os.path.exists(dest_file_path):
|
||
|
|
print("Overwriting {}".format(dest_file_path))
|
||
|
|
else:
|
||
|
|
print("Copying to {}".format(dest_file_path))
|
||
|
|
shutil.copy(src_file_path, dest_file_path)
|
||
|
|
|
||
|
|
def check_ans_install():
|
||
|
|
print("Checking Ansible installation...")
|
||
|
|
if not os.system("which ansible"):
|
||
|
|
return True
|
||
|
|
return False
|
||
|
|
|
||
|
|
def get_ans_modules_path():
|
||
|
|
cmd = "ansible --version | grep 'ansible python module location' | awk '{ print $6 }'"
|
||
|
|
print("Checking Ansible modules path...")
|
||
|
|
if os.system(cmd):
|
||
|
|
return False
|
||
|
|
MODULES_PATH["root"] = re.sub(r"^\s+|\n+$", '', os.popen(cmd).read())
|
||
|
|
document_fragments_path_old = os.path.join(MODULES_PATH["root"], "utils", "module_docs_fragments")
|
||
|
|
document_fragments_path_new = os.path.join(MODULES_PATH["root"], "plugins", "doc_fragments")
|
||
|
|
if os.path.exists(document_fragments_path_new):
|
||
|
|
MODULES_PATH["doc_fragments"] = document_fragments_path_new
|
||
|
|
elif os.path.exists(document_fragments_path_old):
|
||
|
|
MODULES_PATH["doc_fragments"] = document_fragments_path_old
|
||
|
|
else:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
|
||
|
|
def check_esr_modules():
|
||
|
|
res = True
|
||
|
|
print("Checking installed modules...")
|
||
|
|
if os.system("ansible-doc -l | grep esr_config"):
|
||
|
|
res = False
|
||
|
|
if os.system("ansible-doc -l | grep esr_command"):
|
||
|
|
res = False
|
||
|
|
return res
|
||
|
|
|
||
|
|
def modules_install():
|
||
|
|
print("Installing modules...")
|
||
|
|
for root, subdirs, files in os.walk("./ansible"):
|
||
|
|
relpath = os.path.relpath(root, "./ansible")
|
||
|
|
copy_files(files, root, os.path.join(MODULES_PATH["root"], relpath))
|
||
|
|
copy_files(os.listdir('./doc_fragments/'), './doc_fragments/', MODULES_PATH["doc_fragments"])
|
||
|
|
|
||
|
|
|
||
|
|
if not check_ans_install():
|
||
|
|
print("Ansible software is not installed. Please install it")
|
||
|
|
raise SystemExit(1)
|
||
|
|
|
||
|
|
if not get_ans_modules_path():
|
||
|
|
print("Your Ansible software installation may be corrupted. Please reinstall it")
|
||
|
|
raise SystemExit(1)
|
||
|
|
|
||
|
|
modules_install()
|
||
|
|
|
||
|
|
if not check_esr_modules():
|
||
|
|
print("Modules are not installed. Please check your Ansible software installation, permissions for your user and retry")
|
||
|
|
raise SystemExit(1)
|
||
|
|
|
||
|
|
print("Congratulations! Modules esr_command and esr_config are properly installed.\nPlease use `ansible-doc <module_name>` command for more information about usage")
|