No need for sqlite_ystore_factory()

Author: davidbrochart

jupyter_server_ydoc/app.py

@@ -9,7 +9,7 @@
 from traitlets import Float, Int, Type, Unicode, observe
 from ypy_websocket.ystore import BaseYStore, SQLiteYStore  # type: ignore
 
-from .handlers import YDocRoomIdHandler, YDocWebSocketHandler, sqlite_ystore_factory
+from .handlers import JupyterSQLiteYStore, YDocRoomIdHandler, YDocWebSocketHandler
 
 
 class YDocExtension(ExtensionApp):
@@ -41,7 +41,7 @@ class YDocExtension(ExtensionApp):
     )
 
     ystore_class = Type(
-        default_value=sqlite_ystore_factory(),
+        default_value=JupyterSQLiteYStore,
         klass=BaseYStore,
         config=True,
         help="""The YStore class to use for storing Y updates. Defaults to JupyterSQLiteYStore,

jupyter_server_ydoc/handlers.py

@@ -29,17 +29,9 @@ class JupyterTempFileYStore(TempFileYStore):
     prefix_dir = "jupyter_ystore_"
 
 
-def sqlite_ystore_factory(
-    db_path: str = ".jupyter_ystore.db", document_ttl: Optional[int] = None
-) -> Type[SQLiteYStore]:
-    _db_path = db_path
-    _document_ttl = document_ttl
-
-    class JupyterSQLiteYStore(SQLiteYStore):
-        db_path = _db_path
-        document_ttl = _document_ttl
-
-    return JupyterSQLiteYStore
+class JupyterSQLiteYStore(SQLiteYStore):
+    db_path = ".jupyter_ystore.db"
+    document_ttl = None
 
 
 class DocumentRoom(YRoom):