99 lines
3.4 KiB
Python
99 lines
3.4 KiB
Python
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) |