From bc78e17de6be8619ef41d862f11c04dfa3451536 Mon Sep 17 00:00:00 2001
From: Alexander Hosking <alexander@ahosking.com>
Date: Wed, 27 Apr 2022 08:41:18 -0400
Subject: [PATCH] Add Delete Function

---
 period/cycles.py                   | 15 ++++++++++++---
 period/templates/cycle/index.html  |  8 +++++++-
 period/templates/cycle/update.html |  5 +++++
 3 files changed, 24 insertions(+), 4 deletions(-)

diff --git a/period/cycles.py b/period/cycles.py
index c6cc0f1..be14b24 100644
--- a/period/cycles.py
+++ b/period/cycles.py
@@ -54,8 +54,9 @@ def create():
                 )
             else:
                 db.execute(
-                    'INSERT INTO cycle (start_time, cycle_length) VALUES (?, ?)',
-                    (start_time_stamp, cycle_length)
+                    'INSERT INTO cycle (start_time)'
+                    ' VALUES (?)',
+                    (start_time.to_datetime_string(),)
                 )
             db.commit()
             return redirect(url_for('index'))
@@ -118,4 +119,12 @@ def update(id):
                 )
             db.commit()
             return redirect(url_for('index'))
-    return render_template('cycle/update.html', cycle=cycle)
\ No newline at end of file
+    return render_template('cycle/update.html', cycle=cycle)
+
+@bp.route('/<int:id>/delete', methods=('POST',))
+def delete(id):
+    get_cycle(id)
+    db = get_db()
+    db.execute('DELETE FROM cycle WHERE id = ?', (id,))
+    db.commit()
+    return redirect(url_for('index'))
\ No newline at end of file
diff --git a/period/templates/cycle/index.html b/period/templates/cycle/index.html
index e78633a..9602d32 100644
--- a/period/templates/cycle/index.html
+++ b/period/templates/cycle/index.html
@@ -25,7 +25,13 @@
               <b>Ended: </b>{{ cycle['end_time'] }}</div>
             </div>
           </header>
-          <p class="body">Cycle Length: <b>{{ cycle['cycle_length'] }} days</b>.</p>
+          <p class="body">
+            {% if cycle['cycle_length'] %}
+            Cycle Length: <b>{{ cycle['cycle_length'] }} days</b>.
+            {% else %}
+
+            {% endif %}
+          </p>
           <a class="action" href="{{ url_for('cycles.update', id=cycle['id'])}}">Edit</a> - {{ cycle['id'] }}
         </article>
         {% if not loop.last %}
diff --git a/period/templates/cycle/update.html b/period/templates/cycle/update.html
index a3f8234..6eb0b24 100644
--- a/period/templates/cycle/update.html
+++ b/period/templates/cycle/update.html
@@ -12,4 +12,9 @@
     <input name="end_time" id="end_time" value="{{ request.form['end_time'] or cycle['end_time'][:10] if cycle['end_time'] is not none }}" type="date">
     <input type="submit" value="Save">
   </form>
+
+  
+  <form action="{{ url_for('cycles.delete', id=cycle['id']) }}" method="post">
+    <input class="danger" type="submit" value="Delete" onclick="return confirm('Are you sure you want to delete this cycle?');">
+  </form>
 {% endblock %}
\ No newline at end of file