snapshot of https://github.com/ActivityWatch/aw-core.git (master, 2026-09-16; blobs > 131072B omitted)

aw_datastore/__init__.py

@@ -0,0 +1,19 @@
+from typing import Callable, Dict
+
+from . import storages
+from .datastore import Datastore
+from .migration import check_for_migration
+
+
+def get_storage_methods() -> Dict[str, Callable[..., storages.AbstractStorage]]:
+    from .storages import MemoryStorage, PeeweeStorage, SqliteStorage
+
+    methods: Dict[str, Callable[..., storages.AbstractStorage]] = {
+        PeeweeStorage.sid: PeeweeStorage,
+        MemoryStorage.sid: MemoryStorage,
+        SqliteStorage.sid: SqliteStorage,
+    }
+    return methods
+
+
+__all__ = ["Datastore", "get_storage_methods", "check_for_migration"]