mirror of
https://github.com/ynput/ayon-core.git
synced 2025-12-25 05:14:40 +01:00
131 lines
3.5 KiB
Python
131 lines
3.5 KiB
Python
import json
|
|
import datetime
|
|
|
|
from bson.objectid import ObjectId
|
|
|
|
from aiohttp.web_response import Response
|
|
|
|
from openpype.client import (
|
|
get_projects,
|
|
get_project,
|
|
get_assets,
|
|
get_asset_by_name,
|
|
)
|
|
from openpype_modules.webserver.base_routes import RestApiEndpoint
|
|
|
|
|
|
class _RestApiEndpoint(RestApiEndpoint):
|
|
def __init__(self, resource):
|
|
self.resource = resource
|
|
super(_RestApiEndpoint, self).__init__()
|
|
|
|
|
|
class AvalonProjectsEndpoint(_RestApiEndpoint):
|
|
async def get(self) -> Response:
|
|
output = [
|
|
project_doc
|
|
for project_doc in get_projects()
|
|
]
|
|
return Response(
|
|
status=200,
|
|
body=self.resource.encode(output),
|
|
content_type="application/json"
|
|
)
|
|
|
|
|
|
class AvalonProjectEndpoint(_RestApiEndpoint):
|
|
async def get(self, project_name) -> Response:
|
|
project_doc = get_project(project_name)
|
|
if project_doc:
|
|
return Response(
|
|
status=200,
|
|
body=self.resource.encode(project_doc),
|
|
content_type="application/json"
|
|
)
|
|
return Response(
|
|
status=404,
|
|
reason="Project name {} not found".format(project_name)
|
|
)
|
|
|
|
|
|
class AvalonAssetsEndpoint(_RestApiEndpoint):
|
|
async def get(self, project_name) -> Response:
|
|
asset_docs = list(get_assets(project_name))
|
|
return Response(
|
|
status=200,
|
|
body=self.resource.encode(asset_docs),
|
|
content_type="application/json"
|
|
)
|
|
|
|
|
|
class AvalonAssetEndpoint(_RestApiEndpoint):
|
|
async def get(self, project_name, asset_name) -> Response:
|
|
asset_doc = get_asset_by_name(project_name, asset_name)
|
|
if asset_doc:
|
|
return Response(
|
|
status=200,
|
|
body=self.resource.encode(asset_doc),
|
|
content_type="application/json"
|
|
)
|
|
return Response(
|
|
status=404,
|
|
reason="Asset name {} not found in project {}".format(
|
|
asset_name, project_name
|
|
)
|
|
)
|
|
|
|
|
|
class AvalonRestApiResource:
|
|
def __init__(self, avalon_module, server_manager):
|
|
self.module = avalon_module
|
|
self.server_manager = server_manager
|
|
|
|
self.prefix = "/avalon"
|
|
|
|
self.endpoint_defs = (
|
|
(
|
|
"GET",
|
|
"/projects",
|
|
AvalonProjectsEndpoint(self)
|
|
),
|
|
(
|
|
"GET",
|
|
"/projects/{project_name}",
|
|
AvalonProjectEndpoint(self)
|
|
),
|
|
(
|
|
"GET",
|
|
"/projects/{project_name}/assets",
|
|
AvalonAssetsEndpoint(self)
|
|
),
|
|
(
|
|
"GET",
|
|
"/projects/{project_name}/assets/{asset_name}",
|
|
AvalonAssetEndpoint(self)
|
|
)
|
|
)
|
|
|
|
self.register()
|
|
|
|
def register(self):
|
|
for methods, url, endpoint in self.endpoint_defs:
|
|
final_url = self.prefix + url
|
|
self.server_manager.add_route(
|
|
methods, final_url, endpoint.dispatch
|
|
)
|
|
|
|
@staticmethod
|
|
def json_dump_handler(value):
|
|
if isinstance(value, datetime.datetime):
|
|
return value.isoformat()
|
|
if isinstance(value, ObjectId):
|
|
return str(value)
|
|
raise TypeError(value)
|
|
|
|
@classmethod
|
|
def encode(cls, data):
|
|
return json.dumps(
|
|
data,
|
|
indent=4,
|
|
default=cls.json_dump_handler
|
|
).encode("utf-8")
|