Mercurial > repos > tduigou > get_db_info
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")
