WIP: demo invoicing workflow

Author: clstaudt

app/Tuttle.py

@@ -1,5 +1,6 @@
 from loguru import logger
 from textwrap import dedent
+from pathlib import Path
 
 import flet
 from flet import (
@@ -25,6 +26,7 @@
 import widgets
 
 from tuttle.controller import Controller
+from tuttle.preferences import Preferences
 from tuttle.model import (
     Contact,
     Contract,
@@ -187,15 +189,15 @@ def update_content(self):
 
         projects = self.app.con.query(Project)
 
-        project_select = Dropdown(
+        self.project_select = Dropdown(
             label="Project",
             hint_text="Select the project",
-            options=[dropdown.Option(project.title) for project in projects],
+            options=[dropdown.Option(project.tag) for project in projects],
             autofocus=True,
         )
 
-        date_from_select = widgets.DateSelector()
-        date_to_select = widgets.DateSelector()
+        self.date_from_select = widgets.DateSelector()
+        self.date_to_select = widgets.DateSelector()
 
         self.main_column.controls = [
             Row(
@@ -236,19 +238,19 @@ def update_content(self):
             Row(
                 [
                     Icon(icons.WORK),
-                    project_select,
+                    self.project_select,
                 ]
             ),
             Row(
                 [
                     # Icon(icons.DATE_RANGE),
-                    date_from_select,
+                    self.date_from_select,
                     Icon(icons.ARROW_FORWARD),
-                    date_to_select,
+                    self.date_to_select,
                     TextButton(
                         "Select",
                         on_click=lambda event: logger.info(
-                            f"Selected date range: {date_from_select.get_date()} -> {date_to_select.get_date()}"
+                            f"Selected date range: {self.date_from_select.get_date()} -> {self.date_to_select.get_date()}"
                         ),
                     ),
                 ]
@@ -258,19 +260,33 @@ def update_content(self):
                     ElevatedButton(
                         "Generate invoice",
                         icon=icons.EDIT_NOTE,
-                        # on_click=self.add_demo_data,
+                        on_click=self.generate_invoices_clicked,
                     ),
                 ]
             ),
         ]
         self.update()
 
+    def generate_invoices_clicked(self, event):
+        """Generate invoices for the selected project and date range."""
+        logger.info("Generate invoices clicked")
+        self.app.con.billing(
+            project_tags=[self.project_select.value],
+            period_start=self.date_from_select.get_date(),
+            period_end=self.date_to_select.get_date(),
+            timetracking_method="file_calendar",
+            calendar_file_path=None,
+        )
+
 
 def main(page: Page):
 
     con = Controller(
         in_memory=True,
         verbose=False,
+        preferences=Preferences(
+            invoice_dir=Path("Invoices"),
+        ),
     )
 
     app = App(

tuttle/controller.py

@@ -12,16 +12,20 @@
 from loguru import logger
 
 from . import model, timetracking, dataviz, rendering, invoicing, calendar, cloud
+from .preferences import Preferences
 
 
 class Controller:
     """The application controller."""
 
-    def __init__(self, home_dir=None, verbose=False, in_memory=False):
-        if home_dir is None:
+    def __init__(self, preferences: Preferences, verbose=False, in_memory=False):
+        self.preferences = preferences
+        # set home directory
+        if preferences.home_dir is None:
             self.home = Path.home() / ".tuttle"
         else:
-            self.home = Path(home_dir)
+            self.home = self.preferences.home_dir
+        # make directories
         if not os.path.exists(self.home):
             os.mkdir(self.home)
         if in_memory:
@@ -204,19 +208,25 @@ def duration_to_revenue(
     def billing(
         self,
         project_tags,
-        out_dir,
         period_start,
         period_end=None,
-        timetracking_method="calendar",
+        timetracking_method="cloud_calendar",
+        calendar_file_path=None,
     ):
         """Generate time sheets and invoices for a given period"""
+        out_dir = self.home / self.preferences.invoice_dir
         # TODO: read method from user settings
-        if timetracking_method == "calendar":
+        if timetracking_method == "cloud_calendar":
             timetracking_calendar = calendar.ICloudCalendar(
                 icloud=cloud.login_iCloud(user_name=self.user.icloud_account.user_name),
                 # TODO: read from user settings
                 name="TimeTracking",
             )
+        elif timetracking_method == "file_calendar":
+            timetracking_calendar = calendar.FileCalendar(
+                path=calendar_file_path,
+                name=calendar_file_path.stem,
+            )
         else:
             raise ValueError(f"unsupported time tracking method: {timetracking_method}")
 

tuttle/preferences.py

@@ -0,0 +1,10 @@
+from dataclasses import dataclass
+from pathlib import Path
+
+
+@dataclass
+class Preferences:
+    """Settings for the application."""
+
+    home_dir: Path = None
+    invoice_dir: Path = Path("Invoices")