Skip to content

Commit

Permalink
chore(use_cases): refactor main pipeline use_cases (#51)
Browse files Browse the repository at this point in the history
  • Loading branch information
kennedykori committed Aug 25, 2022
1 parent 638d436 commit 7c3456a
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 4 deletions.
9 changes: 5 additions & 4 deletions app/__main__.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,14 @@
from argparse import ArgumentParser
from typing import Any, Optional, Sequence
from typing import Optional, Sequence

import app
from app.__version__ import __title__, __version__
from app.core import DataSourceType, Transport
from app.lib import Pipeline
from app.use_cases.main_pipeline import (
from app.use_cases import (
FetchMetadata,
RunExtraction,
UploadExtractResult,
UploadExtracts,
)

Expand Down Expand Up @@ -75,7 +76,7 @@ def argparse_factory(prog_name: str = __title__) -> ArgumentParser:

def main_pipeline_factory(
transport: Optional[Transport] = None,
) -> Pipeline[Sequence[DataSourceType], Any]:
) -> Pipeline[Sequence[DataSourceType], Sequence[UploadExtractResult]]:
"""A factory for the main application pipeline.
Returns a fully initialized pipeline ready for use. An optional
Expand Down Expand Up @@ -124,7 +125,7 @@ def main() -> None: # pragma: no cover
transport_factory = app.registry.get_default_transport_factory_or_raise()
with transport_factory() as transport:
main_pipeline: Pipeline[
Sequence[DataSourceType], Any
Sequence[DataSourceType], Sequence[UploadExtractResult]
] = main_pipeline_factory(transport=transport)
main_pipeline.execute(tuple(app.registry.data_source_types.values()))
print("Done ...")
Expand Down
2 changes: 2 additions & 0 deletions app/use_cases/__init__.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,9 @@
from .main_pipeline import FetchMetadata, RunExtraction, UploadExtracts
from .types import UploadExtractResult

__all__ = [
"FetchMetadata",
"RunExtraction",
"UploadExtractResult",
"UploadExtracts",
]

0 comments on commit 7c3456a

Please sign in to comment.