import os

os.chdir("C:/Users/Louis/Desktop/TIPE/fichiers_molecules")

nom = input("Nom de la molécule : ")

### Sécurité pour éviter de détruire un fichier existant

try:
    f = open(nom + '.py', 'r')
    print("Fichier déjà existant")

except FileNotFoundError:
    f = open(nom + ".py", 'w')
    f.write("import classes\n\n")
    
    n_at = int(input("Nombre d'atomes dans la molécule : "))
    
    for i in range(1, n_at + 1):
        nat = input("Atome " + str(i) + " ; nature de l'atome (C, H, O, etc) : ")
        n_vois = int(input("Nombre de voisins : "))
        l_vois = []
        for j in range(n_vois):
            l_vois.append(int(input("Numéro voisin (1 à la fois) : ")))
        f.write("atome" + str(i) + " = classes.Atome('" + nat + "', " + str(i) + ", " + str(l_vois) + ")\n")

    f.write("\n")

    f.write("tableau_coeffs_HO = classes.Tableau_de_coefficients([")

    for i in range(1, n_at):
        f.write("atome" + str(i) + ", ")
    f.write("atome" + str(n_at) + "], [")

    for i in range(1, n_at):
        f.write(input("Coefficient HO de l'atome " + str(i) + " : ") + ", ")
    f.write(input("Coefficient HO de l'atome " + str(n_at) + " : ") + "])\n\n")


    f.write("tableau_coeffs_BV = classes.Tableau_de_coefficients([")

    for i in range(1, n_at):
        f.write("atome" + str(i) + ", ")
    f.write("atome" + str(n_at) + "], [")

    for i in range(1, n_at):
        f.write(input("Coefficient BV de l'atome " + str(i) + " : ") + ", ")
    f.write(input("Coefficient BV de l'atome " + str(n_at) + " : ") + "])\n\n")


    f.write("HO = classes.Orbitale_moleculaire(")
    f.write(input("Energie de la HO : "))
    f.write(", tableau_coeffs_HO, 'HO')\n\n")

    f.write("BV = classes.Orbitale_moleculaire(")
    f.write(input("Energie de la BV : "))
    f.write(", tableau_coeffs_BV, 'BV')\n\n")

    f.write("molecule = classes.Molecule('" + nom + "', HO, BV)")

    f.close()

    print("Molécule enregistrée")    
