fix db upsert bug in new_month

Author: tensorturtle

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "fob"
 # Remember to update the version number in 'utils/utils.py' manually also.
-version = "0.2.2"
+version = "0.2.3"
 description = "Focus Blocks: Simple and elegant daily time management tool."
 readme = "README.md"
 authors = [{ name = "tensorturtle", email = "[email protected]" }]

src/fob/commands/new_month.py

@@ -31,7 +31,7 @@ def write_new_month(data: MonthBlockData, db: TinyDBWrapper) -> None:
 
     # Overwrite the existing month data (shouldn't happen but avoids duplicates)
     q = Query()
-    db.update({
+    db.upsert({
         "year": data.year,
         "month": data.month,
         "work_days_allocated": data.work_days_allocated,

src/fob/db/wrapper.py

@@ -14,6 +14,9 @@ def insert(self, data):
         # Insert a document (dictionary) into the database
         return self.db.insert(data)
 
+    def upsert(self, fields, query):
+        return self.db.upsert(fields, query)
+
     def all(self):
         # Get all documents stored in the database
         return self.db.all()

src/fob/utils/utils.py

@@ -7,7 +7,7 @@
 from rich import print
 
 def get_version_number() -> tuple[int, int, int]:
-    return (0,2,2)
+    return (0,2,3)
 
 def get_db_path(args: Namespace) -> Path:
     '''