def connection(self, host, port, user, password, ssl=False, **kwargs):
# must set X-Requested-By in newer versions of Ambari
self.x_requested_by = user
if user == 'admin':
self.x_requested_by = os.getenv('USER', user)
#log.info("contacting Ambari as '%s'" % self.user)
if not isHost(host) or not isPort(port) or not isUser(user) or not password:
raise InvalidOptionException('invalid options passed to AmbariBlueprint()')
proto = 'http' # pylint: disable=unused-variable
if ssl:
proto = 'https'
self.host = host
self.port = port
self.user = user
self.password = password
# if kwargs.has_key('strip_config') and kwargs['strip_config']:
if 'strip_config' in kwargs and kwargs['strip_config']:
self.strip_config = True
self.url_base = '%(proto)s://%(host)s:%(port)s/api/v1' % locals()
if 'dir' in kwargs and kwargs['dir']:
self.blueprint_dir = kwargs['dir']
if not isDirname(self.blueprint_dir):
qquit('UNKNOWN', 'invalid dir arg passed to AmbariBlueprintTool')
try:
if not self.blueprint_dir or not os.path.exists(self.blueprint_dir):
log.info("creating blueprint data dir '%s'" % self.blueprint_dir)
os.mkdir(self.blueprint_dir)
if not os.path.isdir(self.blueprint_dir):
raise IOError("blueprint dir '%s'already taken and is not a directory" % self.blueprint_dir)
except IOError as _:
die("'failed to create dir '%s': %s" % (self.blueprint_dir, _))
def connection(self, host, port, user, password, ssl=False, **kwargs):
# must set X-Requested-By in newer versions of Ambari
self.x_requested_by = user
if user == 'admin':
self.x_requested_by = os.getenv('USER', user)
#log.info("contacting Ambari as '%s'" % self.user)
if not isHost(host) or not isPort(port) or not isUser(user) or not password:
raise InvalidOptionException('invalid options passed to AmbariBlueprint()')
proto = 'http' # pylint: disable=unused-variable
if ssl:
proto = 'https'
self.host = host
self.port = port
self.user = user
self.password = password
# if kwargs.has_key('strip_config') and kwargs['strip_config']:
if 'strip_config' in kwargs and kwargs['strip_config']:
self.strip_config = True
self.url_base = '%(proto)s://%(host)s:%(port)s/api/v1' % locals()
if 'dir' in kwargs and kwargs['dir']:
self.blueprint_dir = kwargs['dir']
if not isDirname(self.blueprint_dir):
qquit('UNKNOWN', 'invalid dir arg passed to AmbariBlueprintTool')
try:
if not self.blueprint_dir or not os.path.exists(self.blueprint_dir):
log.info("creating blueprint data dir '%s'" % self.blueprint_dir)
os.mkdir(self.blueprint_dir)
if not os.path.isdir(self.blueprint_dir):
raise IOError("blueprint dir '%s'already taken and is not a directory" % self.blueprint_dir)
except IOError as _:
die("'failed to create dir '%s': %s" % (self.blueprint_dir, _))
def construct_msg(self):
# user = os.getenv('USER', '').strip()
user = getpass.getuser()
if not isUser(user):
# print("invalid user '%s' determined from environment variable $USER, failed regex validation" % user)
print(
"invalid user '%s' returned by getpass.getuser(), failed regex validation"
% user)
sys.exit(ERRORS['CRITICAL'])
user = self.case_user(user)
msg = 'Welcome %s - ' % user
last = ''
if which("last"):
_ = os.popen('last -100')
_.readline()
re_skip = re.compile(r'^(?:reboot|wtmp)|^\s*$')
last = ''
for line in _:
last = line.rstrip('\n')
if re_skip.match(last):
last = ''
continue
break
_.close()
else:
printerr(
"WARNING: 'last' command not found, will not be able to get last login information"
)
if last:
msg += 'last login was '
last_user = re.sub(r'\s+.*$', '', last)
if last_user == 'root':
last_user = '******'
# strip up to "Day Mon NN" ie "%a %b %e ..."
(last, num_replacements) = re.subn(r'.*(\w{3}\s+\w{3}\s+\d+)',
r'\g<1>', last)
if not num_replacements:
print('failed to find the date format in the last log')
sys.exit(ERRORS['CRITICAL'])
last = re.sub(' *$', '', last)
if last_user == 'ROOT':
msg += 'ROOT'
elif last_user.lower() == user.lower():
msg += 'by you'
else:
msg += 'by %s' % last_user
msg += ' => %s' % last
else:
msg += 'no last login information available!'
return msg
def construct_msg(self): # pylint: disable=no-self-use
# user = os.getenv('USER', '').strip()
user = getpass.getuser()
if not isUser(user):
# print("invalid user '%s' determined from environment variable $USER, failed regex validation" % user)
print("invalid user '%s' returned by getpass.getuser(), failed regex validation" % user)
sys.exit(ERRORS['CRITICAL'])
user = self.case_user(user)
msg = 'Welcome %s - ' % user
last = ''
if which("last"):
_ = os.popen('last -100')
_.readline()
re_skip = re.compile(r'^(?:reboot|wtmp)|^\s*$')
last = ''
for line in _:
last = line.rstrip('\n')
if re_skip.match(last):
last = ''
continue
break
_.close()
else:
printerr("WARNING: 'last' command not found, will not be able to get last login information")
if last:
msg += 'last login was '
last_user = re.sub(r'\s+.*$', '', last)
if last_user == 'root':
last_user = '******'
# strip up to "Day Mon NN" ie "%a %b %e ..."
(last, num_replacements) = re.subn(r'.*(\w{3}\s+\w{3}\s+\d+)', r'\g<1>', last)
if not num_replacements:
print('failed to find the date format in the last log')
sys.exit(ERRORS['CRITICAL'])
last = re.sub(' *$', '', last)
if last_user == 'ROOT':
msg += 'ROOT'
elif last_user.lower() == user.lower():
msg += 'by you'
else:
msg += 'by %s' % last_user
msg += ' => %s' % last
else:
msg += 'no last login information available!'
return msg