tested database setup

Author: clstaudt

tuttle/app.py

@@ -25,6 +25,9 @@ def get_session(self):
         return sqlmodel.Session(self.db_engine)
 
     def clear_database(self):
+        """
+        Delete the database and rebuild database model.
+        """
         self.db_path.unlink()
         self.db_engine = sqlmodel.create_engine(f"sqlite:///{self.db_path}", echo=True)
         sqlmodel.SQLModel.metadata.create_all(self.db_engine)
@@ -34,5 +37,27 @@ def get_user(self):
             user = session.exec(sqlmodel.select(model.User)).one()
         return user
 
+    def store_all(self, entities):
+        """Store a collection of entities in the database."""
+        with self.get_session() as session:
+            for entity in entities:
+                session.add(entity)
+            session.commit()
+
+    def retrieve_all(self, entity_type):
+        with self.get_session() as session:
+            entities = session.exec(
+                sqlmodel.select(entity_type),
+            ).all()
+            return entities
+
+    @property
+    def contracts(self):
+        with self.get_session() as session:
+            contracts = session.exec(
+                sqlmodel.select(model.Contract),
+            ).all()
+            return contracts
+
 
 the_app = App()

tuttle/model.py

@@ -108,7 +108,7 @@ class User(SQLModel, table=True):
     bank_account_id: Optional[int] = Field(default=None, foreign_key="bankaccount.id")
     bank_account: Optional["BankAccount"] = Relationship(back_populates="user")
     # TODO: path to logo image
-    logo: str
+    logo: Optional[str]
 
 
 class ICloudAccount(SQLModel, table=True):
@@ -135,7 +135,7 @@ class BankAccount(SQLModel, table=True):
     name: str
     IBAN: str
     BIC: str
-    username: str  # online banking user name
+    # username: str  # online banking user name
     user: User = Relationship(back_populates="bank_account")
 
 
@@ -192,7 +192,7 @@ class Contract(SQLModel, table=True):
     )
     currency: str  # TODO: currency representation
     unit: TimeUnit = Field(sa_column=sqlalchemy.Column(sqlalchemy.Enum(TimeUnit)))
-    volume: int = Field(
+    volume: Optional[int] = Field(
         description="Number of units agreed on",
     )
     term_of_payment: Optional[int] = Field(