Make utility accept Path-like objects.

bluesky · Nov 5, 2024 · 2aed6f4 · 2aed6f4
1 parent 0b1181a
commit 2aed6f4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/tiled/_tests/test_utils.py b/tiled/_tests/test_utils.py
@@ -1,3 +1,5 @@
+from pathlib import Path
+
 from ..utils import ensure_specified_sql_driver
 
 
@@ -62,6 +64,8 @@ def test_ensure_specified_sql_driver():
     # Filepaths are implicitly SQLite databases.
     # Relative path
     assert ensure_specified_sql_driver("test.db") == "sqlite+aiosqlite:///test.db"
+    # Path object
+    assert ensure_specified_sql_driver(Path("test.db")) == "sqlite+aiosqlite:///test.db"
     # Relative path anchored to .
     assert ensure_specified_sql_driver("./test.db") == "sqlite+aiosqlite:///test.db"
     # Absolute path

diff --git a/tiled/utils.py b/tiled/utils.py
@@ -740,7 +740,7 @@ def ensure_specified_sql_driver(uri: str) -> str:
     'postgresql+my_custom_driver://...' -> 'postgresql+my_custom_driver://...'
     '/path/to/file.db' -> 'sqlite+aiosqlite:////path/to/file.db'
     """
-    if not SCHEME_PATTERN.match(uri):
+    if not SCHEME_PATTERN.match(str(uri)):
         # Interpret URI as filepath.
         uri = f"sqlite+aiosqlite:///{Path(uri)}"
     scheme, rest = uri.split(":", 1)