<reponame>PradyumnaKrishna/electionguard-python<filename>src/electionguard_cli/e2e/e2e_input_retrieval_step.py<gh_stars>0
from io import TextIOWrapper
from electionguard.ballot import PlaintextBallot

from electionguard.key_ceremony import CeremonyDetails
from electionguard.manifest import Manifest
from electionguard_tools.helpers.key_ceremony_orchestrator import (
    KeyCeremonyOrchestrator,
)

from ..cli_steps import (
    InputRetrievalStepBase,
)
from .e2e_inputs import E2eInputs


class E2eInputRetrievalStep(InputRetrievalStepBase):
    """Responsible for retrieving and parsing user provided inputs for the CLI's e2e command."""

    def get_inputs(
        self,
        guardian_count: int,
        quorum: int,
        manifest_file: TextIOWrapper,
        ballots_path: str,
        spoil_id: str,
        output_record: str,
        output_keys: str,
    ) -> E2eInputs:
        self.print_header("Retrieving Inputs")
        guardians = KeyCeremonyOrchestrator.create_guardians(
            CeremonyDetails(guardian_count, quorum)
        )
        manifest: Manifest = self._get_manifest(manifest_file)
        ballots = E2eInputRetrievalStep._get_ballots(ballots_path, PlaintextBallot)
        self.print_value("Guardians", guardian_count)
        self.print_value("Quorum", quorum)
        return E2eInputs(
            guardian_count,
            quorum,
            guardians,
            manifest,
            ballots,
            spoil_id,
            output_record,
            output_keys,
        )
