diff --git a/src/teilchensammler_cli/__init__.py b/src/teilchensammler_cli/__init__.py index 18235ac..85148c0 100644 --- a/src/teilchensammler_cli/__init__.py +++ b/src/teilchensammler_cli/__init__.py @@ -2,7 +2,7 @@ This file exists so the directory becomes a package. """ -from .main import main +from teilchensammler_cli.main import main if __name__ == "__main__": main() diff --git a/src/teilchensammler_cli/main.py b/src/teilchensammler_cli/main.py index 168d584..d8494bd 100644 --- a/src/teilchensammler_cli/main.py +++ b/src/teilchensammler_cli/main.py @@ -16,6 +16,8 @@ from textual.widgets import ( Input, Static, ) +from tortoise import Tortoise, fields +from tortoise.models import Model @final @@ -98,6 +100,10 @@ class AddInventoryScreen(Screen[None]): @final class SammlerApp(App[None]): async def on_mount(self) -> None: + await Tortoise.init( + db_url="sqlite://db.sqlite3", modules={"models": ["teilchensammler_cli"]} + ) + await Tortoise.generate_schemas() _ = self.push_screen(AddInventoryScreen())