checkpoint: demo: create_fake_timesheet

Author: clstaudt

app/demo.py

@@ -25,6 +25,8 @@
     BankAccount,
     Invoice,
     InvoiceItem,
+    Timesheet,
+    TimeTrackingItem,
 )
 from tuttle import rendering
 
@@ -129,6 +131,46 @@ def invoice_number_counting():
 invoice_number_counter = invoice_number_counting()
 
 
+def create_fake_timesheet(
+    project: Project,
+    fake: faker.Faker,
+) -> Timesheet:
+    """
+    Create a fake timesheet object with random values.
+
+    Args:
+    project (Project): The project associated with the timesheet.
+    fake (faker.Faker): An instance of the Faker class to generate random values.
+
+    Returns:
+    Timesheet: A fake timesheet object.
+    """
+    timesheet = Timesheet(
+        title=fake.bs(),
+        comment=fake.paragraph(nb_sentences=2),
+        date=datetime.date.today(),
+        project=project,
+    )
+    number_of_items = fake.random_int(min=1, max=5)
+    for _ in range(number_of_items):
+        unit = fake.random_element(elements=("hours", "days"))
+        if unit == "hours":
+            unit_price = abs(round(numpy.random.normal(50, 20), 2))
+        elif unit == "days":
+            unit_price = abs(round(numpy.random.normal(400, 200), 2))
+        time_tracking_item = TimeTrackingItem(
+            timesheet=timesheet,
+            begin=fake.date_time_this_year(before_now=True, after_now=False),
+            end=fake.date_time_this_year(before_now=True, after_now=False),
+            duration=datetime.timedelta(hours=fake.random_int(min=1, max=8)),
+            title=f"{fake.bs()} for #{project.tag}",
+            tag=project.tag,
+            description=fake.paragraph(nb_sentences=2),
+        )
+        timesheet.items.append(time_tracking_item)
+    return timesheet
+
+
 def create_fake_invoice(
     project: Project,
     user: User,
@@ -146,7 +188,7 @@ def create_fake_invoice(
     """
     invoice_number = next(invoice_number_counter)
     invoice = Invoice(
-        number=invoice_number,
+        number=str(invoice_number),  # TODO: replace with generated number
         date=datetime.date.today(),
         sent=fake.pybool(),
         paid=fake.pybool(),

app/invoicing/intent.py

@@ -117,10 +117,21 @@ def create_invoice(
 
             if render:
                 # TODO: render timesheet
-
+                user = self._user_data_source.get_user()
+                try:
+                    rendering.render_timesheet(
+                        user=user,
+                        timesheet=timesheet,
+                        out_dir=Path.home() / ".tuttle" / "Timesheets",
+                    )
+                    logger.info(f"✅ rendered timesheet for {project.title}")
+                except Exception as ex:
+                    logger.error(
+                        f"❌ Error rendering timesheet for {project.title}: {ex}"
+                    )
+                    logger.exception(ex)
                 # render invoice
                 try:
-                    user = self._user_data_source.get_user()
                     rendering.render_invoice(
                         user=user,
                         invoice=invoice,

tuttle/model.py

@@ -439,9 +439,7 @@ class TimeTrackingItem(SQLModel, table=True):
     timesheet: Optional["Timesheet"] = Relationship(back_populates="items")
     #
     begin: datetime.datetime = Field(description="Start time of the time interval.")
-    end: Optional[datetime.datetime] = Field(
-        description="End time of the time interval."
-    )
+    end: datetime.datetime = Field(description="End time of the time interval.")
     duration: datetime.timedelta = Field(description="Duration of the time interval.")
     title: str = Field(description="A short description of the time interval.")
     tag: str = Field(