23 lines
933 B
Python
23 lines
933 B
Python
|
from core.plugin.impl.base import BasePluginClient
|
||
|
|
|
||
|
|
|
||
|
|
class PluginAssetManager(BasePluginClient):
|
||
|
|
def fetch_asset(self, tenant_id: str, id: str) -> bytes:
|
||
|
|
"""
|
||
|
|
Fetch an asset by id.
|
||
|
|
"""
|
||
|
|
response = self._request(method="GET", path=f"plugin/{tenant_id}/asset/{id}")
|
||
|
|
if response.status_code != 200:
|
||
|
|
raise ValueError(f"can not found asset {id}")
|
||
|
|
return response.content
|
||
|
|
|
||
|
|
def extract_asset(self, tenant_id: str, plugin_unique_identifier: str, filename: str) -> bytes:
|
||
|
|
response = self._request(
|
||
|
|
method="GET",
|
||
|
|
path=f"plugin/{tenant_id}/extract-asset/",
|
||
|
|
params={"plugin_unique_identifier": plugin_unique_identifier, "file_path": filename},
|
||
|
|
)
|
||
|
|
if response.status_code != 200:
|
||
|
|
raise ValueError(f"can not found asset {plugin_unique_identifier}, {str(response.status_code)}")
|
||
|
|
return response.content
|