def setup_init():
Working_dir = os.path.abspath('..')
# Working_dir = tri_config.Working_dir
if len(tri_config.MySQL_Pass) == 0:
import_db = "mysql -u'%s' <TriAquae.sql" % tri_config.MySQL_User
else:
import_db = "mysql -u'%s' -p'%s' <TriAquae.sql" % (
tri_config.MySQL_User, tri_config.MySQL_Pass)
#print import_db
if os.system(import_db) == 0:
asset_collect_user = tri_config.Asset_collect_user
asset_user_password = random_pass.main()
''' modify asset management script username and password by DengLei '''
f = open('tri_config.py', 'r')
content = f.read()
line1 = re.sub("Asset_collect_user.*",
"Asset_collect_user = '******'" % asset_collect_user,
content)
line2 = re.sub("Asset_user_password.*",
"Asset_user_password = '******'" % asset_user_password,
line1)
f = open('tri_config.py', 'w')
f.write(line2)
f.flush()
f.close()
f = open('Tri_Scrwer.py', 'r')
content = f.read()
line1 = re.sub("triaquae_user=.*",
"triaquae_user='******'" % asset_collect_user, content)
line2 = re.sub("old_passwd=.*",
"old_passwd='%s'" % asset_user_password, line1)
f = open('Tri_Scrwer.py', 'w')
f.write(line2)
f.flush()
f.close()
''' Create tri_connector user '''
tri_connector_username = tri_config.Tri_connector_username
tri_connector_password = tri_config.Tri_connector_password
tri_connector_scripts = tri_config.Tri_connector_baoleiuser
if os.system('id %s >/dev/null 2>&1' % tri_connector_username) != 0:
if sys_version == "ubuntu":
os.system('useradd -r -m -s /bin/bash %s' %
(tri_connector_username))
os.system('echo %s:%s |chpasswd' %
(tri_connector_username, tri_connector_password))
#configure .profile
f = open('/home/%s/.profile' % tri_connector_username, 'a')
f.write('python %s\nlogout\n' % tri_connector_scripts)
f.flush()
f.close()
elif sys_version == "redhat":
os.system('useradd %s' % (tri_connector_username))
os.system('echo %s |passwd %s --stdin >/dev/null 2>&1' %
(tri_connector_password, tri_connector_username))
#configure .profile
f = open('/home/%s/.bash_profile' % tri_connector_username,
'a')
f.write('python %s\nlogout\n' % tri_connector_scripts)
f.flush()
f.close()
''' Create all related dirs '''
os.system('python tri_config.py --initial')
''' Timing to perform asset management to collect '''
if sys_version == "ubuntu":
cron_path = "/var/spool/cron/crontabs/root"
elif sys_version == "redhat":
cron_path = "/var/spool/cron/root"
cmd = '''
\n# TriAquae Automatically collect hardware information
\n00 00 * * * /usr/bin/python %s/TriAquae/backend/Hardware_Multiprocess_Run_Collect_And_write.py\n''' % Working_dir
f = open(cron_path, 'a')
f.write(cmd)
f.close()
print "\033[33;1mInitComplete!\033[0m"
else:
print '\033[;31mPlease fix above database configuration error and rerun this step!\033[0m'
sys.exit(1)
def setup_init():
Working_dir = tri_config.Working_dir
MySQL_Name = tri_config.MySQL_Name
MySQL_User = tri_config.MySQL_User
MySQL_Pass = tri_config.MySQL_Pass
Tri_IP = tri_config.Tri_IP
if len(tri_config.MySQL_Pass) == 0:
import_db = "mysql -u'%s' <TriAquae.sql" % tri_config.MySQL_User
else:
import_db = "mysql -u'%s' -p'%s' <TriAquae.sql" % (tri_config.MySQL_User,tri_config.MySQL_Pass)
#print import_db
if os.system(import_db) == 0:
asset_collect_user = tri_config.Asset_collect_user
asset_user_password = random_pass.main()
''' modify asset management script username and password by DengLei '''
f = open('tri_config.py','r')
content = f.read()
line1 = re.sub("Asset_collect_user.*","Asset_collect_user = '******'" % asset_collect_user,content)
line2 = re.sub("Asset_user_password.*","Asset_user_password = '******'" % asset_user_password,line1)
f = open('tri_config.py','w')
f.write(line2)
f.flush()
f.close()
f = open('Tri_Scrwer.py','r')
content = f.read()
line1 = re.sub("triaquae_user=.*","triaquae_user='******'" % asset_collect_user,content)
line2 = re.sub("old_passwd=.*","old_passwd='%s'" % asset_user_password,line1)
f = open('Tri_Scrwer.py','w')
f.write(line2)
f.flush()
f.close()
#for line in fileinput.input('Tri_Scrwer.py',inplace=1):
# content = line.replace("triaquae_user='******'","triaquae_user='******'" % asset_collect_user).replace("old_passwd='123456'","old_passwd='%s'" % asset_user_password)
# print content,
''' Copy the modified files '''
time.sleep(2)
import shutil
src1 = "tri_config.py"
dst1 = "%s/TriAquae/backend/tri_config.py" % Working_dir
shutil.copyfile(src1, dst1)
src2 = "Tri_Scrwer.py"
dst2 = "%s/TriAquae/backend/Tri_Scrwer.py" % Working_dir
shutil.copyfile(src2, dst2)
''' Create tri_connector user '''
tri_connector_username = tri_config.Tri_connector_username
tri_connector_password = tri_config.Tri_connector_password
tri_connector_scripts = tri_config.Tri_connector_baoleiuser
if os.system('id %s >/dev/null 2>&1' % tri_connector_username) != 0:
if sys_version == "ubuntu":
os.system('useradd -r -m -s /bin/bash %s' % (tri_connector_username))
os.system('echo %s:%s |chpasswd' % (tri_connector_username,tri_connector_password))
#configure .profile
f = open('/home/%s/.profile' % tri_connector_username,'a')
f.write('python %s\nlogout\n' % tri_connector_scripts)
f.flush()
f.close()
elif sys_version == "redhat":
os.system('useradd %s' % (tri_connector_username))
os.system('echo %s |passwd %s --stdin >/dev/null 2>&1' % (tri_connector_password,tri_connector_username))
#configure .profile
f = open('/home/%s/.bash_profile' % tri_connector_username,'a')
f.write('python %s\nlogout\n' % tri_connector_scripts)
f.flush()
f.close()
''' Create all related dirs '''
os.system('python tri_config.py --initial')
''' Timing to perform asset management to collect '''
if sys_version == "ubuntu":
cron_path = "/var/spool/cron/crontabs/root"
elif sys_version == "redhat":
cron_path = "/var/spool/cron/root"
cmd = '''
\n# TriAquae Automatically collect hardware information
\n00 00 * * * /usr/bin/python %s/TriAquae/backend/Hardware_Multiprocess_Run_Collect_And_write.py\n''' % Working_dir
f = open(cron_path,'a')
f.write(cmd)
f.close()
print "\033[33;1mInitComplete!\033[0m"
else:
print '\033[;31mPlease fix above database configuration error and rerun this step!\033[0m'
sys.exit(1)
def setup_init():
Working_dir = tri_config.Working_dir
MySQL_Name = tri_config.MySQL_Name
MySQL_User = tri_config.MySQL_User
MySQL_Pass = tri_config.MySQL_Pass
Tri_IP = tri_config.Tri_IP
if len(tri_config.MySQL_Pass) == 0:
import_db = "mysql -u'%s' <TriAquae.sql" % tri_config.MySQL_User
else:
import_db = "mysql -u'%s' -p'%s' <TriAquae.sql" % (tri_config.MySQL_User,tri_config.MySQL_Pass)
#print import_db
if os.system(import_db) == 0:
''' generate "Tri_connector_username" and "Asset_collect_user" random user passwd to tri_config and db(TriConfig) '''
tri_connector_password = random_pass.main()
asset_collect_user = tri_config.Asset_collect_user
asset_user_password = random_pass.main()
#print 'tri_connector_password',tri_connector_password
#print 'asset_user_password',asset_user_password
modify_tri_connector_pass = '''sed -i "s#^Tri_connector_password = .*#Tri_connector_password = '******'#g" %s/install/%s''' % (tri_connector_password,Working_dir,config_file)
modify_asset_pass = '''sed -i "s#^Asset_user_password = .*#Asset_user_password = '******'#g" %s/install/%s''' % (asset_user_password,Working_dir,config_file)
#modify asset management script username and password by DengLei
modify_username = '''sed -i "s#^triaquae_user=.*#triaquae_user='******'#g" %s/TriAquae/backend/Tri_Scrwer.py''' % (asset_collect_user,Working_dir)
modify_password = '''sed -i "s#^old_passwd=.*#old_passwd='%s'#g" %s/TriAquae/backend/Tri_Scrwer.py''' % (asset_user_password,Working_dir)
#modify install directory db connector info
modify_MySQL_Name = '''sed -i "s#^MySQL_Name = .*#MySQL_Name = '%s'#g" %s/install/tri_config.py''' % (MySQL_Name,Working_dir)
modify_MySQL_User = '''sed -i "s#^MySQL_User = .*#MySQL_User = '******'#g" %s/install/tri_config.py''' % (MySQL_User,Working_dir)
modify_MySQL_Pass = '''sed -i "s#^MySQL_Pass = .*#MySQL_Pass = '******'#g" %s/install/tri_config.py''' % (MySQL_Pass,Working_dir)
modify_Tri_IP = '''sed -i "s#^Tri_IP = .*#Tri_IP = '%s'#g" %s/install/tri_config.py''' % (Tri_IP,Working_dir)
os.system(modify_tri_connector_pass)
os.system(modify_asset_pass)
os.system(modify_username)
os.system(modify_password)
os.system(modify_MySQL_Name)
os.system(modify_MySQL_User)
os.system(modify_MySQL_Pass)
os.system(modify_Tri_IP)
time.sleep(2)
''' TriAquae configure tri_connector user '''
tri_connector_username = tri_config.Tri_connector_username
tri_connector_password = tri_connector_password
tri_connector_scripts = tri_config.Tri_connector_baoleiuser
if os.system('id %s >/dev/null 2>&1' % tri_connector_username) != 0:
if sys_version == "ubuntu":
os.system('useradd -r -m -s /bin/bash %s' % (tri_connector_username))
os.system('echo %s:%s |chpasswd' % (tri_connector_username,tri_connector_password))
#configure .profile
f = open('/home/%s/.profile' % tri_connector_username,'a')
f.write('python %s\nlogout\n' % tri_connector_scripts)
f.flush()
f.close()
elif sys_version == "redhat":
os.system('useradd %s' % (tri_connector_username))
os.system('echo %s |passwd %s --stdin >/dev/null 2>&1' % (tri_connector_password,tri_connector_username))
#configure .profile
f = open('/home/%s/.bash_profile' % tri_connector_username,'a')
f.write('python %s\nlogout\n' % tri_connector_scripts)
f.flush()
f.close()
''' Create all related dirs '''
os.system('python tri_config.py --initial')
''' Timing to perform asset management to collect '''
if sys_version == "ubuntu":
cron_path = "/var/spool/cron/crontabs/root"
elif sys_version == "redhat":
cron_path = "/var/spool/cron/root"
cmd = '''
\n# TriAquae Automatically collect hardware information
\n00 00 * * * /usr/bin/python %s/TriAquae/backend/Hardware_Multiprocess_Run_Collect_And_write.py\n''' % Working_dir
f = open(cron_path,'a')
f.write(cmd)
f.close()
'''Customize django_admin'''
import django_admin_bootstrapped
os.system('/bin/cp -rp admin %s/templates/' % ''.join(django_admin_bootstrapped.__path__))
else:
print '\033[;31mYour database configuration errors\033[0m'
sys.exit(1)
