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)