def do_show_node(self, node):
my_configuration = Configuration(self.nodes)
server = my_configuration.get_system(node)
if server != 1:
my_configuration.show_classes(server)
print
self.store_nodes()
else:
print "System not found: {0}".format(node)
def run(self, argv):
self.banner()
try:
opts, args = getopt.getopt(argv,"hd:s:a",["deploy=","show-node=","agent"])
except getopt.GetoptError:
print "pyisac.py --help"
sys.exit(2)
for opt, arg in opts:
if opt in ("-h", "--help"):
print "pyisac help file"
sys.exit(0)
if opt in ("-d", "--deploy"):
self.deploy_script = arg
self.deployment = True
if opt in ("-s", "--show-node"):
self.show_node = True
self.node = arg
if opt in ("-a", "--agent"):
self.agent_mode = True
self.node = arg
self.config_file = self.check_configuration()
self.load_configuration(self.config_file)
self.load_nodes()
if self.deployment:
print "Deploying Script: {0}".format(self.deploy_script)
self.deploy(self.deploy_script)
if self.show_node:
print "showing node {0}".format(self.node)
self.do_show_node(self.node)
if self.agent_mode:
self.my_configuration = Configuration(self.nodes)
self.my_configuration.load_modules()
self.store_nodes()
self.get_node_facts()
print
print "====Working in agent mode===="
for node in self.nodes:
print
print "Applying configurion to {0}".format(node.servername)
self.do_agent_mode(node)
class Main(object):
node_files = []
nodes = []
my_infrastructure = Infrastructure()
my_core = Core()
deployment = False
deploy_script = ""
show_node = False
verbose = False
agent_mode = False
servers = []
def __init__(self):
sys.path.append('/etc/pyisac/config/modules') #include user modules
sys.path.append('/etc/pyisac/config/') #include user configuration
self.node_config_db = Database()
def banner(self):
print "Python InfraStructure As Code - Open Source"
print "Released under the GPU v2 - http://www.gnu.org/licenses/old-licenses/gpl-2.0.html"
def load_configuration(self, config_file):
with open(config_file) as data_file:
configuration = json.load(data_file)
nodes_path = configuration["nodes_path"]
for root, dirs, files in os.walk(nodes_path):
for file in files:
if file.endswith(".json"):
node_file = os.path.join(root,file)
if self.verbose:
print "Found {0}".format(node_file)
self.node_files.append(node_file)
def check_configuration(self):
if not os.path.isdir("/etc/pyisac/config/"):
sys.exit("ERROR: Configuration Directory not found /etc/pyisac/config/")
if not os.path.isfile("/etc/pyisac/config/config.json"):
sys.exit("ERROR: Configuration File not found /etc/pyisac/config/config.json")
else:
if self.verbose:
print("loading environment configuration from /etc/pyisac/config/config.json")
config_file = "/etc/pyisac/config/config.json"
return config_file
def load_nodes(self):
for node_file in self.node_files:
if self.verbose:
print "Loading node configuration from {0}".format(node_file)
self.nodes = self.my_infrastructure.import_json(node_file)
if self.verbose:
print "Found {0} Nodes".format(len(self.nodes))
def store_nodes(self):
for node in self.nodes:
cursor = self.node_config_db.nodefacts_collection.find({"node": node.servername})
if cursor.count() == 0:
result = self.node_config_db.nodefacts_collection.insert_one(node.to_json())
node.ref_id = result.inserted_id
else:
for document in cursor:
node.ref_id = document["_id"]
print "Database synchronized..."
def get_node_facts(self):
print "Updating node facts..."
for node in self.nodes:
node.facts = self.my_core.get_node_facts(node.servername, node.username)
result = self.node_config_db.nodefacts_collection.update_one(
{ "node": node.servername},
{"$set": {"facts": node.facts}
})
print "Updated"
def run(self, argv):
self.banner()
try:
opts, args = getopt.getopt(argv,"hd:s:a",["deploy=","show-node=","agent"])
except getopt.GetoptError:
print "pyisac.py --help"
sys.exit(2)
for opt, arg in opts:
if opt in ("-h", "--help"):
print "pyisac help file"
sys.exit(0)
if opt in ("-d", "--deploy"):
self.deploy_script = arg
self.deployment = True
if opt in ("-s", "--show-node"):
self.show_node = True
self.node = arg
if opt in ("-a", "--agent"):
self.agent_mode = True
self.node = arg
self.config_file = self.check_configuration()
self.load_configuration(self.config_file)
self.load_nodes()
if self.deployment:
print "Deploying Script: {0}".format(self.deploy_script)
self.deploy(self.deploy_script)
if self.show_node:
print "showing node {0}".format(self.node)
self.do_show_node(self.node)
if self.agent_mode:
self.my_configuration = Configuration(self.nodes)
self.my_configuration.load_modules()
self.store_nodes()
self.get_node_facts()
print
print "====Working in agent mode===="
for node in self.nodes:
print
print "Applying configurion to {0}".format(node.servername)
self.do_agent_mode(node)
def deploy(self, script):
if not os.path.isfile(script):
print "Script Not found: {0}".format(script)
else:
print ""
execfile(script)
def do_show_node(self, node):
my_configuration = Configuration(self.nodes)
server = my_configuration.get_system(node)
if server != 1:
my_configuration.show_classes(server)
print
self.store_nodes()
else:
print "System not found: {0}".format(node)
def do_agent_mode(self, node):
self.pyisac_modules = {}
modules = self.my_configuration.modules
for module in modules:
if module.__name__ in node.classes:
klass = getattr(module, module.__name__.title())
self.pyisac_modules[module.__name__] = klass(node)
