Add check modified time for path

src/aiogram_dialog/context/media_storage.py

@@ -1,3 +1,4 @@
+import os
 from typing import Optional
 
 from aiogram.types import ContentType
@@ -19,7 +20,21 @@ async def get_media_id(
     ) -> Optional[MediaId]:
         if not path and not url:
             return None
-        return self.cache.get((path, url, type))
+        cached = self.cache.get((path, url, type))
+        if cached is None:
+            return None
+        if cached[1] is not None:
+            mtime = self._get_file_mtime(path)
+            if mtime is not None and mtime != cached[1]:
+                return None
+        return cached[0]
+
+    def _get_file_mtime(self, path: Optional[str]) -> Optional[float]:
+        if not path:
+            return None
+        if not os.path.exists(path):  # noqa: PTH110
+            return None
+        return os.path.getmtime(path)  # noqa: PTH204
 
     async def save_media_id(
             self,
@@ -30,4 +45,4 @@ async def save_media_id(
     ) -> None:
         if not path and not url:
             return
-        self.cache[(path, url, type)] = media_id
+        self.cache[(path, url, type)] = (media_id, self._get_file_mtime(path))

tests/widgets/media/test_media_storage.py

@@ -0,0 +1,50 @@
+import asyncio
+import os
+import tempfile
+
+import pytest
+from aiogram.enums import ContentType
+
+from aiogram_dialog.context.media_storage import MediaIdStorage
+
+
+@pytest.mark.asyncio
+async def test_get_media_id():
+    manager = MediaIdStorage()
+    with tempfile.TemporaryDirectory() as d:
+        filename = os.path.join(d, "file_test")  # noqa: PTH118
+        media_id = await manager.get_media_id(
+            filename,
+            None,
+            ContentType.DOCUMENT,
+        )
+        assert media_id is None
+
+        with open(filename, "w") as file:  # noqa: PTH123
+            file.write("test1")
+
+        await manager.save_media_id(
+            filename,
+            None,
+            ContentType.DOCUMENT,
+            "test1",
+        )
+
+        media_id = await manager.get_media_id(
+            filename,
+            None,
+            ContentType.DOCUMENT,
+        )
+        assert media_id == "test1"
+
+        await asyncio.sleep(0.1)
+
+        with open(filename, "w") as file:  # noqa: PTH123
+            file.write("test2")
+
+        media_id = await manager.get_media_id(
+            filename,
+            None,
+            ContentType.DOCUMENT,
+        )
+        assert media_id is None