fix: create parent dirs from the real directory of the save file

telebot/asyncio_storage/pickle_storage.py

@@ -74,8 +74,9 @@ def create_dir(self):
         """
         Create directory .save-handlers.
         """
-        dirs, filename = os.path.split(self.file_path)
-        os.makedirs(dirs, exist_ok=True)
+        dirs = os.path.dirname(self.file_path)
+        if dirs:
+            os.makedirs(dirs, exist_ok=True)
         if not os.path.isfile(self.file_path):
             with open(self.file_path, "wb") as file:
                 pickle.dump({}, file)

telebot/handler_backends.py

@@ -99,8 +99,9 @@ def load_handlers(self, filename=None, del_file_after_loading=True):
 
     @staticmethod
     def dump_handlers(handlers, filename, file_mode="wb"):
-        dirs = filename.rsplit('/', maxsplit=1)[0]
-        os.makedirs(dirs, exist_ok=True)
+        dirs = os.path.dirname(filename)
+        if dirs:
+            os.makedirs(dirs, exist_ok=True)
 
         with open(filename + ".tmp", file_mode) as file:
             if apihelper.CUSTOM_SERIALIZER is None:

telebot/storage/pickle_storage.py

@@ -63,8 +63,9 @@ def create_dir(self):
         """
         Create directory .save-handlers.
         """
-        dirs, filename = os.path.split(self.file_path)
-        os.makedirs(dirs, exist_ok=True)
+        dirs = os.path.dirname(self.file_path)
+        if dirs:
+            os.makedirs(dirs, exist_ok=True)
         if not os.path.isfile(self.file_path):
             with open(self.file_path, "wb") as file:
                 pickle.dump({}, file)

tests/test_handler_backends.py

@@ -285,3 +285,11 @@ def start(message):
 
     telegram_bot.process_new_updates([update_type])
     assert update_type.message.text == 'entered start'
+
+
+def test_dump_handlers_without_directory(tmp_path, monkeypatch):
+    monkeypatch.chdir(tmp_path)
+    FileHandlerBackend.dump_handlers({1: ['handler']}, 'step.save')
+
+    assert os.path.isfile('step.save')
+    assert not os.path.isdir('step.save')

tests/test_pickle_storage.py

@@ -0,0 +1,22 @@
+import os
+
+import pytest
+
+from telebot.storage import StatePickleStorage
+
+
+def test_state_pickle_storage_accepts_filename_without_directory(tmp_path, monkeypatch):
+    monkeypatch.chdir(tmp_path)
+    StatePickleStorage(file_path="states.pkl")
+
+    assert os.path.isfile("states.pkl")
+
+
+def test_async_state_pickle_storage_accepts_filename_without_directory(tmp_path, monkeypatch):
+    pytest.importorskip("aiofiles")
+    from telebot.asyncio_storage import StatePickleStorage as AsyncStatePickleStorage
+
+    monkeypatch.chdir(tmp_path)
+    AsyncStatePickleStorage(file_path="states.pkl")
+
+    assert os.path.isfile("states.pkl")