feat: better dates + clear data before generation

tests/generate_test_data.py

@@ -1,16 +1,27 @@
 import faker
 from random import randint
+
+from sqlalchemy import MetaData
+from sqlalchemy.orm import Session
 from dooit.api import Todo, Workspace, manager
 
 manager.register_engine()
 
 f = faker.Faker()
 
 
+def delete_all_data(session: Session):
+    meta = MetaData()
+    meta.reflect(bind=session.get_bind())
+    for table in reversed(meta.sorted_tables):
+        session.execute(table.delete())
+    session.commit()
+
+
 def gen_todo(parent):
     words = randint(2, 9)
     description = " ".join(f.words(nb=words))
-    due = f.date_time()
+    due = f.date_between(start_date="-1y", end_date="+1y")
     urgency = randint(2, 5) if randint(0, 10) == 5 else 1
 
     todo = Todo(
@@ -22,6 +33,8 @@ def gen_todo(parent):
 
     if isinstance(parent, Todo):
         todo.parent_todo = parent
+        todo.due = None
+        todo.urgency = 0
     else:
         todo.parent_workspace = parent
 
@@ -43,6 +56,8 @@ def gen_workspace(parent=None):
     return workspace
 
 
+delete_all_data(manager.session)
+
 w1 = gen_workspace()
 w1_childs = [gen_workspace(w1) for _ in range(5)]