X-Git-Url: https://plomlompom.com/repos/?p=misc;a=blobdiff_plain;f=new_todo%2Ftodo.py;fp=new_todo%2Ftodo.py;h=7a5ed2dbcf7da166439302c2c251733e8031f476;hp=b7ee52b56d742ba93f754d022d80ee0dda12aaa5;hb=0cd0b92d9e261dbe75fa45001aeca74592c053f8;hpb=33a4db0a8a1eacee8c15eb42c3f4a56c239048a1

diff --git a/new_todo/todo.py b/new_todo/todo.py
index b7ee52b..7a5ed2d 100755
--- a/new_todo/todo.py
+++ b/new_todo/todo.py
@@ -80,6 +80,7 @@ class TodoDBConnection:
     def __init__(self, db_file):
         self.file = db_file
         self.conn = sql_connect(self.file.path)
+        self.conn.execute('PRAGMA foreign_keys = ON')
     
     def commit(self):
         self.file.backup()
@@ -117,6 +118,9 @@ class VersionedAttribute:
             db_conn.exec(f'REPLACE INTO {self.table_name} VALUES (?, ?, ?)',
                          (self.parent.id_, date, value))
 
+    def delete(self, db_conn):
+        db_conn.exec(f'DELETE FROM {self.table_name} WHERE template = ?', (self.parent.id_,))
+
     @property
     def _newest_datetime(self):
         return sorted(self.history.keys())[-1]
@@ -290,8 +294,19 @@ class TodoTemplate:
         self.default_effort.save(db_conn)
         self.description.save(db_conn)
 
+    def delete(self, db_conn):
+        for row in db_conn.exec('SELECT * FROM templates WHERE forked_from = ?', (self.id_,)):
+            raise HandledException('cannot delete forking reference') 
+        self.title.delete(db_conn)
+        self.default_effort.delete(db_conn)
+        self.description.delete(db_conn)
+        db_conn.exec('DELETE FROM templates WHERE id = ?', (self.id_,))
+
     def fork(self, db_conn):
-        return self.__class__(db_conn, id_=None, forked_from=self.id_, forked_at=Day.todays_date(with_time=True))
+        forked = self.__class__(db_conn, id_=None, forked_from=self.id_,
+                                forked_at=Day.todays_date(with_time=True))
+        forked.save(db_conn)
+        return forked
 
 
 
@@ -299,9 +314,9 @@ class TodoTemplate:
 
 class ParamsParser:
 
-    def __init__(self, url_query, site_cookie):
+    def __init__(self, url_query, site_cookie_dict):
         self.params = parse_qs(url_query, keep_blank_values=True)
-        self.cookie = site_cookie
+        self.cookie = site_cookie_dict
 
     def get(self, key, default):
         return self.params.get(key, [default])[0]
@@ -392,10 +407,15 @@ class TodoHandler(BaseHTTPRequestHandler):
     def do_POST_template(self):
         id_ = self.params.get('id', None)
         tmpl = TodoTemplate.by_id(self.db_conn, id_, make_if_none=True)
-        tmpl.title.set(self.postvars['title'][0])
-        tmpl.default_effort.set(float(self.postvars['default_effort'][0]))
-        tmpl.description.set(self.postvars['description'][0])
-        tmpl.save(self.db_conn)
+        if 'update' in self.postvars.keys():
+            tmpl.title.set(self.postvars['title'][0])
+            tmpl.default_effort.set(float(self.postvars['default_effort'][0]))
+            tmpl.description.set(self.postvars['description'][0])
+            tmpl.save(self.db_conn)
+        elif 'fork' in self.postvars.keys():
+            tmpl.fork(self.db_conn)
+        elif 'delete' in self.postvars.keys():
+            tmpl.delete(self.db_conn)
         self.redir_url = 'templates'
 
     # GET routes
@@ -565,6 +585,23 @@ class TestWithDB(TestCase):
         tmpl_2.save(self.db_conn)
         self.assertEqual({tmpl_1.id_, tmpl_2.id_}, set(t.id_ for t in TodoTemplate.all(self.db_conn)))
 
+    def test_TodoTemplate_delete(self):
+        tmpl = TodoTemplate(self.db_conn)
+        tmpl.title.set('foo')
+        tmpl.save(self.db_conn)
+        self.assertIsInstance(TodoTemplate.by_id(self.db_conn, tmpl.id_), TodoTemplate)
+        tmpl.delete(self.db_conn)
+        self.assertEqual(None, TodoTemplate.by_id(self.db_conn, tmpl.id_))
+        tmpl = TodoTemplate(self.db_conn)
+        tmpl.save(self.db_conn)
+        fork = tmpl.fork(self.db_conn)
+        with self.assertRaises(HandledException):
+            tmpl.delete(self.db_conn)
+        self.assertIsInstance(TodoTemplate.by_id(self.db_conn, tmpl.id_), TodoTemplate)
+        fork.delete(self.db_conn)
+        tmpl.delete(self.db_conn)
+        self.assertEqual(None, TodoTemplate.by_id(self.db_conn, tmpl.id_))
+
     def test_versioned_attributes(self):
         def test(name, default, values):
             def wait_till_next_timestamp(timestamp):
@@ -580,8 +617,8 @@ class TestWithDB(TestCase):
             self.assertEqual(tmpl_attr.newest, default)
             self.assertEqual({}, tmpl_attr.history)
             # check we get new value when set 
-            timestamp_1 = Day.todays_date(with_time=True)
             tmpl_attr.set(values[0])
+            timestamp_1 = Day.todays_date(with_time=True)
             self.assertEqual(tmpl_attr.newest, values[0])
             # check history remains unchanged if setting same value as .newest 
             timestamp_2 = wait_till_next_timestamp(timestamp_1)
@@ -652,3 +689,22 @@ class TestSansDB(TestCase):
         day3 = Day('2024-01-03')
         days = [day3, day1, day2]
         self.assertEqual(sorted(days), [day1, day2, day3])
+
+    def test_ParamsParser(self):
+        from urllib.parse import urlencode
+        q = urlencode({'is_foo': 'foo', 'is_whitespace': '', 'is_None': None})
+        c = {'is_bar': 'bar'}
+        p = ParamsParser(q, c)
+        # check basic retrieval and default behavior
+        self.assertEqual(p.get('is_foo', None), 'foo')
+        self.assertEqual(p.get('missing', None), None)
+        self.assertEqual(p.get('missing', 'default'), 'default')
+        # check handling of empty and None values
+        self.assertEqual(p.get('missing', ''), '')
+        self.assertEqual(p.get('is_whitespace', None), '')
+        self.assertEqual(p.get('is_None', None), 'None')  # TODO: unwanted behavior, or urlencode fault?
+        # check retrieval and setting of cookied values
+        self.assertEqual(p.get_cookied('missing', None), None)
+        self.assertEqual(c['missing'], None)
+        self.assertEqual(p.get_cookied('missing', 'default'), None)
+        self.assertEqual(p.get_cookied('is_bar', None), 'bar')