A little refinement.

Author: alinzh

Diff for: graphviz_dot_handler.py

@@ -143,11 +143,7 @@ def diagram_bilder(self):
         if not os.path.exists('diagram_folder'):
             os.makedirs('diagram_folder')
 
-        # if not self.output_path:
         cmd = f'dot -Tpng demo.dot -o ./diagram_folder/{self.name_db}_{self.date_today}.png'
-        # else:
-        #     self.output_path = self.output_path.replace("\\", '/')
-        #     cmd = f'dot -Tpng demo.dot -o {self.output_path}/{self.name_db}_{self.date_today}.png'
 
         try:
             subprocess.run(cmd)

Diff for: main.py

@@ -19,6 +19,9 @@ def main(host, port, user, password, db_name, schema_name, engine=None, directio
     except NameError:
         print('No tables found!')
         raise
+    else:
+        if not answer:
+            print('Failed to get database data.')
 
     tables_structure, foreign_keys_for_diagram_builder, keys_in_table, \
         number_of_keys, primary_keys, column_types = answer