data-pebbles 0.0.1__py3-none-any.whl

data_pebbles/__init__.py

@@ -0,0 +1,10 @@
+import mlflow
+
+from data_pebbles.sdk import (
+	BronzeLayer,
+	DataPebbles,
+	GoldLayer,
+	SilverLayer,
+)
+
+__all__ = ["DataPebbles", "BronzeLayer", "SilverLayer", "GoldLayer", "mlflow"]

data_pebbles/client/__init__.py

@@ -0,0 +1,8 @@
+"""A client library for accessing FastAPI"""
+
+from .client import AuthenticatedClient, Client
+
+__all__ = (
+	"AuthenticatedClient",
+	"Client",
+)

data_pebbles/client/api/__init__.py

@@ -0,0 +1 @@
+"""Contains methods for accessing the API"""

data_pebbles/client/api/api_endpoints_for_interacting_with_the_bronze_layer/__init__.py

@@ -0,0 +1 @@
+"""Contains endpoint functions for accessing the API"""

data_pebbles/client/api/api_endpoints_for_interacting_with_the_bronze_layer/activate_version_bronze_source_id_versions_version_patch.py

@@ -0,0 +1,172 @@
+from http import HTTPStatus
+from typing import Any
+from urllib.parse import quote
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.http_validation_error import HTTPValidationError
+from ...types import Response
+
+
+def _get_kwargs(
+	source_id: int,
+	version: int,
+) -> dict[str, Any]:
+	_kwargs: dict[str, Any] = {
+		"method": "patch",
+		"url": "/bronze/{source_id}/versions/{version}".format(
+			source_id=quote(str(source_id), safe=""),
+			version=quote(str(version), safe=""),
+		),
+	}
+
+	return _kwargs
+
+
+def _parse_response(
+	*, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Any | HTTPValidationError | None:
+	if response.status_code == 200:
+		response_200 = response.json()
+		return response_200
+
+	if response.status_code == 422:
+		response_422 = HTTPValidationError.from_dict(response.json())
+
+		return response_422
+
+	if client.raise_on_unexpected_status:
+		raise errors.UnexpectedStatus(response.status_code, response.content)
+	else:
+		return None
+
+
+def _build_response(
+	*, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[Any | HTTPValidationError]:
+	return Response(
+		status_code=HTTPStatus(response.status_code),
+		content=response.content,
+		headers=response.headers,
+		parsed=_parse_response(client=client, response=response),
+	)
+
+
+def sync_detailed(
+	source_id: int,
+	version: int,
+	*,
+	client: AuthenticatedClient | Client,
+) -> Response[Any | HTTPValidationError]:
+	"""Activate Version
+
+	Args:
+	    source_id (int):
+	    version (int):
+
+	Raises:
+	    errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+	    httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+	Returns:
+	    Response[Any | HTTPValidationError]
+	"""
+
+	kwargs = _get_kwargs(
+		source_id=source_id,
+		version=version,
+	)
+
+	response = client.get_httpx_client().request(
+		**kwargs,
+	)
+
+	return _build_response(client=client, response=response)
+
+
+def sync(
+	source_id: int,
+	version: int,
+	*,
+	client: AuthenticatedClient | Client,
+) -> Any | HTTPValidationError | None:
+	"""Activate Version
+
+	Args:
+	    source_id (int):
+	    version (int):
+
+	Raises:
+	    errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+	    httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+	Returns:
+	    Any | HTTPValidationError
+	"""
+
+	return sync_detailed(
+		source_id=source_id,
+		version=version,
+		client=client,
+	).parsed
+
+
+async def asyncio_detailed(
+	source_id: int,
+	version: int,
+	*,
+	client: AuthenticatedClient | Client,
+) -> Response[Any | HTTPValidationError]:
+	"""Activate Version
+
+	Args:
+	    source_id (int):
+	    version (int):
+
+	Raises:
+	    errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+	    httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+	Returns:
+	    Response[Any | HTTPValidationError]
+	"""
+
+	kwargs = _get_kwargs(
+		source_id=source_id,
+		version=version,
+	)
+
+	response = await client.get_async_httpx_client().request(**kwargs)
+
+	return _build_response(client=client, response=response)
+
+
+async def asyncio(
+	source_id: int,
+	version: int,
+	*,
+	client: AuthenticatedClient | Client,
+) -> Any | HTTPValidationError | None:
+	"""Activate Version
+
+	Args:
+	    source_id (int):
+	    version (int):
+
+	Raises:
+	    errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+	    httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+	Returns:
+	    Any | HTTPValidationError
+	"""
+
+	return (
+		await asyncio_detailed(
+			source_id=source_id,
+			version=version,
+			client=client,
+		)
+	).parsed

data_pebbles/client/api/api_endpoints_for_interacting_with_the_bronze_layer/create_source_bronze_post.py

@@ -0,0 +1,164 @@
+from http import HTTPStatus
+from typing import Any
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.create_source_request import CreateSourceRequest
+from ...models.http_validation_error import HTTPValidationError
+from ...types import Response
+
+
+def _get_kwargs(
+	*,
+	body: CreateSourceRequest,
+) -> dict[str, Any]:
+	headers: dict[str, Any] = {}
+
+	_kwargs: dict[str, Any] = {
+		"method": "post",
+		"url": "/bronze/",
+	}
+
+	_kwargs["json"] = body.to_dict()
+
+	headers["Content-Type"] = "application/json"
+
+	_kwargs["headers"] = headers
+	return _kwargs
+
+
+def _parse_response(
+	*, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Any | HTTPValidationError | None:
+	if response.status_code == 200:
+		response_200 = response.json()
+		return response_200
+
+	if response.status_code == 422:
+		response_422 = HTTPValidationError.from_dict(response.json())
+
+		return response_422
+
+	if client.raise_on_unexpected_status:
+		raise errors.UnexpectedStatus(response.status_code, response.content)
+	else:
+		return None
+
+
+def _build_response(
+	*, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[Any | HTTPValidationError]:
+	return Response(
+		status_code=HTTPStatus(response.status_code),
+		content=response.content,
+		headers=response.headers,
+		parsed=_parse_response(client=client, response=response),
+	)
+
+
+def sync_detailed(
+	*,
+	client: AuthenticatedClient | Client,
+	body: CreateSourceRequest,
+) -> Response[Any | HTTPValidationError]:
+	"""Create Source
+
+	Args:
+	    body (CreateSourceRequest):
+
+	Raises:
+	    errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+	    httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+	Returns:
+	    Response[Any | HTTPValidationError]
+	"""
+
+	kwargs = _get_kwargs(
+		body=body,
+	)
+
+	response = client.get_httpx_client().request(
+		**kwargs,
+	)
+
+	return _build_response(client=client, response=response)
+
+
+def sync(
+	*,
+	client: AuthenticatedClient | Client,
+	body: CreateSourceRequest,
+) -> Any | HTTPValidationError | None:
+	"""Create Source
+
+	Args:
+	    body (CreateSourceRequest):
+
+	Raises:
+	    errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+	    httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+	Returns:
+	    Any | HTTPValidationError
+	"""
+
+	return sync_detailed(
+		client=client,
+		body=body,
+	).parsed
+
+
+async def asyncio_detailed(
+	*,
+	client: AuthenticatedClient | Client,
+	body: CreateSourceRequest,
+) -> Response[Any | HTTPValidationError]:
+	"""Create Source
+
+	Args:
+	    body (CreateSourceRequest):
+
+	Raises:
+	    errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+	    httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+	Returns:
+	    Response[Any | HTTPValidationError]
+	"""
+
+	kwargs = _get_kwargs(
+		body=body,
+	)
+
+	response = await client.get_async_httpx_client().request(**kwargs)
+
+	return _build_response(client=client, response=response)
+
+
+async def asyncio(
+	*,
+	client: AuthenticatedClient | Client,
+	body: CreateSourceRequest,
+) -> Any | HTTPValidationError | None:
+	"""Create Source
+
+	Args:
+	    body (CreateSourceRequest):
+
+	Raises:
+	    errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+	    httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+	Returns:
+	    Any | HTTPValidationError
+	"""
+
+	return (
+		await asyncio_detailed(
+			client=client,
+			body=body,
+		)
+	).parsed

data_pebbles/client/api/api_endpoints_for_interacting_with_the_bronze_layer/delete_source_bronze_source_id_delete.py

@@ -0,0 +1,158 @@
+from http import HTTPStatus
+from typing import Any
+from urllib.parse import quote
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.http_validation_error import HTTPValidationError
+from ...types import Response
+
+
+def _get_kwargs(
+	source_id: int,
+) -> dict[str, Any]:
+	_kwargs: dict[str, Any] = {
+		"method": "delete",
+		"url": "/bronze/{source_id}".format(
+			source_id=quote(str(source_id), safe=""),
+		),
+	}
+
+	return _kwargs
+
+
+def _parse_response(
+	*, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Any | HTTPValidationError | None:
+	if response.status_code == 200:
+		response_200 = response.json()
+		return response_200
+
+	if response.status_code == 422:
+		response_422 = HTTPValidationError.from_dict(response.json())
+
+		return response_422
+
+	if client.raise_on_unexpected_status:
+		raise errors.UnexpectedStatus(response.status_code, response.content)
+	else:
+		return None
+
+
+def _build_response(
+	*, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[Any | HTTPValidationError]:
+	return Response(
+		status_code=HTTPStatus(response.status_code),
+		content=response.content,
+		headers=response.headers,
+		parsed=_parse_response(client=client, response=response),
+	)
+
+
+def sync_detailed(
+	source_id: int,
+	*,
+	client: AuthenticatedClient | Client,
+) -> Response[Any | HTTPValidationError]:
+	"""Delete Source
+
+	Args:
+	    source_id (int):
+
+	Raises:
+	    errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+	    httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+	Returns:
+	    Response[Any | HTTPValidationError]
+	"""
+
+	kwargs = _get_kwargs(
+		source_id=source_id,
+	)
+
+	response = client.get_httpx_client().request(
+		**kwargs,
+	)
+
+	return _build_response(client=client, response=response)
+
+
+def sync(
+	source_id: int,
+	*,
+	client: AuthenticatedClient | Client,
+) -> Any | HTTPValidationError | None:
+	"""Delete Source
+
+	Args:
+	    source_id (int):
+
+	Raises:
+	    errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+	    httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+	Returns:
+	    Any | HTTPValidationError
+	"""
+
+	return sync_detailed(
+		source_id=source_id,
+		client=client,
+	).parsed
+
+
+async def asyncio_detailed(
+	source_id: int,
+	*,
+	client: AuthenticatedClient | Client,
+) -> Response[Any | HTTPValidationError]:
+	"""Delete Source
+
+	Args:
+	    source_id (int):
+
+	Raises:
+	    errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+	    httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+	Returns:
+	    Response[Any | HTTPValidationError]
+	"""
+
+	kwargs = _get_kwargs(
+		source_id=source_id,
+	)
+
+	response = await client.get_async_httpx_client().request(**kwargs)
+
+	return _build_response(client=client, response=response)
+
+
+async def asyncio(
+	source_id: int,
+	*,
+	client: AuthenticatedClient | Client,
+) -> Any | HTTPValidationError | None:
+	"""Delete Source
+
+	Args:
+	    source_id (int):
+
+	Raises:
+	    errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+	    httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+	Returns:
+	    Any | HTTPValidationError
+	"""
+
+	return (
+		await asyncio_detailed(
+			source_id=source_id,
+			client=client,
+		)
+	).parsed

data_pebbles/client/api/api_endpoints_for_interacting_with_the_bronze_layer/delete_version_bronze_source_id_versions_version_delete.py

@@ -0,0 +1,172 @@
+from http import HTTPStatus
+from typing import Any
+from urllib.parse import quote
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.http_validation_error import HTTPValidationError
+from ...types import Response
+
+
+def _get_kwargs(
+	source_id: int,
+	version: int,
+) -> dict[str, Any]:
+	_kwargs: dict[str, Any] = {
+		"method": "delete",
+		"url": "/bronze/{source_id}/versions/{version}".format(
+			source_id=quote(str(source_id), safe=""),
+			version=quote(str(version), safe=""),
+		),
+	}
+
+	return _kwargs
+
+
+def _parse_response(
+	*, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Any | HTTPValidationError | None:
+	if response.status_code == 200:
+		response_200 = response.json()
+		return response_200
+
+	if response.status_code == 422:
+		response_422 = HTTPValidationError.from_dict(response.json())
+
+		return response_422
+
+	if client.raise_on_unexpected_status:
+		raise errors.UnexpectedStatus(response.status_code, response.content)
+	else:
+		return None
+
+
+def _build_response(
+	*, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[Any | HTTPValidationError]:
+	return Response(
+		status_code=HTTPStatus(response.status_code),
+		content=response.content,
+		headers=response.headers,
+		parsed=_parse_response(client=client, response=response),
+	)
+
+
+def sync_detailed(
+	source_id: int,
+	version: int,
+	*,
+	client: AuthenticatedClient | Client,
+) -> Response[Any | HTTPValidationError]:
+	"""Delete Version
+
+	Args:
+	    source_id (int):
+	    version (int):
+
+	Raises:
+	    errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+	    httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+	Returns:
+	    Response[Any | HTTPValidationError]
+	"""
+
+	kwargs = _get_kwargs(
+		source_id=source_id,
+		version=version,
+	)
+
+	response = client.get_httpx_client().request(
+		**kwargs,
+	)
+
+	return _build_response(client=client, response=response)
+
+
+def sync(
+	source_id: int,
+	version: int,
+	*,
+	client: AuthenticatedClient | Client,
+) -> Any | HTTPValidationError | None:
+	"""Delete Version
+
+	Args:
+	    source_id (int):
+	    version (int):
+
+	Raises:
+	    errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+	    httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+	Returns:
+	    Any | HTTPValidationError
+	"""
+
+	return sync_detailed(
+		source_id=source_id,
+		version=version,
+		client=client,
+	).parsed
+
+
+async def asyncio_detailed(
+	source_id: int,
+	version: int,
+	*,
+	client: AuthenticatedClient | Client,
+) -> Response[Any | HTTPValidationError]:
+	"""Delete Version
+
+	Args:
+	    source_id (int):
+	    version (int):
+
+	Raises:
+	    errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+	    httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+	Returns:
+	    Response[Any | HTTPValidationError]
+	"""
+
+	kwargs = _get_kwargs(
+		source_id=source_id,
+		version=version,
+	)
+
+	response = await client.get_async_httpx_client().request(**kwargs)
+
+	return _build_response(client=client, response=response)
+
+
+async def asyncio(
+	source_id: int,
+	version: int,
+	*,
+	client: AuthenticatedClient | Client,
+) -> Any | HTTPValidationError | None:
+	"""Delete Version
+
+	Args:
+	    source_id (int):
+	    version (int):
+
+	Raises:
+	    errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+	    httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+	Returns:
+	    Any | HTTPValidationError
+	"""
+
+	return (
+		await asyncio_detailed(
+			source_id=source_id,
+			version=version,
+			client=client,
+		)
+	).parsed