156 lines
4.9 KiB
Python
156 lines
4.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# 2010-2015 Nico Schottelius (nico-cdist at schottelius.org)
|
|
# 2012-2017 Steven Armstrong (steven-cdist at armstrong.cc)
|
|
#
|
|
# 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 os
|
|
import hashlib
|
|
|
|
import cdist.log
|
|
import cdist.version
|
|
|
|
VERSION = cdist.version.VERSION
|
|
|
|
BANNER = """
|
|
.. . .x+=:. s
|
|
dF @88> z` ^% :8
|
|
'88bu. %8P . <k .88
|
|
. '*88888bu . .@8Ned8" :888ooo
|
|
.udR88N ^"*8888N .@88u .@^%8888" -*8888888
|
|
<888'888k beWE "888L ''888E` x88: `)8b. 8888
|
|
9888 'Y" 888E 888E 888E 8888N=*8888 8888
|
|
9888 888E 888E 888E %8" R88 8888
|
|
9888 888E 888F 888E @8Wou 9% .8888Lu=
|
|
?8888u../ .888N..888 888& .888888P` ^%888*
|
|
"8888P' `"888*"" R888" ` ^"F 'Y"
|
|
"P' "" ""
|
|
"""
|
|
|
|
REMOTE_COPY = "scp -o User=root -q"
|
|
REMOTE_EXEC = "ssh -o User=root"
|
|
REMOTE_CMDS_CLEANUP_PATTERN = "ssh -o User=root -O exit -S {}"
|
|
|
|
|
|
class Error(Exception):
|
|
"""Base exception class for this project"""
|
|
pass
|
|
|
|
|
|
class UnresolvableRequirementsError(cdist.Error):
|
|
"""Resolving requirements failed"""
|
|
pass
|
|
|
|
|
|
class CdistBetaRequired(cdist.Error):
|
|
"""Beta functionality is used but beta is not enabled"""
|
|
|
|
def __init__(self, command, arg=None):
|
|
self.command = command
|
|
self.arg = arg
|
|
|
|
def __str__(self):
|
|
if self.arg is None:
|
|
err_msg = ("\'{}\' command is beta, but beta is "
|
|
"not enabled. If you want to use it please enable beta "
|
|
"functionalities by using the -b/--beta command "
|
|
"line flag or setting CDIST_BETA env var.")
|
|
fmt_args = [self.command, ]
|
|
else:
|
|
err_msg = ("\'{}\' argument of \'{}\' command is beta, but beta "
|
|
"is not enabled. If you want to use it please enable "
|
|
"beta functionalities by using the -b/--beta "
|
|
"command line flag or setting CDIST_BETA env var.")
|
|
fmt_args = [self.arg, self.command, ]
|
|
return err_msg.format(*fmt_args)
|
|
|
|
|
|
class CdistObjectError(Error):
|
|
"""Something went wrong while working on a specific cdist object"""
|
|
def __init__(self, cdist_object, subject=''):
|
|
self.cdist_object = cdist_object
|
|
self.object_name = cdist_object.name.center(len(cdist_object.name)+2)
|
|
if isinstance(subject, Error):
|
|
self.original_error = subject
|
|
else:
|
|
self.original_error = None
|
|
self.message = str(subject)
|
|
self.line_length = 74
|
|
|
|
@property
|
|
def stderr(self):
|
|
output = []
|
|
for stderr_name in os.listdir(self.cdist_object.stderr_path):
|
|
stderr_path = os.path.join(self.cdist_object.stderr_path,
|
|
stderr_name)
|
|
# label = '---- '+ stderr_name +':stderr '
|
|
label = stderr_name + ':stderr '
|
|
if os.path.getsize(stderr_path) > 0:
|
|
# output.append(label)
|
|
# output.append('{0:-^50}'.format(label.center(len(label)+2)))
|
|
output.append('{0:-<{1}}'.format(label, self.line_length))
|
|
with open(stderr_path, 'r') as fd:
|
|
output.append(fd.read())
|
|
return '\n'.join(output)
|
|
|
|
def __str__(self):
|
|
output = []
|
|
output.append(self.message)
|
|
output.append('''{label:-<{length}}
|
|
name: {o.name}
|
|
path: {o.absolute_path}
|
|
source: {o.source}
|
|
type: {o.cdist_type.absolute_path}'''.format(
|
|
label='---- object ',
|
|
length=self.line_length,
|
|
o=self.cdist_object)
|
|
)
|
|
output.append(self.stderr)
|
|
return '\n'.join(output)
|
|
|
|
|
|
def file_to_list(filename):
|
|
"""Return list from \n seperated file"""
|
|
if os.path.isfile(filename):
|
|
file_fd = open(filename, "r")
|
|
lines = file_fd.readlines()
|
|
file_fd.close()
|
|
|
|
# Remove \n from all lines
|
|
lines = map(lambda s: s.strip(), lines)
|
|
else:
|
|
lines = []
|
|
|
|
return lines
|
|
|
|
|
|
def str_hash(s):
|
|
"""Return hash of string s"""
|
|
if isinstance(s, str):
|
|
return hashlib.md5(s.encode('utf-8')).hexdigest()
|
|
else:
|
|
raise Error("Param should be string")
|
|
|
|
|
|
def home_dir():
|
|
if 'HOME' in os.environ:
|
|
return os.path.join(os.environ['HOME'], ".cdist")
|
|
else:
|
|
return None
|