Frame13/ayon-core
1
0
Fork 1
mirror of https://github.com/ynput/ayon-core.git synced 2026-01-01 08:24:53 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

use query functions in avalon rest api calls

Browse source
This commit is contained in:
Jakub Trllo 2022-07-01 09:16:51 +02:00
parent 91e5f5aa27
commit b35d8886e2
1 changed files with 13 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

37
openpype/modules/avalon_apps/rest_api.py
View file

@ -5,7 +5,12 @@ from bson.objectid import ObjectId
from aiohttp.web_response import Response from aiohttp.web_response import Response
from openpype.pipeline import AvalonMongoDB from openpype.client import (
get_projects,
get_project,
get_assets,
get_asset_by_name,
)
from openpype_modules.webserver.base_routes import RestApiEndpoint from openpype_modules.webserver.base_routes import RestApiEndpoint
@ -14,19 +19,13 @@ class _RestApiEndpoint(RestApiEndpoint):
self.resource = resource self.resource = resource
super(_RestApiEndpoint, self).__init__() super(_RestApiEndpoint, self).__init__()
@property
def dbcon(self):
return self.resource.dbcon
class AvalonProjectsEndpoint(_RestApiEndpoint): class AvalonProjectsEndpoint(_RestApiEndpoint):
async def get(self) -> Response: async def get(self) -> Response:
output = [] output = [
for project_name in self.dbcon.database.collection_names(): project_doc
project_doc = self.dbcon.database[project_name].find_one({ for project_doc in get_projects()
"type": "project" ]
})
output.append(project_doc)
return Response( return Response(
status=200, status=200,
body=self.resource.encode(output), body=self.resource.encode(output),
@ -36,9 +35,7 @@ class AvalonProjectsEndpoint(_RestApiEndpoint):
class AvalonProjectEndpoint(_RestApiEndpoint): class AvalonProjectEndpoint(_RestApiEndpoint):
async def get(self, project_name) -> Response: async def get(self, project_name) -> Response:
project_doc = self.dbcon.database[project_name].find_one({ project_doc = get_project(project_name)
"type": "project"
})
if project_doc: if project_doc:
return Response( return Response(
status=200, status=200,
@ -53,9 +50,7 @@ class AvalonProjectEndpoint(_RestApiEndpoint):
class AvalonAssetsEndpoint(_RestApiEndpoint): class AvalonAssetsEndpoint(_RestApiEndpoint):
async def get(self, project_name) -> Response: async def get(self, project_name) -> Response:
asset_docs = list(self.dbcon.database[project_name].find({ asset_docs = list(get_assets(project_name))
"type": "asset"
}))
return Response( return Response(
status=200, status=200,
body=self.resource.encode(asset_docs), body=self.resource.encode(asset_docs),
@ -65,10 +60,7 @@ class AvalonAssetsEndpoint(_RestApiEndpoint):
class AvalonAssetEndpoint(_RestApiEndpoint): class AvalonAssetEndpoint(_RestApiEndpoint):
async def get(self, project_name, asset_name) -> Response: async def get(self, project_name, asset_name) -> Response:
asset_doc = self.dbcon.database[project_name].find_one({ asset_doc = get_asset_by_name(project_name, asset_name)
"type": "asset",
"name": asset_name
})
if asset_doc: if asset_doc:
return Response( return Response(
status=200, status=200,
@ -88,9 +80,6 @@ class AvalonRestApiResource:
self.module = avalon_module self.module = avalon_module
self.server_manager = server_manager self.server_manager = server_manager
self.dbcon = AvalonMongoDB()
self.dbcon.install()
self.prefix = "/avalon" self.prefix = "/avalon"
self.endpoint_defs = ( self.endpoint_defs = (