def edit_profile(profile, command):
width = 50
if command == 'modify':
if not os.path.isfile(os.path.join(config_path, profile + '.conf')):
print "Profile (%s) does not exist. Try 'agileprofile create' first. Exiting." % (
profile)
sys.exit(1)
if command == 'create':
if os.path.isfile(os.path.join(config_path, profile + '.conf')):
print "Profile (%s) already exists. You must use delete first, or use modify. Exiting." % (
profile)
sys.exit(1)
username = ''
password = ''
egress_protocol = 'http'
egress_port = '80'
egress_hostname = 'global.mt.lldns.net'
egress_basepath = ''
ingest_protocol = 'https'
ingest_port = '443'
ingest_hostname = 'api.agile.lldns.net'
elif command == 'modify':
config.read(os.path.join(config_path, profile + '.conf'))
try:
cfgver = config.get("AgileCLU", "version")
username = config.get("Identity", "username")
password = config.get("Identity", "password")
egress_protocol = config.get("Egress", "protocol")
egress_hostname = config.get("Egress", "hostname")
egress_port = config.get("Egress", "port")
egress_basepath = config.get("Egress", "basepath")
ingest_protocol = config.get("Ingest", "protocol")
ingest_hostname = config.get("Ingest", "hostname")
ingest_port = config.get("Ingest", "port")
except:
print "Profile (%s) is incomplete. %s should be %s. Exiting." % (
profile, cfgver, AgileCLU.AgileCLU.__version__)
sys.exit(1)
ciphered = 0
print command.upper() + " PROFILE: " + profile
try:
ok = 0
while not ok:
username = prompt('Username', username, command)
ok = len(username) >= 2
if not ok: print "Username has to be at least 3 characters."
if command == 'modify':
password2 = password
password = prompt('Password', password, command, True)
if (password != password2):
while (password != password2):
password = prompt('Password', '', command, True)
password2 = prompt("Re-enter password", '', command, True)
else:
ciphered = 1
else:
password = "******"
password2 = "2"
while (password != password2):
password = prompt('Password', password, command, True)
password2 = prompt("Re-enter password", password2, command,
True)
egress_port = "80"
egress_protocol = 'http'
""" Temporarily leave off https, since MTs are not presently configured for https, default http will be used
ok=0
while not ok:
egress_protocol = prompt( 'Egress protocol', egress_protocol, command ).lower()
ok = egress_protocol in ['http','https']
if not ok:
print "Supported protocols are 'http' and 'https'."
egress_protocol = 'http'
"""
ok = 0
while not ok:
egress_hostname = prompt('Egress hostname', egress_hostname,
command)
try:
addrinfo = socket.getaddrinfo(egress_hostname, egress_port)
ok = 1
except:
print "Egress hostname (%s) does not resolve correctly." % (
egress_hostname)
ok = 0
egress_basepath = prompt("Egress base path", egress_basepath, command)
ingest_protocol = 'https'
# ingest_protocol = prompt( 'Ingest protocol', ingest_protocol, command )
ingest_port = '80'
# ingest_port = prompt( 'Ingest port', ingest_port, command )
ok = 0
while not ok:
ingest_hostname = prompt('Ingest hostname', ingest_hostname,
command)
try:
addrinfo = socket.getaddrinfo(ingest_hostname, ingest_port)
ok = 1
except:
print "Ingest hostname (%s) does not resolve correctly." % (
ingest_hostname)
ok = 0
if not ciphered:
cipher = AgileCLU.e_pw_hash(password, username, egress_protocol,
egress_hostname, egress_basepath)
else:
cipher = password
except (KeyboardInterrupt, SystemExit):
print "\nAborting..."
sys.exit(1)
if command == 'modify':
try:
os.unlink(os.path.join(config_path, profile + '.conf'))
except:
print "Failed to remove (%s) profile. Exiting." % (profile)
sys.exit(1)
config.remove_section('AgileCLU')
config.remove_section('Identity')
config.remove_section('Egress')
config.remove_section('Ingest')
config.remove_section('Logging')
config.add_section("AgileCLU")
config.set("AgileCLU", "version", AgileCLU.AgileCLU.__version__)
config.add_section("Identity")
config.set("Identity", "username", username)
config.set("Identity", "password", cipher)
config.add_section("Egress")
config.set("Egress", "protocol", egress_protocol)
config.set("Egress", "port", egress_port)
config.set("Egress", "hostname", egress_hostname)
config.set("Egress", "basepath", egress_basepath)
config.add_section("Ingest")
config.set("Ingest", "port", ingest_port)
config.set("Ingest", "protocol", ingest_protocol)
config.set("Ingest", "hostname", ingest_hostname)
config.add_section("Logging")
config.set("Logging", "enabled", "no")
config.set("Logging", "logfile", "/var/log/agileclu.log")
config.set("Logging", "level", "info")
config.write(open(os.path.join(config_path, profile + '.conf'), "w"))
print "\nProfile (%s) has been" % (profile),
if command == 'modify':
print "updated.",
else:
print "saved.",
print " Exiting.\n"
def edit_profile( profile, command ):
width = 50
if command=='modify':
if not os.path.isfile( os.path.join( config_path, profile+'.conf' ) ):
print "Profile (%s) does not exist. Try 'agileprofile create' first. Exiting." % (profile)
sys.exit(1)
if command=='create':
if os.path.isfile( os.path.join( config_path, profile+'.conf' ) ):
print "Profile (%s) already exists. You must use delete first, or use modify. Exiting." % (profile)
sys.exit(1)
username='' ; password=''
egress_protocol='http' ; egress_port='80' ; egress_hostname='global.mt.lldns.net' ; egress_basepath='' ;
ingest_protocol='https' ; ingest_port='443' ; ingest_hostname='api.agile.lldns.net'
elif command=='modify':
config.read( os.path.join( config_path, profile+'.conf' ) )
try:
cfgver=config.get("AgileCLU", "version" )
username=config.get( "Identity", "username" )
password=config.get( "Identity", "password" )
egress_protocol=config.get( "Egress", "protocol" )
egress_hostname=config.get( "Egress", "hostname" )
egress_port=config.get( "Egress", "port" )
egress_basepath=config.get( "Egress", "basepath" )
ingest_protocol=config.get( "Ingest", "protocol" )
ingest_hostname=config.get( "Ingest", "hostname" )
ingest_port=config.get( "Ingest", "port" )
except:
print "Profile (%s) is incomplete. %s should be %s. Exiting." % (profile, cfgver, AgileCLU.AgileCLU.__version__ )
sys.exit(1)
ciphered = 0
print command.upper()+ " PROFILE: "+profile
try:
ok=0
while not ok:
username = prompt( 'Username', username, command )
ok = len(username)>=2
if not ok: print "Username has to be at least 3 characters."
if command=='modify':
password2 = password
password = prompt( 'Password', password, command, True )
if ( password != password2 ):
while (password != password2 ):
password = prompt( 'Password', '', command, True )
password2 = prompt( "Re-enter password", '', command, True )
else: ciphered=1
else:
password = "******" ; password2 = "2"
while (password != password2 ):
password = prompt( 'Password', password, command, True )
password2 = prompt( "Re-enter password", password2, command, True )
egress_port = "80"
egress_protocol = 'http'
""" Temporarily leave off https, since MTs are not presently configured for https, default http will be used
ok=0
while not ok:
egress_protocol = prompt( 'Egress protocol', egress_protocol, command ).lower()
ok = egress_protocol in ['http','https']
if not ok:
print "Supported protocols are 'http' and 'https'."
egress_protocol = 'http'
"""
ok=0
while not ok:
egress_hostname = prompt( 'Egress hostname', egress_hostname, command )
try:
addrinfo = socket.getaddrinfo( egress_hostname, egress_port )
ok = 1
except:
print "Egress hostname (%s) does not resolve correctly." % (egress_hostname)
ok = 0
egress_basepath = prompt( "Egress base path", egress_basepath, command )
ingest_protocol = 'https'
# ingest_protocol = prompt( 'Ingest protocol', ingest_protocol, command )
ingest_port = '80'
# ingest_port = prompt( 'Ingest port', ingest_port, command )
ok=0
while not ok:
ingest_hostname = prompt( 'Ingest hostname', ingest_hostname, command )
try:
addrinfo = socket.getaddrinfo( ingest_hostname, ingest_port )
ok = 1
except:
print "Ingest hostname (%s) does not resolve correctly." % (ingest_hostname)
ok = 0
if not ciphered:
cipher = AgileCLU.e_pw_hash( password, username, egress_protocol, egress_hostname, egress_basepath )
else:
cipher = password
except (KeyboardInterrupt, SystemExit):
print "\nAborting..."
sys.exit(1)
if command=='modify':
try:
os.unlink( os.path.join( config_path, profile+'.conf' ) )
except:
print "Failed to remove (%s) profile. Exiting." % (profile)
sys.exit(1)
config.remove_section('AgileCLU')
config.remove_section('Identity')
config.remove_section('Egress')
config.remove_section('Ingest')
config.remove_section('Logging')
config.add_section("AgileCLU")
config.set("AgileCLU", "version", AgileCLU.AgileCLU.__version__ )
config.add_section("Identity")
config.set("Identity", "username", username )
config.set("Identity", "password", cipher )
config.add_section("Egress")
config.set("Egress", "protocol", egress_protocol )
config.set("Egress", "port", egress_port )
config.set("Egress", "hostname", egress_hostname )
config.set("Egress", "basepath", egress_basepath )
config.add_section("Ingest")
config.set("Ingest", "port", ingest_port )
config.set("Ingest", "protocol", ingest_protocol )
config.set("Ingest", "hostname", ingest_hostname )
config.add_section("Logging")
config.set("Logging", "enabled", "no" )
config.set("Logging", "logfile", "/var/log/agileclu.log" )
config.set("Logging", "level", "info" )
config.write( open( os.path.join( config_path, profile+'.conf' ) , "w"))
print "\nProfile (%s) has been" % (profile),
if command=='modify':
print "updated.",
else:
print "saved.",
print " Exiting.\n"
def test_e_pw_hash(self):
hash=AgileCLU.e_pw_hash('teststr', 'test', 'test', 'test.example.com', '/')
self.assertEqual(hash, 'jyH0M5b9OyM=')