Mercurial > repos > iuc > data_manager_funannotate

annotate data_manager/extract.py @ 5:d2eb758a1d7d draft default tip

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
author iuc
date Fri, 07 Mar 2025 22:28:32 +0000
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
5
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
1 import json
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
2 import os
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
3 import sys
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
4
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
5 fun_db = sys.argv[1]
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
6 fun_db_value = sys.argv[2]
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
7 dmjson = sys.argv[3]
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
8
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
9 content = []
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
10 # get options for parameter --busco_db
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
11 # which are just the subfolders of the db dir (minus outgroups/ and trained_species/)
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
12 # https://github.com/nextgenusfs/funannotate/blob/8cc40728fee61566fdf736c1f2292e14cc117660/funannotate/predict.py#L319
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
13 for d in os.scandir(fun_db):
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
14 if not d.is_dir():
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
15 continue
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
16 if d.name in ['outgroups', 'trained_species']:
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
17 continue
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
18 if not os.path.exists(os.path.join(d, "dataset.cfg")):
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
19 continue
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
20 name = d.name.replace("_", " ").capitalize()
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
21 content.append({'value': d.name, 'name': name, 'select': 'busco_db', 'db_value': fun_db_value})
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
22
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
23 # --busco_seed_species
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
24 # trained_species
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
25 for d in os.scandir(os.path.join(fun_db, "trained_species")):
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
26 if not d.is_dir():
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
27 continue
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
28 if not os.path.exists(os.path.join(d, "info.json")):
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
29 continue
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
30 name = d.name.replace("_", " ").capitalize()
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
31 content.append({'value': d.name, 'name': name, 'select': 'trained_species', 'db_value': fun_db_value})
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
32
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
33 # --busco_seed_species
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
34 # outgroups
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
35 for f in os.scandir(os.path.join(fun_db, "outgroups")):
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
36 if f.is_dir():
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
37 continue
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
38 if not f.name.endswith("_buscos.fa"):
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
39 continue
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
40 value = f.name[:-10]
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
41 name = ' - '.join([x.replace("_", " ").capitalize() for x in value.split('.')])
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
42 content.append({'value': value, 'name': name, 'select': 'outgroup', 'db_value': fun_db_value})
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
43
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
44 with open(dmjson, "w") as fh:
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
45 json.dump({"data_tables": {"funannotate_options": content}}, fh)
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
46
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
47 print(f'{len([c for c in content if c["select"]=="busco_db"])} x busco_db\n')
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
48 print(f'{len([c for c in content if c["select"]=="trained_species"])} x trained_species\n')
d2eb758a1d7d planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/main/data_managers/data_manager_funannotate commit 8341270dd36185ebf59d15282bc79f1215e936a4
iuc
parents:
diff changeset
49 print(f'{len([c for c in content if c["select"]=="outgroup"])} x outgroup\n')