Merge pull request Tishka17#345 from Vladyslav49/develop

add pathlib.Path support for path in media
lubaskinc0de · Dec 26, 2023 · f381e4a · f381e4a
2 parents e71f047 + 0b31cba
commit f381e4a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/src/aiogram_dialog/api/entities/media.py b/src/aiogram_dialog/api/entities/media.py
@@ -1,5 +1,6 @@
 from dataclasses import dataclass
-from typing import Optional
+from pathlib import Path
+from typing import Optional, Union
 
 from aiogram.types import ContentType
 
@@ -22,7 +23,7 @@ def __init__(
             self,
             type: ContentType,
             url: Optional[str] = None,
-            path: Optional[str] = None,
+            path: Union[str, Path, None] = None,
             file_id: Optional[MediaId] = None,
             use_pipe: bool = False,
             **kwargs,

diff --git a/src/aiogram_dialog/widgets/media/static.py b/src/aiogram_dialog/widgets/media/static.py
@@ -1,3 +1,4 @@
+from pathlib import Path
 from typing import Optional, Union
 
 from aiogram.types import ContentType
@@ -13,7 +14,7 @@ class StaticMedia(Media):
     def __init__(
             self,
             *,
-            path: Union[Text, str, None] = None,
+            path: Union[Text, str, Path, None] = None,
             url: Union[Text, str, None] = None,
             type: ContentType = ContentType.PHOTO,
             use_pipe: bool = False,
@@ -24,7 +25,9 @@ def __init__(
         if not (url or path):
             raise ValueError("Neither url nor path are provided")
         self.type = type
-        if isinstance(path, str):
+        if isinstance(path, Path):
+            path = Const(str(path))
+        elif isinstance(path, str):
             path = Const(path)
         self.path = path
         if isinstance(url, str):