Add basic concept

Author: erwindouna

scheduler.py

@@ -35,8 +35,7 @@ def start(
         _LOGGER.info("Starting the scheduler, with schedule: %s", self._schedule)
         if self._schedule is None:
             _LOGGER.warning("No schedule set, not starting the scheduler")
-            self._schedule = "*/1 * * * *"  # TODO: for debugging purposes
-            # return
+            return
 
         if self._import_job:
             self._scheduler.remove_job(self._import_job.id)
@@ -62,6 +61,28 @@ def run_import() -> None:
         self._scheduler.start()
         _LOGGER.info("Scheduler started")
 
+    def set_schedule(self, schedule: str) -> None:
+        """Set the schedule for the import job."""
+        self._schedule = schedule
+        _LOGGER.info("Scheduler schedule set to: %s", self._schedule)
+
+        if not schedule:
+            _LOGGER.info("Disabling the scheduler")
+            if self._import_job:
+                self._scheduler.remove_job(self._import_job.id)
+                _LOGGER.info("Scheduler job removed")
+            else:
+                _LOGGER.warning("No import job to remove")
+            self._import_job = None
+            return
+
+        if self._import_job:
+            self._scheduler.reschedule_job(
+                self._import_job.id,
+                trigger=CronTrigger.from_crontab(self._schedule),
+            )
+            _LOGGER.info("Scheduler job rescheduled to: %s", self._schedule)
+
     def stop(self) -> None:
         """Stop the scheduler."""
         if not self._scheduler.running:

templates/configuration.html

@@ -32,6 +32,32 @@ <h5 class="modal-title" id="resetModalLabel">Confirm Reset</h5>
     </div>
 </div>
 
+<div class="row">
+    <div class="col-md-12">
+        <h3>Scheduler</h3>
+        <p>Set a scheduler to automatically import your transactions.</p>
+        <hr>
+        <form action="/set-schedule" method="post">
+            <div class="mb-3">
+                <label for="schedule" class="form-label">Scheduler</label>
+                <select class="form-select" id="schedule" name="schedule">
+                    <option value="">Disabled</option>
+                    <option value="*/1 * * * *">Every minute</option>
+                    <option value="*/5 * * * *">Every 5 minutes</option>
+                    <option value="*/10 * * * *">Every 10 minutes</option>
+                    <option value="*/15 * * * *">Every 15 minutes</option>
+                    <option value="*/30 * * * *">Every 30 minutes</option>
+                    <option value="0 * * * *">Every hour</option>
+                    <option value="0 0 * * *">Every day at midnight</option>
+                    <option value="0 0 * * 0">Every week at midnight</option>
+                </select>
+            </div>
+            <button type="submit" class="btn btn-primary">Save Scheduler</button>
+        </form>
+    </div>
+</div>
+<hr>
+
 <div class="row">
     <div class="col-md-6">
         <div class="card mb-4">
@@ -107,4 +133,4 @@ <h5 class="modal-title" id="resetModalLabel">Confirm Reset</h5>
     </div>
 </div>
 
-{% endblock %}
+{% endblock %}

truelayer2firefly.py

@@ -113,6 +113,14 @@ async def get_firefly_client() -> FireflyClient:
     return client
 
 
+async def get_scheduler() -> Scheduler:
+    """Get the scheduler from the application state."""
+    scheduler = app.state.scheduler
+    if not scheduler:
+        raise RuntimeError("Scheduler is not initialized.")
+    return scheduler
+
+
 @app.get("/", response_class=HTMLResponse)
 async def index(request: Request):
     """Render the index page."""
@@ -365,6 +373,27 @@ async def reset_configuration(request: Request):
     return RedirectResponse(str(request.url_for("index")), status_code=302)
 
 
+@app.post("/set-schedule")
+async def set_schedule(
+    request: Request,
+    schedule: str = Form(...),
+    scheduler: Scheduler = Depends(get_scheduler),
+) -> RedirectResponse:
+    """Set the import schedule."""
+    _LOGGER.info("Setting schedule to %s", schedule)
+    config.set("import_schedule", schedule)
+
+    try:
+        scheduler.set_schedule(schedule)
+    except Exception as e:
+        _LOGGER.error("Error setting schedule: %s", e)
+        return JSONResponse(
+            status_code=500,
+            content={"error": "Error setting schedule", "details": str(e)},
+        )
+    return RedirectResponse(str(request.url_for("index")), status_code=302)
+
+
 app.add_exception_handler(
     TrueLayer2FireflyAuthorizationError, truelayer_authorization_error_handler
 )