Source code for rail_pz_service.client.dataset

"""python for client API for managing Dataset tables"""

from __future__ import annotations

from typing import TYPE_CHECKING

import httpx

from .. import models
from . import wrappers

if TYPE_CHECKING:
    from .client import PZRailClient

# Template specialization
# Specify the pydantic model for Step
ResponseModelClass = models.Dataset

# Construct derived templates
router_string = "dataset"




class PZRailDatasetClient:
    """Interface for accessing remote pz-rail-service to manipulate
    Dataset Tables
    """

    def __init__(self, parent: PZRailClient) -> None:
        self._client = parent.client

    @property
    def client(self) -> httpx.Client:
        """Return the httpx.Client"""
        return self._client

    # Add functions to the client class
    get_rows = wrappers.get_rows_function(ResponseModelClass, f"{router_string}/list")

    get_row = wrappers.get_row_function(ResponseModelClass, f"{router_string}/get")

    get_row_by_name = wrappers.get_row_by_name_function(
        ResponseModelClass, f"{router_string}/get_row_by_name"
    )

    get_estimators = wrappers.get_row_attribute_list_function(
        ResponseModelClass, f"{router_string}/get", "estimators"
    )

    get_models = wrappers.get_row_attribute_list_function(
        ResponseModelClass, f"{router_string}/get", "models"
    )

    get_requests = wrappers.get_row_attribute_list_function(
        ResponseModelClass, f"{router_string}/get", "requests"
    )

    download = wrappers.download_file_function(f"{router_string}/download")