Skip to content

Commit

Permalink
Update results exporters
Browse files Browse the repository at this point in the history
  • Loading branch information
jlumpe committed Dec 1, 2024
1 parent 9ccddce commit 36784fd
Showing 1 changed file with 7 additions and 11 deletions.
18 changes: 7 additions & 11 deletions src/gambit/results.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@

from gambit.util.io import FilePath, maybe_open
import gambit.util.json as gjson
from gambit.query import QueryResults, QueryResultItem, QueryInput
from gambit.query import QueryResults, QueryResultItem
from gambit.db import ReferenceGenomeSet, Taxon, AnnotatedGenome, Genome


Expand Down Expand Up @@ -84,8 +84,9 @@ class CSVResultsExporter(AbstractResultsExporter):
"""
format_opts: dict[str, Any]

# Pairs of column name and QueryResultItem attribute
COLUMNS = [
('query', 'input.label'),
('query', 'label'),
('predicted.name', 'report_taxon.name'),
('predicted.rank', 'report_taxon.rank'),
('predicted.ncbi_id', 'report_taxon.ncbi_id'),
Expand Down Expand Up @@ -140,20 +141,15 @@ def _results_to_json(self, results: QueryResults):
@to_json.register(QueryResultItem)
def _item_to_json(self, item: QueryResultItem):
return dict(
query=item.input,
query=dict(
name=item.label,
path=item.file,
),
predicted_taxon=item.report_taxon,
next_taxon=item.classifier_result.next_taxon,
closest_genomes=item.closest_genomes,
)

@to_json.register(QueryInput)
def _input_to_json(self, input: QueryInput):
return dict(
name=input.label,
path=None if input.file is None else input.file.path,
format=None if input.file is None else input.file.format,
)

@to_json.register(ReferenceGenomeSet)
def _genomeset_to_json(self, gset: ReferenceGenomeSet):
return _todict(gset, ['id', 'key', 'version', 'name', 'description'])
Expand Down

0 comments on commit 36784fd

Please sign in to comment.