Ei kuvausta

app.py 1.7KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. from flask import Flask, render_template, request
  2. from fpl_lib import SFL, Team
  3. from flask_pymongo import PyMongo
  4. from bson.objectid import ObjectId
  5. app = Flask(__name__)
  6. app.config["MONGO_URI"] = "mongodb://localhost:27017/fpl_db"
  7. mongo = PyMongo(app)
  8. sfp = SFL("ABC")
  9. sfp.setup()
  10. @app.route('/', methods=["GET", "POST"])
  11. def index():
  12. players = sfp.players[:400]
  13. print("=== players ===")
  14. print(players)
  15. selectTeam = request.args.get("t", "")
  16. if request.method == "POST":
  17. teamCreate = request.form.get('teamCreate', None)
  18. if 'selectPlayerBtn' in request.form:
  19. print(f"selectPlayerBtn == {selectTeam}")
  20. selectPlayers = request.form.getlist('players')
  21. selectPlayers = [int(x) for x in selectPlayers]
  22. mongo.db.teams.update_one({"_id": ObjectId(selectTeam)}, {"$set": {"players": selectPlayers}})
  23. print("select players")
  24. print(selectPlayers)
  25. if teamCreate:
  26. print("Team Create")
  27. teamName = request.form.get('teamName', None)
  28. mongo.db.teams.insert_one({'title': teamName})
  29. teams = mongo.db.teams.find()
  30. currentTeam = None
  31. if selectTeam != "":
  32. currentTeam = mongo.db.teams.find_one({"_id": ObjectId(selectTeam)})
  33. currentPlayers = []
  34. for p in currentTeam['players']:
  35. try:
  36. currentPlayers.append(sfp.get_player(p))
  37. except:
  38. currentPlayers.append(None)
  39. #currentPlayers = sfp.get_players(currentTeam['players'])
  40. return render_template('index.html', players = players, teams=teams, selectTeam = selectTeam, currentTeam = currentTeam, currentPlayers = currentPlayers)