diff get_db_info.py @ 9:6401a277ee72 draft

planemo upload for repository https://github.com/brsynth commit 6ae809b563b40bcdb6be2e74fe2a84ddad5484ae
author tduigou
date Fri, 18 Apr 2025 13:36:40 +0000
parents 8984fabea52c
children 703fda6c48c5
line wrap: on
line diff
--- a/get_db_info.py	Fri Apr 18 13:07:58 2025 +0000
+++ b/get_db_info.py	Fri Apr 18 13:36:40 2025 +0000
@@ -81,10 +81,10 @@
             time.sleep(2)
     raise Exception("Database connection failed after timeout.")
 
-def fetch_annotations(tsv_file, sequence_column, annotation_columns, db_uri, table_name, fragment_column_name, output):
+def fetch_annotations(csv_file, sequence_column, annotation_columns, db_uri, table_name, fragment_column_name, output):
     """Fetch annotations from the database and save the result as GenBank files."""
     db_uri = fix_db_uri(db_uri)
-    df = pd.read_csv(tsv_file, sep='\t')
+    df = pd.read_csv(csv_file, sep=',')
     
     engine = create_engine(db_uri)
     connection = engine.connect()
@@ -216,7 +216,7 @@
 
 def main():
     parser = argparse.ArgumentParser(description="Fetch annotations from PostgreSQL database and save as JSON.")
-    parser.add_argument("--input", required=True, help="Input TSV file")
+    parser.add_argument("--input", required=True, help="Input CSV file")
     parser.add_argument("--sequence_column", required=True, help="DB column contains sequence for ganbank file")
     parser.add_argument("--annotation_columns", required=True, help="DB column contains head for ganbank file")
     parser.add_argument("--db_uri", required=True, help="Database URI connection string")