Replace generate_citation.py

generate_citation.py

@@ -36,34 +36,39 @@ def citation_generation(cit_authors,cit_title,cit_year,cit_data_repository,cit_v
 	cit_authors_len_test = cit_authors_len
 	for cit_author in cit_authors:
 		if cit_authors_len_test == cit_authors_len : # le nom du premier auteur est suivi d'un point-virgule, les autres d'une simple virgule
-			print(f"{cit_author}; ",end='')
+			citation = f"{cit_author}; "
 		else:
-			print(f"{cit_author}, ",end='')
+			citation = f"{cit_author}, "
 		cit_authors_len_test = cit_authors_len_test - 1
-	print (f"{cit_year}, ", end='')
-	print (f"\"{cit_title}\", ",end='')
-	print (f"{cit_data_repository}, ",end='')
-	print (f"V{cit_version_number}.{cit_version_minor_number}, ", end='')
-	print (f"http://dx.doi.org/{cit_doi}", end = '')
+	citation += f"{cit_year}, "
+	citation += f"\"{cit_title}\", "
+	citation += f"{cit_data_repository}, "
+	citation += f"V{cit_version_number}.{cit_version_minor_number}, "
+	citation += f"http://dx.doi.org/{cit_doi}"
 	if cit_unf !="" :
-		print (f"UNF: {cit_unf}.")
+		citation += f"UNF: {cit_unf}."
 	else : 
-		print (".\n")
+		citation += ".\n"
+	return (citation)
 
 
 # ----------------
 # Fichier d'entrée
 # ----------------
 
-filename_input = input ("Entrez le nom du fichier à traiter [export_json.json] : ")
-if filename_input =="":
-	filename = "export_json.json"
-else:
-	filename = filename_input
+# filename_input = input ("Entrez le nom du fichier à traiter [export_json.json] : ")
+# if filename_input =="":
+# 	filename = "export_json.json"
+# else:
+# 	filename = filename_input
+
+filename = "export_json.json"
 
 with open (filename) as f:
 	complete_list = json.load(f)
 
+bibliography_list = []
+
 for doi,metadata in complete_list.items(): # 
 	# print (f"\n{doi} : \n")
 	authors_list=[]
@@ -124,7 +129,27 @@ for doi,metadata in complete_list.items(): #
 
 	# Appel de la fonction
 
-	citation_generation(cit_authors = authors_list,cit_title=title,cit_year=dataset_yearofPublication,cit_data_repository=dataset_publisher,cit_version_number=dataset_versionNumber,cit_version_minor_number=dataset_versionMinorNumber, cit_doi=doi,cit_unf=dataset_unf)
+	citation_current = citation_generation(cit_authors = authors_list,cit_title=title,cit_year=dataset_yearofPublication,cit_data_repository=dataset_publisher,cit_version_number=dataset_versionNumber,cit_version_minor_number=dataset_versionMinorNumber, cit_doi=doi,cit_unf=dataset_unf)
+	print (citation_current)
+	bibliography_list.append(citation_current)
+
+# Génération de la bibliographie
+bibliography = ""
+for reference in bibliography_list:
+	bibliography += (f"{reference}\n")
+export_filename_input = input ("Entrez un nom de fichier pour la sortie en format texte (défaut datasets_list.txt) : ")
+if export_filename_input == "":
+	export_filename = "datasets_list.txt"
+else:
+	export_filename = export_filename_input
+
+with open(export_filename, 'w') as f:
+	f.write(bibliography)
+
+
+
+
+
 
 
  	
\ No newline at end of file