app.py

import os
from datetime import datetime
from flask import Flask, render_template, redirect, url_for, flash, request
from .forms import TeamRegistrationForm, MatchResultsForm, TeamSearchForm, EditResultsForm
from .models import update_registration, update_results, clear_stored_data, calculate_rankings_grouped, search, no_data_stored


app = Flask(__name__)
# csrf validation is included in FlaskForm
app.config['SECRET_KEY'] = os.urandom(24)

# Store team data locally
DATA_FILE = './teams_data.txt'
RESULT_FILE = './results_data.txt'
LOG_FILE = './log.txt'

def store_team_data(data):
    with open(DATA_FILE, 'a') as file:
        file.write(data + '\n')

def store_result_data(data):
    with open(RESULT_FILE, 'a') as file:
        file.write(data + '\n')

@app.route('/view_teams')
def view_teams():
    if os.path.exists(DATA_FILE):
        with open(DATA_FILE, 'r') as file:
            teams = file.readlines()
    else:
        teams = []
    return render_template('view_teams.html', teams=teams)

@app.route('/view_results')
def view_results():
    if os.path.exists(RESULT_FILE):
        with open(RESULT_FILE, 'r') as file:
            results = file.readlines()
    else:
        results = []

    return render_template('view_results.html', results=results)

@app.route('/register_teams', methods=['GET', 'POST'])
def register_teams():
    form = TeamRegistrationForm()
    if form.validate_on_submit():
        log_action("register teams")
        teams_data = form.teams_data.data
        teams_list = teams_data.split('\n')
        for team_info in teams_list:
            try:
                team_name, reg_date, group_number = team_info.split()
                datetime.strptime(reg_date, "%d/%m")
                update_registration(team_name, reg_date, group_number)
                store_team_data(f"{team_name} {reg_date} {group_number}")
            except ValueError:
                flash('Invalid input format or invalid date. Make sure each line follows the format: <Team Name> <DD/MM> <Group Number>', 'error')
                return redirect(url_for('register_teams'))

        return redirect(url_for('register_teams'))

    return render_template('register_teams.html', form=form)

@app.route('/register_results', methods=['GET', 'POST'])
def register_results():
    form = MatchResultsForm()
    if form.validate_on_submit():
        log_action("register results")
        results_data = form.teams_data.data
        results_list = results_data.split('\n')
        for result_info in results_list:
            try:
                team_a_name, team_b_name, team_a_goals, team_b_goals = result_info.split()
                update_results(team_a_name, team_b_name, team_a_goals, team_b_goals)
                store_result_data(f"{team_a_name} {team_b_name} {team_a_goals} {team_b_goals}")
            except ValueError:
                flash('Invalid input format. Make sure each line follows the format: <Team A name> <Team B name> <Team A goals scored> <Team B goals scored>', 'error')
                return redirect(url_for('register_results'))

        return redirect(url_for('register_results'))

    return render_template('register_results.html', form=form)

@app.route('/edit_results', methods=['GET', 'POST'])
def edit_results():
    form = EditResultsForm()
    if request.method == 'GET':
        if os.path.exists('results_data.txt'):
            with open('results_data.txt', 'r') as file:
                results_data = file.read()
        else:
            results_data = ''
        form.results_data.data = results_data

    if form.validate_on_submit():
        log_action("edit results")
        new_results_data = form.results_data.data
        clear_file(RESULT_FILE)
        clear_stored_data()
        reload_team_data()
        # write then read new results
        results_list = new_results_data.split('\n')
        for result_info in results_list:
            try:
                team_a_name, team_b_name, team_a_goals, team_b_goals = result_info.split()
                update_results(team_a_name, team_b_name, team_a_goals, team_b_goals)
                store_result_data(f"{team_a_name} {team_b_name} {team_a_goals} {team_b_goals}")
            except ValueError:
                flash('Invalid input format. Make sure each line follows the format: <Team A name> <Team B name> <Team A goals scored> <Team B goals scored>', 'error')

        flash('Results updated successfully', 'success')
        return redirect(url_for('edit_results'))

    return render_template('edit_results.html', form=form)

# clear ALL data
@app.route('/clear_data')
def clear_data():
    log_action("clear data")
    clear_file(DATA_FILE)
    clear_file(RESULT_FILE)
    clear_stored_data()
    return redirect(url_for('index'))

@app.route('/search_team', methods=['GET', 'POST'])
def search_team():
    form = TeamSearchForm()
    team_data = None
    if form.validate_on_submit():
        team_name = form.team_name.data
        team_data = search(team_name)
        if not team_data:
            flash(f"No results found for team: {team_name}", 'error')

    return render_template('search_team.html', form=form, team_data=team_data)

def log_action(action):
    with open(LOG_FILE, 'a') as file:
        timestamp = datetime.now().strftime('%d/%m %H:%M:%S')
        file.write(f"{timestamp} - {action}\n")

def clear_file(filename):
    if os.path.exists(filename):
        os.remove(filename)

@app.route('/clear_log')
def clear_log():
    clear_file(LOG_FILE)
    return redirect(url_for('index'))

@app.route('/view_log')
def view_log():
    if os.path.exists(LOG_FILE):
        with open(LOG_FILE, 'r') as file:
            logs = file.readlines()
    else:
        logs = ["No logs available"]

    return render_template('view_log.html', logs=logs)

def reload_data():
    # used on start, and on edit
    log_action("reloaded data")
    reload_team_data()
    reload_result_data()

def reload_team_data():
    if os.path.exists(DATA_FILE):
        with open(DATA_FILE, 'r') as file:
            teams = file.readlines()
        for team_info in teams:
            team_name, reg_date, group_number = team_info.split()
            update_registration(team_name, reg_date, group_number)

def reload_result_data():
    if os.path.exists(RESULT_FILE):
        with open(RESULT_FILE, 'r') as file:
            results = file.readlines()
        for result_info in results:
            team_a_name, team_b_name, team_a_goals, team_b_goals = result_info.split()
            update_results(team_a_name, team_b_name, team_a_goals, team_b_goals)

@app.route('/')
def index():
    if no_data_stored():
        reload_data()
    ranked_groups = calculate_rankings_grouped()
    return render_template('index.html', ranked_groups=ranked_groups)

if __name__ == '__main__':
    app.run(debug=True)