test impl

app.py

@@ -0,0 +1,99 @@
+from flask import Flask, render_template, request, jsonify
+from tinydb import TinyDB, Query
+from datetime import datetime, timedelta
+
+app = Flask(__name__)
+db = TinyDB('habits.json')
+Habit = Query()
+
+# Helper function to get the current date string
+def get_current_date():
+    return datetime.now().strftime('%Y-%m-%d')
+
+# Helper function to get a list of dates for the last 30 days
+def get_last_30_days():
+    dates = []
+    for i in range(30):
+        date = datetime.now() - timedelta(days=i)
+        dates.append(date.strftime('%Y-%m-%d'))
+    return dates[::-1] # Reverse to have oldest first
+
+@app.route('/')
+def index():
+    return render_template('index.html')
+
+@app.route('/habits', methods=['GET'])
+def get_habits():
+    habits_data = db.all()
+    # Add completion status for the last 30 days
+    last_30_days = get_last_30_days()
+    for habit in habits_data:
+        habit['completion_history'] = {}
+        for date_str in last_30_days:
+            habit['completion_history'][date_str] = date_str in habit.get('completed_dates', [])
+    return jsonify(habits_data)
+
+@app.route('/habits', methods=['POST'])
+def add_habit():
+    data = request.json
+    name = data.get('name')
+    if not name:
+        return jsonify({'error': 'Habit name is required'}), 400
+    
+    habit_id = db.insert({'name': name, 'completed_dates': []})
+    return jsonify({'id': habit_id, 'name': name, 'completed_dates': []}), 201
+
+@app.route('/habits/<int:habit_id>/complete', methods=['POST'])
+def complete_habit(habit_id):
+    date_to_complete = request.json.get('date', get_current_date())
+    
+    habit = db.get(doc_id=habit_id)
+    if not habit:
+        return jsonify({'error': 'Habit not found'}), 404
+    
+    if date_to_complete not in habit['completed_dates']:
+        habit['completed_dates'].append(date_to_complete)
+        db.update({'completed_dates': habit['completed_dates']}, doc_ids=[habit_id])
+    
+    return jsonify({'message': f'Habit {habit_id} marked as completed for {date_to_complete}', 'completed_dates': habit['completed_dates']})
+
+@app.route('/habits/<int:habit_id>/uncomplete', methods=['POST'])
+def uncomplete_habit(habit_id):
+    date_to_uncomplete = request.json.get('date', get_current_date())
+
+    habit = db.get(doc_id=habit_id)
+    if not habit:
+        return jsonify({'error': 'Habit not found'}), 404
+    
+    if date_to_uncomplete in habit['completed_dates']:
+        habit['completed_dates'].remove(date_to_uncomplete)
+        db.update({'completed_dates': habit['completed_dates']}, doc_ids=[habit_id])
+    
+    return jsonify({'message': f'Habit {habit_id} marked as uncompleted for {date_to_uncomplete}', 'completed_dates': habit['completed_dates']})
+
+@app.route('/habits/<int:habit_id>', methods=['PUT'])
+def update_habit(habit_id):
+    data = request.json
+    name = data.get('name')
+    
+    if not name:
+        return jsonify({'error': 'Habit name is required'}), 400
+    
+    habit = db.get(doc_id=habit_id)
+    if not habit:
+        return jsonify({'error': 'Habit not found'}), 404
+    
+    db.update({'name': name}, doc_ids=[habit_id])
+    return jsonify({'message': f'Habit {habit_id} updated', 'name': name})
+
+@app.route('/habits/<int:habit_id>', methods=['DELETE'])
+def delete_habit(habit_id):
+    habit = db.get(doc_id=habit_id)
+    if not habit:
+        return jsonify({'error': 'Habit not found'}), 404
+    
+    db.remove(doc_ids=[habit_id])
+    return jsonify({'message': f'Habit {habit_id} deleted'})
+
+if __name__ == '__main__':
+    app.run(debug=True)