cdist-backup/bin/cdist
Nico Schottelius 7a7413f1f4 pause
Signed-off-by: Nico Schottelius <nico@kr.ethz.ch>
2011-09-08 03:31:58 +02:00

140 lines
4.4 KiB
Python
Executable file

#!/usr/bin/env python3
#
# 2010-2011 Nico Schottelius (nico-cdist at schottelius.org)
#
# This file is part of cdist.
#
# cdist is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# cdist is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with cdist. If not, see <http://www.gnu.org/licenses/>.
#
#
import sys # argv
import subprocess # execute stuff
import os
__cdist_version="2.0.0"
def logger(type, *args):
"""Ignore type for now, support later"""
print(*args)
def exit_error(*args):
logger("error", *args)
sys.exit(1)
def run_or_fail(*args):
# newargs = ["echo"]
newargs = []
newargs.extend(*args)
print(newargs)
try:
subprocess.check_call(newargs)
except subprocess.CalledProcessError:
exit_error("Command failed:", " ".join(newargs))
def remote_run_or_fail(hostname, *args):
"""Run something on the remote side and fail is something breaks"""
newargs = ["ssh", "root@" + hostname]
newargs.extend(*args)
run_or_fail(newargs)
def remove_remote_dir(hostname, destination):
remote_run_or_fail(hostname, ["rm", "-rf", destination])
def transfer_dir(hostname, source, destination):
remove_remote_dir(hostname, destination)
run_or_fail(["scp", "-r", source, "root@" + hostname + ":" + destination])
def base_directory():
"""Returns the directory in which all cdist stuff is based in"""
print("Going to", __file__, os.path.join(os.path.dirname(__file__), os.pardir))
os.chdir(os.path.join(os.path.dirname(__file__), os.pardir))
return os.getcwd()
def remote_base_directory():
return "/var/lib/cdist"
def conf_directory():
"""Returns path to main configuration directory"""
return os.path.join(base_directory(), "conf")
def remote_conf_directory():
"""Returns path to remote main configuration directory"""
return os.path.join(remote_base_directory(), "conf")
def out_dir():
FIXME: stopped - probably need static temp know!
"""Local directory containing output"""
return os.path.join(base_directory(), "conf")
def global_explorer_directory():
"""Returns path to directory containing the global explorers"""
return os.path.join(conf_directory(), "explorer")
def remote_global_explorer_directory():
"""Returns path to the remote directory containing the global explorers"""
return os.path.join(remote_conf_directory(), "explorer")
def remote_global_explorer_path(explorer):
"""Returns path to the remote explorer"""
return os.path.join(remote_global_explorer_directory(), explorer)
def list_global_explorers():
"""Return list of available explorers"""
return os.listdir(global_explorer_directory())
def transfer_global_explorers(hostname):
transfer_dir(hostname, global_explorer_directory(), remote_global_explorer_directory())
def global_explore(hostname):
"""Run global explorers"""
explorers = list_global_explorers()
if(len(explorers) == 0):
exit_error("No explorers found in", global_explorer_directory())
transfer_global_explorers(hostname)
for explorer in explorers:
remote_run_or_fail(hostname, [remote_global_explorer_path(explorer)])
def init_deploy(hostname):
logger("info", "Creating clean directory structure")
# Ensure there is no old stuff, neither local nor remote
# run_or_fail(["rm -rf", "$__cdist_local_base_dir"])
#
# remote_run_or_fail(hostname, ["rm -rf", "${__cdist_remote_base_dir}"])
#
# # Create base directories
# run_or_fail(["mkdir -p", "$__cdist_local_base_dir"])
# remote_run_or_fail(hostname,["mkdir -p", "${__cdist_remote_base_dir}"])
#
# # Link configuraion source directory - consistent with remote
# run_or_fail(["ln -sf", "$__cdist_conf_dir", "$__cdist_local_base_dir/$__cdist_name_conf_dir"])
def cdist_deploy_to(hostname):
"""Mimic the old deploy to: Deploy to one host"""
logger("info", "Deploying to host", hostname)
init_deploy(hostname)
global_explore(hostname)
if __name__ == "__main__":
hostname=sys.argv[1]
logger("info", "cdist", __cdist_version, ": Configuring host", hostname)
cdist_deploy_to(hostname)
print(list_global_explorers())