Skip to content

Commit

Permalink
Add from_runs to esmecata_create_db.
Browse files Browse the repository at this point in the history
  • Loading branch information
ArnaudBelcour committed Nov 9, 2024
1 parent 4276046 commit 1cc03bf
Show file tree
Hide file tree
Showing 2 changed files with 38 additions and 1 deletion.
1 change: 1 addition & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@

## Add

* Create database from different output folders of esmecata (`from_runs`).
* Check the good format of the gzip file.

## Fix
Expand Down
38 changes: 37 additions & 1 deletion esmecata/__main_create_database__.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,30 @@ def main():
help='EsMeCaTa output folder.',
metavar='INPUT_FOLDER')

parent_parser_iproteomes = argparse.ArgumentParser(add_help=False)
parent_parser_iproteomes.add_argument(
'-iproteomes',
dest='input_proteomes',
required=True,
help='EsMeCaTa proteomes output folder.',
metavar='INPUT_FOLDER')

parent_parser_iclustering = argparse.ArgumentParser(add_help=False)
parent_parser_iclustering.add_argument(
'-iclustering',
dest='input_clustering',
required=True,
help='EsMeCaTa clustering output folder.',
metavar='INPUT_FOLDER')

parent_parser_iannotation = argparse.ArgumentParser(add_help=False)
parent_parser_iannotation.add_argument(
'-iannotation',
dest='input_annotation',
required=True,
help='EsMeCaTa annotation output folder.',
metavar='INPUT_FOLDER')

parent_parser_i_zip = argparse.ArgumentParser(add_help=False)
parent_parser_i_zip.add_argument(
'-i',
Expand Down Expand Up @@ -105,6 +129,15 @@ def main():
],
allow_abbrev=False)

from_esmecata_runs = subparsers.add_parser(
'from_runs',
help='Create database from esmecata proteomes, clustering and annotation output folders.',
parents=[
parent_parser_iproteomes, parent_parser_iclustering, parent_parser_iannotation, parent_parser_o,
parent_parser_database_version, parent_parser_c
],
allow_abbrev=False)

merge_db = subparsers.add_parser(
'merge_db',
help='Merge multiple zip files corresponding to EsMeCaTa databases.',
Expand Down Expand Up @@ -146,7 +179,10 @@ def main():
if ' ' in args.input:
list_db_files = args.input.split(' ')
merge_db_files(list_db_files, args.output)

elif args.cmd == 'from_runs':
create_database_from_esmecata_run(args.input_proteomes, args.input_clustering, args.parent_parser_iannotation,
args.output, args.database_version, args.core)

logger.info("--- Total runtime %.2f seconds ---" % (time.time() - start_time))
logger.warning(f'--- Logs written in {log_file_path} ---')

Expand Down

0 comments on commit 1cc03bf

Please sign in to comment.