Commit c66fd718 authored by Jan Reimes's avatar Jan Reimes
Browse files

refactor(specs): consolidate query models into models.py and remove dead code

parent 18226a3c

src/tdoc_crawler/specs/models.py

+8 −0
Original line number Diff line number Diff line
@@ -79,6 +79,10 @@ class SpecQueryFilters: 
    working_group: str | None = None

    status: str | None = None



    def __rich_console__(self, console: Console, options: ConsoleOptions) -> RenderResult:

        _ = (console, options)

        yield Text(f"SpecQueryFilters(spec_numbers={self.spec_numbers})")





@dataclass

class SpecQueryResult:
@@ -90,6 +94,10 @@ class SpecQueryResult: 
    working_group: str | None = None

    source_differences: dict[str, dict[str, str | None]] = field(default_factory=dict)



    def __rich_console__(self, console: Console, options: ConsoleOptions) -> RenderResult:

        _ = (console, options)

        yield Text(f"{self.spec_number} - {self.title or 'Unknown'}")





__all__ = [

    "SpecQueryFilters",

src/tdoc_crawler/specs/query.py

deleted100644 → 0
+0 −35
Original line number Diff line number Diff line 
"""Spec query filters and results."""



from dataclasses import dataclass, field



from rich.console import Console, ConsoleOptions, RenderResult

from rich.text import Text





@dataclass

class SpecQueryFilters:

    """Filter inputs for spec queries."""



    spec_numbers: list[str] | None = None

    title: str | None = None

    working_group: str | None = None

    status: str | None = None



    def __rich_console__(self, console: Console, options: ConsoleOptions) -> RenderResult:

        _ = (console, options)

        yield Text(f"SpecQueryFilters(spec_numbers={self.spec_numbers})")





@dataclass

class SpecQueryResult:

    """Query result row for a spec."""



    spec_number: str

    title: str | None = None

    status: str | None = None

    working_group: str | None = None

    source_differences: dict[str, dict[str, str | None]] = field(default_factory=dict)



    def __rich_console__(self, console: Console, options: ConsoleOptions) -> RenderResult:

        _ = (console, options)

        yield Text(f"{self.spec_number} - {self.title or 'Unknown'}")

tests/test_specs_database.py

+1 −1
Original line number Diff line number Diff line
@@ -4,7 +4,7 @@ from pathlib import Path 


from tdoc_crawler.database.specs import SpecDatabase

from tdoc_crawler.specs.models import Specification, SpecificationDownload, SpecificationSourceRecord, SpecificationVersion

from tdoc_crawler.specs.query import SpecQueryFilters

from tdoc_crawler.specs.models import Specification, SpecificationDownload, SpecificationSourceRecord, SpecificationVersion, SpecQueryFilters





def test_upsert_specification_roundtrip(tmp_path: Path) -> None: