feat(dataset): add "get_url()" method to "Data"

PR Closed: #789

tensorbay/dataset/data.py

@@ -14,8 +14,8 @@
 from http.client import HTTPResponse
 from string import printable
 from typing import Any, Callable, Dict, Optional, Type, TypeVar, Union
-from urllib.parse import quote
-from urllib.request import urlopen
+from urllib.parse import quote, urljoin
+from urllib.request import pathname2url, urlopen
 
 from _io import BufferedReader
 
@@ -213,6 +213,15 @@ def dumps(self) -> Dict[str, Any]:
         """
         return super()._dumps()
 
+    def get_url(self) -> str:
+        """Return the url of the local data file.
+
+        Returns:
+            The url of the local data.
+
+        """
+        return urljoin("file:", pathname2url(os.path.abspath(self.path)))
+
 
 class RemoteData(DataBase):
     """RemoteData is a combination of a specific tensorbay dataset file and its label.

tensorbay/dataset/tests/test_data.py

@@ -3,6 +3,8 @@
 # Copyright 2021 Graviti. Licensed under MIT License.
 #
 
+from pathlib import Path
+
 import pytest
 
 from ..data import Data, RemoteData
@@ -37,6 +39,13 @@ def test_target_remote_path(self):
         data_object.target_remote_path = target_remote_path
         assert data_object.target_remote_path == target_remote_path
 
+    def test_get_url(self):
+        local_relative_path = Path(__file__).relative_to(Path.cwd())
+        data = Data(str(local_relative_path))
+        assert data.get_url() == local_relative_path.resolve().as_uri()
+        data = Data(__file__)
+        assert data.get_url() == Path(__file__).resolve().as_uri()
+
 
 class TestRemoteData:
     def test_init(self):