v1.1.0 of main script update

shadmehr-gh · May 17, 2024 · 3cce708 · 3cce708
1 parent 25f2eda
commit 3cce708
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/zinc_downloader.py b/zinc_downloader.py
@@ -23,7 +23,6 @@ def download_molecule(zinc_id, zinc_version, zinc_file_type):
         if response.status_code == 200:
             with open(file_path, "wb") as f:
                 f.write(response.content)
-            print(Fore.GREEN + f"Downloaded {zinc_id}.{zinc_file_type}" + Fore.RESET)
         else:
             print(Fore.RED + f"Failed to download {zinc_id}.{zinc_file_type}" + Fore.RESET)
     except requests.exceptions.RequestException as e:
@@ -70,7 +69,7 @@ def zinc_merger(zinc_file_type, zinc_id_list):
             except Exception as e:
                 print(Fore.RED + f"Error while merging {zinc_id}.{zinc_file_type}: {e}" + Fore.RESET)
 
-    print(Fore.GREEN + "All molecules merged successfully" + Fore.RESET)
+    print(Fore.GREEN + "All molecules merged successfully\n" + Fore.RESET)
 
 def main():
     print(Fore.BLUE + "\n****************************************************")
@@ -115,6 +114,8 @@ def main():
                 print(Fore.RED + f"Error: {e}" + Fore.RESET)
 
     print("****************************************************\n")
+    print(Fore.GREEN + "Download job finished." + Fore.RESET)
+    print("****************************************************\n")
 
     if merge_bool == "yes":
         zinc_merger(zinc_file_type, zinc_id_list)