Skip to content

Commit 36784fd

Browse files
committed
Update results exporters
1 parent 9ccddce commit 36784fd

File tree

1 file changed

+7
-11
lines changed

1 file changed

+7
-11
lines changed

src/gambit/results.py

+7-11
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111

1212
from gambit.util.io import FilePath, maybe_open
1313
import gambit.util.json as gjson
14-
from gambit.query import QueryResults, QueryResultItem, QueryInput
14+
from gambit.query import QueryResults, QueryResultItem
1515
from gambit.db import ReferenceGenomeSet, Taxon, AnnotatedGenome, Genome
1616

1717

@@ -84,8 +84,9 @@ class CSVResultsExporter(AbstractResultsExporter):
8484
"""
8585
format_opts: dict[str, Any]
8686

87+
# Pairs of column name and QueryResultItem attribute
8788
COLUMNS = [
88-
('query', 'input.label'),
89+
('query', 'label'),
8990
('predicted.name', 'report_taxon.name'),
9091
('predicted.rank', 'report_taxon.rank'),
9192
('predicted.ncbi_id', 'report_taxon.ncbi_id'),
@@ -140,20 +141,15 @@ def _results_to_json(self, results: QueryResults):
140141
@to_json.register(QueryResultItem)
141142
def _item_to_json(self, item: QueryResultItem):
142143
return dict(
143-
query=item.input,
144+
query=dict(
145+
name=item.label,
146+
path=item.file,
147+
),
144148
predicted_taxon=item.report_taxon,
145149
next_taxon=item.classifier_result.next_taxon,
146150
closest_genomes=item.closest_genomes,
147151
)
148152

149-
@to_json.register(QueryInput)
150-
def _input_to_json(self, input: QueryInput):
151-
return dict(
152-
name=input.label,
153-
path=None if input.file is None else input.file.path,
154-
format=None if input.file is None else input.file.format,
155-
)
156-
157153
@to_json.register(ReferenceGenomeSet)
158154
def _genomeset_to_json(self, gset: ReferenceGenomeSet):
159155
return _todict(gset, ['id', 'key', 'version', 'name', 'description'])

0 commit comments

Comments
 (0)