| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253 |
- 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)
|