# -*- coding: utf-8 -*-
# Generated by https://github.com/verloop/twirpy/protoc-gen-twirpy.  DO NOT EDIT!
# source: catalog.proto

from google.protobuf import symbol_database as _symbol_database

from twirp.base import Endpoint
from twirp.server import TwirpServer
from twirp.client import TwirpClient

_sym_db = _symbol_database.Default()

class CatalogServiceServer(TwirpServer):

	def __init__(self, *args, service):
		super().__init__(service=service)
		self._prefix = "/twirp/catalog.CatalogService"
		self._endpoints = {
			"ListDatasets": Endpoint(
				service_name="CatalogService",
				name="ListDatasets",
				function=getattr(service, "ListDatasets"),
				input=_sym_db.GetSymbol("catalog.ListDatasetsRequest"),
				output=_sym_db.GetSymbol("catalog.ListDatasetsResponse"),
			),
			"ListOrganizationDatasets": Endpoint(
				service_name="CatalogService",
				name="ListOrganizationDatasets",
				function=getattr(service, "ListOrganizationDatasets"),
				input=_sym_db.GetSymbol("catalog.ListOrganizationDatasetsRequest"),
				output=_sym_db.GetSymbol("catalog.ListOrganizationDatasetsResponse"),
			),
			"ListGroupDatasets": Endpoint(
				service_name="CatalogService",
				name="ListGroupDatasets",
				function=getattr(service, "ListGroupDatasets"),
				input=_sym_db.GetSymbol("catalog.ListGroupDatasetsRequest"),
				output=_sym_db.GetSymbol("catalog.ListGroupDatasetsResponse"),
			),
			"GetDataset": Endpoint(
				service_name="CatalogService",
				name="GetDataset",
				function=getattr(service, "GetDataset"),
				input=_sym_db.GetSymbol("catalog.GetDatasetRequest"),
				output=_sym_db.GetSymbol("catalog.GetDatasetResponse"),
			),
			"UpdateDataset": Endpoint(
				service_name="CatalogService",
				name="UpdateDataset",
				function=getattr(service, "UpdateDataset"),
				input=_sym_db.GetSymbol("catalog.UpdateDatasetRequest"),
				output=_sym_db.GetSymbol("catalog.UpdateDatasetResponse"),
			),
			"UpdateProperties": Endpoint(
				service_name="CatalogService",
				name="UpdateProperties",
				function=getattr(service, "UpdateProperties"),
				input=_sym_db.GetSymbol("catalog.UpdatePropertiesRequest"),
				output=_sym_db.GetSymbol("catalog.UpdatePropertiesResponse"),
			),
			"TagRevision": Endpoint(
				service_name="CatalogService",
				name="TagRevision",
				function=getattr(service, "TagRevision"),
				input=_sym_db.GetSymbol("catalog.TagRevisionRequest"),
				output=_sym_db.GetSymbol("catalog.TagRevisionResponse"),
			),
			"UntagRevision": Endpoint(
				service_name="CatalogService",
				name="UntagRevision",
				function=getattr(service, "UntagRevision"),
				input=_sym_db.GetSymbol("catalog.UntagRevisionRequest"),
				output=_sym_db.GetSymbol("catalog.UntagRevisionResponse"),
			),
			"ListTags": Endpoint(
				service_name="CatalogService",
				name="ListTags",
				function=getattr(service, "ListTags"),
				input=_sym_db.GetSymbol("catalog.ListTagsRequest"),
				output=_sym_db.GetSymbol("catalog.ListTagsResponse"),
			),
			"PruneVersion": Endpoint(
				service_name="CatalogService",
				name="PruneVersion",
				function=getattr(service, "PruneVersion"),
				input=_sym_db.GetSymbol("catalog.PruneVersionRequest"),
				output=_sym_db.GetSymbol("catalog.PruneVersionResponse"),
			),
			"PruneRevision": Endpoint(
				service_name="CatalogService",
				name="PruneRevision",
				function=getattr(service, "PruneRevision"),
				input=_sym_db.GetSymbol("catalog.PruneRevisionRequest"),
				output=_sym_db.GetSymbol("catalog.PruneRevisionResponse"),
			),
			"Healthz": Endpoint(
				service_name="CatalogService",
				name="Healthz",
				function=getattr(service, "Healthz"),
				input=_sym_db.GetSymbol("catalog.Empty"),
				output=_sym_db.GetSymbol("catalog.Empty"),
			),
		}

class CatalogServiceClient(TwirpClient):

	def ListDatasets(self, *args, ctx, request, **kwargs):
		return self._make_request(
			url="/twirp/catalog.CatalogService/ListDatasets",
			ctx=ctx,
			request=request,
			response_obj=_sym_db.GetSymbol("catalog.ListDatasetsResponse"),
			**kwargs,
		)

	def ListOrganizationDatasets(self, *args, ctx, request, **kwargs):
		return self._make_request(
			url="/twirp/catalog.CatalogService/ListOrganizationDatasets",
			ctx=ctx,
			request=request,
			response_obj=_sym_db.GetSymbol("catalog.ListOrganizationDatasetsResponse"),
			**kwargs,
		)

	def ListGroupDatasets(self, *args, ctx, request, **kwargs):
		return self._make_request(
			url="/twirp/catalog.CatalogService/ListGroupDatasets",
			ctx=ctx,
			request=request,
			response_obj=_sym_db.GetSymbol("catalog.ListGroupDatasetsResponse"),
			**kwargs,
		)

	def GetDataset(self, *args, ctx, request, **kwargs):
		return self._make_request(
			url="/twirp/catalog.CatalogService/GetDataset",
			ctx=ctx,
			request=request,
			response_obj=_sym_db.GetSymbol("catalog.GetDatasetResponse"),
			**kwargs,
		)

	def UpdateDataset(self, *args, ctx, request, **kwargs):
		return self._make_request(
			url="/twirp/catalog.CatalogService/UpdateDataset",
			ctx=ctx,
			request=request,
			response_obj=_sym_db.GetSymbol("catalog.UpdateDatasetResponse"),
			**kwargs,
		)

	def UpdateProperties(self, *args, ctx, request, **kwargs):
		return self._make_request(
			url="/twirp/catalog.CatalogService/UpdateProperties",
			ctx=ctx,
			request=request,
			response_obj=_sym_db.GetSymbol("catalog.UpdatePropertiesResponse"),
			**kwargs,
		)

	def TagRevision(self, *args, ctx, request, **kwargs):
		return self._make_request(
			url="/twirp/catalog.CatalogService/TagRevision",
			ctx=ctx,
			request=request,
			response_obj=_sym_db.GetSymbol("catalog.TagRevisionResponse"),
			**kwargs,
		)

	def UntagRevision(self, *args, ctx, request, **kwargs):
		return self._make_request(
			url="/twirp/catalog.CatalogService/UntagRevision",
			ctx=ctx,
			request=request,
			response_obj=_sym_db.GetSymbol("catalog.UntagRevisionResponse"),
			**kwargs,
		)

	def ListTags(self, *args, ctx, request, **kwargs):
		return self._make_request(
			url="/twirp/catalog.CatalogService/ListTags",
			ctx=ctx,
			request=request,
			response_obj=_sym_db.GetSymbol("catalog.ListTagsResponse"),
			**kwargs,
		)

	def PruneVersion(self, *args, ctx, request, **kwargs):
		return self._make_request(
			url="/twirp/catalog.CatalogService/PruneVersion",
			ctx=ctx,
			request=request,
			response_obj=_sym_db.GetSymbol("catalog.PruneVersionResponse"),
			**kwargs,
		)

	def PruneRevision(self, *args, ctx, request, **kwargs):
		return self._make_request(
			url="/twirp/catalog.CatalogService/PruneRevision",
			ctx=ctx,
			request=request,
			response_obj=_sym_db.GetSymbol("catalog.PruneRevisionResponse"),
			**kwargs,
		)

	def Healthz(self, *args, ctx, request, **kwargs):
		return self._make_request(
			url="/twirp/catalog.CatalogService/Healthz",
			ctx=ctx,
			request=request,
			response_obj=_sym_db.GetSymbol("catalog.Empty"),
			**kwargs,
		)