Merge pull request #8 from gabrielfava/banch_jp

geração de arquivo de times

Modulos/Times/generate.py

@@ -1,49 +1,65 @@
 import string
 import random
-import mysql.connector
+from mysql.connector import MySQLConnection, Error
 import database_conf as cfg
 import team
 
+__select_ALL = "SELECT team_number, team_site, team_name, password, team_short_name, enabled, type, multi_login, team_full_name FROM TEAMS"
+
 
 def select_teams(cursor):
-	query = ("SELECT team_number, team_name from TEAMS")
+	query = ("SELECT team_number, team_name, password from TEAMS")
 	cursor.execute(query)
 	return cursor
 
 def update_passwords(cnx, cursor):
-	for(team_number, team_name) in cursor:
-		new_pass = ''.join(random.SystemRandom().choice(string.ascii_uppercase + string.digits) for _ in range(6))
-		cursor.execute = ("""
-			UPDATE TEAMS
-			SET password=%s
-			WHERE team_number=%s
-		""", (new_pass, team_number))
-		print(team_number, team_name, new_pass)
+	for(team_number, team_name, password) in cursor:
+		if(not password):
+			new_pass = ''.join(random.SystemRandom().choice(string.ascii_uppercase + string.digits) for _ in range(6))
+			query = """UPDATE TEAMS
+					SET password=%s
+					WHERE team_number=%s""" 
+			data = (new_pass, team_number)
+
+			try:
+				cursor.execute(query,data)
+			except Error as error:
+				print(error)
+
 	cnx.commit()
 
 
+def generate_file(cursor):
+	cursor.execute(__select_ALL)
+	f = open("import.txt", 'w')
+	f.write("[user]")
+	first_team = True
+	for(team_number, team_site, team_name, password, team_short_name, enabled, type, multi_login, team_full_name) in cursor:
+		if(first_team):
+			f.write('\n')
+			first_team = False
+		else:
+			f.write('\n\n')
+		f.write('usernumber=' + str(team_number))	
+		f.write('\nusersitenumber=' + team_site)
+		f.write('\nusername=' + team_short_name)
+		f.write('\nusertype=' + type)
+		f.write('\nuserenabled=' + enabled)
+		f.write('\nusermultilogin=' + multi_login)
+		f.write('\nuserfullname=' + team_full_name)
+		f.write('\nuserdesc=' + team_full_name)
+		f.write('\nuserpassword=' + str(password))
+	f.close()
+
 
 def main():
-	cnx = mysql.connector.connect(**cfg.mysql)
+	cnx = MySQLConnection(**cfg.mysql)
 	cursor = cnx.cursor()
 	select_teams(cursor)
 	update_passwords(cnx, cursor)
+	generate_file(cursor)
 	cursor.close()
 	cnx.close()
 
 if __name__ == "__main__":
-	main()
-
-
-
-
-# [user]
-# usernumber=900
-# usersitenumber=1
-# username=time1
-# usertype=team
-# userenabled=t
-# usermultilogin=f
-# userfullname=[TIME 1]ASAP
-# userdesc=[TIME 1]ASAP
-# userpassword=20194143
+	main()

Modulos/Times/import.txt

@@ -0,0 +1,30 @@
+[user]
+usernumber=900
+usersitenumber=1
+username=team1
+usertype=team
+userenabled=t
+usermultilogin=f
+userfullname=[Team 1] ASAP
+userdesc=[Team 1] ASAP
+userpassword=5EUQDV
+
+usernumber=901
+usersitenumber=1
+username=team2
+usertype=team
+userenabled=t
+usermultilogin=f
+userfullname=[Team 2] Sudocode
+userdesc=[Team 2] Sudocode
+userpassword=ZHODQV
+
+usernumber=902
+usersitenumber=1
+username=team3
+usertype=team
+userenabled=t
+usermultilogin=f
+userfullname=[Team 3] JAVAIScript 
+userdesc=[Team 3] JAVAIScript 
+userpassword=FV5Y78