32 lines
696 B
Python
32 lines
696 B
Python
from logging import getLogger
|
|
from os import path
|
|
from typing import Optional
|
|
from urllib.parse import urlparse
|
|
|
|
from ..utils import ConversionContext
|
|
from .base import BaseClient
|
|
|
|
logger = getLogger(__name__)
|
|
|
|
|
|
class FileClient(BaseClient):
|
|
protocol = "file://"
|
|
|
|
root: str
|
|
|
|
def __init__(self, root: str):
|
|
self.root = root
|
|
|
|
def download(
|
|
self,
|
|
_conversion: ConversionContext,
|
|
_name: str,
|
|
uri: str,
|
|
format: Optional[str] = None,
|
|
dest: Optional[str] = None,
|
|
**kwargs,
|
|
) -> str:
|
|
parts = urlparse(uri)
|
|
logger.info("loading model from: %s", parts.path)
|
|
return path.join(self.root, parts.path)
|