fix: detect full driver name before configuring engine (#149)

chore: detect full driver name before configuring engine
litestar-org · Apr 9, 2024 · 9d3da13 · 9d3da13
1 parent 4f3e29e
commit 9d3da13
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/app/config/base.py b/src/app/config/base.py
@@ -70,7 +70,7 @@ def engine(self) -> AsyncEngine:
     def get_engine(self) -> AsyncEngine:
         if self._engine_instance is not None:
             return self._engine_instance
-        if self.URL.startswith("postgres"):
+        if self.URL.startswith("postgresql+asyncpg"):
             engine = create_async_engine(
                 url=self.URL,
                 future=True,
@@ -128,7 +128,7 @@ def decoder(bin_value: bytes) -> Any:
                         format="binary",
                     ),
                 )
-        elif self.URL.startswith("sqlite"):
+        elif self.URL.startswith("sqlite+aiosqlite"):
             engine = create_async_engine(
                 url=self.URL,
                 future=True,