365 lines
No EOL
13 KiB
Python
365 lines
No EOL
13 KiB
Python
from flask import Blueprint, render_template, redirect, url_for, request, flash, jsonify
|
|
from flask_login import login_required, current_user
|
|
from datetime import datetime, date
|
|
from . import db
|
|
from .models import init_db_models
|
|
|
|
# Import model classes (stored on init_db_models function)
|
|
DailyLog = init_db_models.DailyLog
|
|
DailyLogItem = init_db_models.DailyLogItem
|
|
UserSettings = init_db_models.UserSettings
|
|
|
|
bp = Blueprint('main', __name__)
|
|
|
|
|
|
# ─── Step 2: Daily Log (main page) ───────────────────────────────────
|
|
@bp.route('/')
|
|
@login_required
|
|
def index():
|
|
"""Daily Log screen — main page (Step 2)."""
|
|
today = date.today()
|
|
log = DailyLog.query.filter_by(date=today, user_id=current_user.id).first()
|
|
|
|
if not log:
|
|
log = DailyLog(date=today, user_id=current_user.id)
|
|
db.session.add(log)
|
|
db.session.commit()
|
|
|
|
# Get settings
|
|
settings = UserSettings.query.filter_by(user_id=current_user.id).first()
|
|
if not settings:
|
|
settings = UserSettings(user_id=current_user.id)
|
|
db.session.add(settings)
|
|
db.session.commit()
|
|
|
|
return render_template('daily_log.html', log=log, settings=settings)
|
|
|
|
|
|
# ─── Step 2: All Logs ────────────────────────────────────────────────
|
|
@bp.route('/logs')
|
|
@login_required
|
|
def log_list():
|
|
"""All Logs screen — list of dates (Step 2)."""
|
|
logs = DailyLog.query.filter_by(user_id=current_user.id)\
|
|
.order_by(DailyLog.date.desc()).all()
|
|
return render_template('log_list.html', logs=logs)
|
|
|
|
|
|
# ─── Step 2: Daily Log Detail (by date) ──────────────────────────────
|
|
@bp.route('/logs/<int:log_id>')
|
|
@login_required
|
|
def log_detail(log_id):
|
|
"""Open the Daily Log for a specific date."""
|
|
log = DailyLog.query.get_or_404(log_id)
|
|
if log.user_id != current_user.id:
|
|
flash('Access denied.')
|
|
return redirect(url_for('main.log_list'))
|
|
return render_template('daily_log.html', log=log, settings=None)
|
|
|
|
|
|
# ─── Step 4: Create / Edit Log ───────────────────────────────────────
|
|
@bp.route('/logs/new', methods=['GET', 'POST'])
|
|
@login_required
|
|
def log_new():
|
|
"""Create a new Daily Log entry."""
|
|
if request.method == 'POST':
|
|
log_date = request.form.get('date')
|
|
try:
|
|
log_date = datetime.strptime(log_date, '%Y-%m-%d').date()
|
|
except ValueError:
|
|
flash('Invalid date format.')
|
|
return render_template('daily_log_form.html', action='New')
|
|
|
|
existing = DailyLog.query.filter_by(date=log_date, user_id=current_user.id).first()
|
|
if existing:
|
|
flash('Log for this date already exists.')
|
|
return redirect(url_for('main.log_detail', log_id=existing.id))
|
|
|
|
log = DailyLog(
|
|
date=log_date,
|
|
prompt=request.form.get('prompt', ''),
|
|
notes=request.form.get('notes', ''),
|
|
bedtime=request.form.get('bedtime', ''),
|
|
wake_time=request.form.get('wake_time', ''),
|
|
sleep_quality=request.form.get('sleep_quality'),
|
|
mood=request.form.get('mood', ''),
|
|
user_id=current_user.id,
|
|
)
|
|
db.session.add(log)
|
|
db.session.commit()
|
|
|
|
# Handle gratitude (3 fields)
|
|
gratitude = [request.form.get(f'gratitude_{i}', '') for i in range(3)]
|
|
log.gratitude = gratitude
|
|
|
|
# Handle bullets / items
|
|
item_types = request.form.getlist('item_type[]')
|
|
item_symbols = request.form.getlist('item_symbol[]')
|
|
item_texts = request.form.getlist('item_text[]')
|
|
item_statuses = request.form.getlist('item_status[]')
|
|
|
|
for i in range(len(item_types)):
|
|
if item_types[i] and item_texts[i].strip():
|
|
item = DailyLogItem(
|
|
log_id=log.id,
|
|
type=item_types[i],
|
|
symbol=item_symbols[i] if item_symbols[i] else '•',
|
|
text=item_texts[i].strip(),
|
|
status=item_statuses[i] if item_statuses[i] in ['todo', 'doing', 'done', 'cancelled'] else 'todo',
|
|
)
|
|
db.session.add(item)
|
|
|
|
db.session.commit()
|
|
flash('Log saved!')
|
|
return redirect(url_for('main.log_detail', log_id=log.id))
|
|
|
|
return render_template('daily_log_form.html', action='New')
|
|
|
|
|
|
@bp.route('/logs/<int:log_id>/edit', methods=['GET', 'POST'])
|
|
@login_required
|
|
def log_edit(log_id):
|
|
"""Edit an existing Daily Log entry."""
|
|
log = DailyLog.query.get_or_404(log_id)
|
|
if log.user_id != current_user.id:
|
|
flash('Access denied.')
|
|
return redirect(url_for('main.log_list'))
|
|
|
|
if request.method == 'POST':
|
|
log.prompt = request.form.get('prompt', '')
|
|
log.notes = request.form.get('notes', '')
|
|
log.bedtime = request.form.get('bedtime', '')
|
|
log.wake_time = request.form.get('wake_time', '')
|
|
log.sleep_quality = request.form.get('sleep_quality')
|
|
log.mood = request.form.get('mood', '')
|
|
|
|
# Handle gratitude
|
|
gratitude = [request.form.get(f'gratitude_{i}', '') for i in range(3)]
|
|
log.gratitude = gratitude
|
|
|
|
# Handle existing items and new items
|
|
item_ids = request.form.getlist('item_id[]')
|
|
item_types = request.form.getlist('item_type[]')
|
|
item_symbols = request.form.getlist('item_symbol[]')
|
|
item_texts = request.form.getlist('item_text[]')
|
|
item_statuses = request.form.getlist('item_status[]')
|
|
|
|
# Remove items not in item_ids
|
|
existing_ids = [int(i) for i in item_ids if i.isdigit()]
|
|
for item in log.items:
|
|
if item.id not in existing_ids:
|
|
db.session.delete(item)
|
|
|
|
for i in range(len(item_types)):
|
|
if item_ids[i] and item_ids[i].isdigit():
|
|
item = DailyLogItem.query.get(int(item_ids[i]))
|
|
if item:
|
|
item.type = item_types[i]
|
|
item.symbol = item_symbols[i] if item_symbols[i] else '•'
|
|
item.text = item_texts[i].strip()
|
|
item.status = item_statuses[i] if item_statuses[i] in ['todo', 'doing', 'done', 'cancelled'] else 'todo'
|
|
elif item_types[i] and item_texts[i].strip():
|
|
item = DailyLogItem(
|
|
log_id=log.id,
|
|
type=item_types[i],
|
|
symbol=item_symbols[i] if item_symbols[i] else '•',
|
|
text=item_texts[i].strip(),
|
|
status=item_statuses[i] if item_statuses[i] in ['todo', 'doing', 'done', 'cancelled'] else 'todo',
|
|
)
|
|
db.session.add(item)
|
|
|
|
db.session.commit()
|
|
flash('Log updated!')
|
|
return redirect(url_for('main.log_detail', log_id=log.id))
|
|
|
|
return render_template('daily_log_form.html', log=log, action='Edit')
|
|
|
|
|
|
# ─── Step 4: Add / Update Bullet (AJAX) ──────────────────────────────
|
|
@bp.route('/logs/<int:log_id>/items', methods=['POST'])
|
|
@login_required
|
|
def add_log_item(log_id):
|
|
"""Add a bullet/item to a log (AJAX endpoint)."""
|
|
log = DailyLog.query.get_or_404(log_id)
|
|
if log.user_id != current_user.id:
|
|
return jsonify({'success': False, 'error': 'Access denied.'}), 403
|
|
|
|
data = request.get_json()
|
|
item_type = data.get('type', 'bullet')
|
|
symbol = data.get('symbol', '•')
|
|
text = data.get('text', '').strip()
|
|
status = data.get('status', 'todo')
|
|
|
|
if not text:
|
|
return jsonify({'success': False, 'error': 'Text is required.'}), 400
|
|
|
|
item = DailyLogItem(
|
|
log_id=log.id,
|
|
type=item_type,
|
|
symbol=symbol,
|
|
text=text,
|
|
status=status,
|
|
)
|
|
db.session.add(item)
|
|
db.session.commit()
|
|
|
|
return jsonify({
|
|
'success': True,
|
|
'id': item.id,
|
|
'type': item.type,
|
|
'symbol': item.symbol,
|
|
'text': item.text,
|
|
'status': item.status,
|
|
})
|
|
|
|
|
|
@bp.route('/logs/items/<int:item_id>', methods=['PUT'])
|
|
@login_required
|
|
def update_log_item(item_id):
|
|
"""Update a bullet/item status or text (AJAX endpoint)."""
|
|
item = DailyLogItem.query.get_or_404(item_id)
|
|
if item.log.user_id != current_user.id:
|
|
return jsonify({'success': False, 'error': 'Access denied.'}), 403
|
|
|
|
data = request.get_json()
|
|
if 'status' in data:
|
|
item.status = data['status']
|
|
if 'text' in data:
|
|
item.text = data['text'].strip()
|
|
|
|
db.session.commit()
|
|
return jsonify({'success': True, 'status': item.status, 'text': item.text})
|
|
|
|
|
|
@bp.route('/logs/items/<int:item_id>', methods=['DELETE'])
|
|
@login_required
|
|
def delete_log_item(item_id):
|
|
"""Delete a bullet/item (AJAX endpoint)."""
|
|
item = DailyLogItem.query.get_or_404(item_id)
|
|
if item.log.user_id != current_user.id:
|
|
return jsonify({'success': False, 'error': 'Access denied.'}), 403
|
|
|
|
db.session.delete(item)
|
|
db.session.commit()
|
|
return jsonify({'success': True})
|
|
|
|
|
|
# ─── Step 4: Export / Import ─────────────────────────────────────────
|
|
@bp.route('/export')
|
|
@login_required
|
|
def export_logs():
|
|
"""Export all logs as JSON (Step 4)."""
|
|
logs = DailyLog.query.filter_by(user_id=current_user.id).all()
|
|
data = []
|
|
for log in logs:
|
|
log_data = {
|
|
'date': log.date.isoformat(),
|
|
'prompt': log.prompt,
|
|
'notes': log.notes,
|
|
'gratitude': log.gratitude or [],
|
|
'bedtime': log.bedtime,
|
|
'wake_time': log.wake_time,
|
|
'sleep_quality': log.sleep_quality,
|
|
'mood': log.mood,
|
|
'items': [
|
|
{
|
|
'type': item.type.value,
|
|
'symbol': item.symbol,
|
|
'text': item.text,
|
|
'status': item.status.value,
|
|
}
|
|
for item in log.items
|
|
],
|
|
}
|
|
data.append(log_data)
|
|
|
|
return jsonify(data), 200, {'Content-Type': 'application/json', 'Content-Disposition': 'attachment; filename="logs.json"'}
|
|
|
|
|
|
@bp.route('/import', methods=['POST'])
|
|
@login_required
|
|
def import_logs():
|
|
"""Import logs from JSON file (Step 4)."""
|
|
if 'file' not in request.files:
|
|
flash('No file uploaded.')
|
|
return redirect(url_for('main.settings'))
|
|
|
|
file = request.files['file']
|
|
if not file or not file.filename.endswith('.json'):
|
|
flash('Invalid file. Please upload a JSON file.')
|
|
return redirect(url_for('main.settings'))
|
|
|
|
import json as json_module
|
|
try:
|
|
data = file.read().decode('utf-8')
|
|
logs_data = json_module.loads(data)
|
|
|
|
imported = 0
|
|
for log_data in logs_data:
|
|
log_date = datetime.strptime(log_data['date'], '%Y-%m-%d').date()
|
|
existing = DailyLog.query.filter_by(date=log_date, user_id=current_user.id).first()
|
|
if existing:
|
|
continue # Skip existing
|
|
|
|
log = DailyLog(
|
|
date=log_date,
|
|
prompt=log_data.get('prompt', ''),
|
|
notes=log_data.get('notes', ''),
|
|
gratitude=log_data.get('gratitude', []),
|
|
bedtime=log_data.get('bedtime', ''),
|
|
wake_time=log_data.get('wake_time', ''),
|
|
sleep_quality=log_data.get('sleep_quality'),
|
|
mood=log_data.get('mood', ''),
|
|
user_id=current_user.id,
|
|
)
|
|
db.session.add(log)
|
|
db.session.flush()
|
|
|
|
for item_data in log_data.get('items', []):
|
|
item = DailyLogItem(
|
|
log_id=log.id,
|
|
type=item_data.get('type', 'bullet'),
|
|
symbol=item_data.get('symbol', '•'),
|
|
text=item_data.get('text', ''),
|
|
status=item_data.get('status', 'todo'),
|
|
)
|
|
db.session.add(item)
|
|
|
|
imported += 1
|
|
|
|
db.session.commit()
|
|
flash(f'{imported} logs imported!')
|
|
except Exception as e:
|
|
db.session.rollback()
|
|
flash(f'Import failed: {str(e)}')
|
|
|
|
return redirect(url_for('main.settings'))
|
|
|
|
|
|
# ─── Step 4: Settings ────────────────────────────────────────────────
|
|
@bp.route('/settings')
|
|
@login_required
|
|
def settings():
|
|
"""Settings screen (Step 2)."""
|
|
s = UserSettings.query.filter_by(user_id=current_user.id).first()
|
|
if not s:
|
|
s = UserSettings(user_id=current_user.id)
|
|
db.session.add(s)
|
|
db.session.commit()
|
|
return render_template('settings.html', settings=s)
|
|
|
|
|
|
@bp.route('/settings/update', methods=['POST'])
|
|
@login_required
|
|
def update_settings():
|
|
"""Update user settings."""
|
|
s = UserSettings.query.filter_by(user_id=current_user.id).first()
|
|
if not s:
|
|
s = UserSettings(user_id=current_user.id)
|
|
db.session.add(s)
|
|
|
|
s.theme = request.form.get('theme', 'light')
|
|
s.font_size = int(request.form.get('font_size', 18))
|
|
db.session.commit()
|
|
flash('Settings saved!')
|
|
return redirect(url_for('main.settings')) |