ndacremont/lego
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: data gen almost done.

Browse Source
This commit is contained in:
damien DELPY 2024-12-04 18:26:52 +01:00
parent 052a69633d
commit f047aff693
No known key found for this signature in database
1 changed files with 4 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
PostgreSQL/generate_data.py
View File

@ -479,10 +479,10 @@ if __name__ == '__main__':
nbMotifs : int = 10 nbMotifs : int = 10
store_table(nbMotifs, "motifs", tablesDict["motifs"]) store_table(nbMotifs, "motifs", tablesDict["motifs"])
nbPieces : int = 100 nbPieces : int = 50
store_table(nbPieces, "pieces", tablesDict["pieces"]) store_table(nbPieces, "pieces", tablesDict["pieces"])
nbPiecesComp : int = 10 nbPiecesComp : int = 5
store_table(nbPiecesComp, "pieces_complexes", store_table(nbPiecesComp, "pieces_complexes",
tablesDict["pieces_complexes"]) tablesDict["pieces_complexes"])
@ -492,7 +492,7 @@ if __name__ == '__main__':
# #
########################################################################### ###########################################################################
nbModeles : int = 5 nbModeles : int = 10
funcGen = lambda : tablesDict["modeles"]( funcGen = lambda : tablesDict["modeles"](
give_used_key(idMembreDict), give_used_key(idMembreDict),
give_used_key(idModeleDict) give_used_key(idModeleDict)
@ -638,25 +638,7 @@ if __name__ == '__main__':
) )
generate_table(nbVarier, "varier", funcGen) generate_table(nbVarier, "varier", funcGen)
########################################################################### ### The following code is in `convert_data.py`.
#
# To write `insert.sql`.
#
###########################################################################
# Remove duplicatas.
for nameTable in tablesDict:
remove_duplicata(give_path_file(nameTable))
# Writing in `insert.sql`.
fileSql = open("src/insert.sql", 'w+')
fileSql.write("TRUNCATE " + ', '.join(tablesDict) + ";\n")
fileSql.write("\n")
fileSql.close()
with open("src/insert.sql", 'a') as fileSql:
for nameTable in tablesDict:
convert_table_to_sql(give_path_file(nameTable), nameTable)
############################################################################### ###############################################################################