from flask import Flask, render_template, request
from fpl_lib import SFL, Team
from flask_pymongo import PyMongo
from bson.objectid import ObjectId


app = Flask(__name__)
app.config["MONGO_URI"] = "mongodb://localhost:27017/fpl_db"
mongo = PyMongo(app)

sfp = SFL("ABC")
sfp.setup()

@app.route('/', methods=["GET", "POST"])
def index():
    players = sfp.players[:400]
    print("=== players ===")
    print(players)
    selectTeam = request.args.get("t", "")

    if request.method == "POST":

        teamCreate = request.form.get('teamCreate', None)
        if 'selectPlayerBtn' in request.form:
            print(f"selectPlayerBtn == {selectTeam}")
            selectPlayers = request.form.getlist('players')
            selectPlayers = [int(x) for x in selectPlayers]
            mongo.db.teams.update_one({"_id": ObjectId(selectTeam)}, {"$set": {"players": selectPlayers}})

        print("select players")
        print(selectPlayers)

        if teamCreate:
            print("Team Create")
            teamName = request.form.get('teamName', None)
            mongo.db.teams.insert_one({'title': teamName})

    teams = mongo.db.teams.find()
    currentTeam = None

    if selectTeam != "":
        currentTeam = mongo.db.teams.find_one({"_id": ObjectId(selectTeam)})
        currentPlayers = []

        for p in currentTeam['players']:
            try:
                currentPlayers.append(sfp.get_player(p))
            except:
                currentPlayers.append(None)
       #currentPlayers = sfp.get_players(currentTeam['players'])

    return render_template('index.html', players = players, teams=teams, selectTeam =  selectTeam, currentTeam = currentTeam, currentPlayers = currentPlayers)