team

app.py

@@ -54,6 +54,42 @@ def lgs():
 
     return render_template('lgs.html', lgs = lgs, lg=lg, teams = teams)
 
+
+@app.route('/team2/<team_id>', methods=["GET", "POST"])
+def team2(team_id):
+    t = Team.from_id(team_id, mongo)
+    t.sfp = sfp
+
+    currentPlayers = t.get_players()
+
+    results = sfp.search_player("")
+
+    if 'searchBtn' in request.form:
+        name = request.form.get('name', '')
+        results = sfp.search_player(name)
+
+    if 'selectPlayers' in request.form:
+        selectPlayers = request.form.getlist('players')
+        selectPlayers = [int(x) for x in selectPlayers]
+
+        for i in selectPlayers:
+            p = sfp.get_player(i)
+            t.add_player(p)
+
+        t.save()
+
+        return redirect(url_for("team2", team_id=team_id))
+
+    if 'deletePlayer' in request.form:
+        deletePlayers = request.form.getlist('deletes')
+        for i in deletePlayers:
+            t.remove_player(int(i))
+
+        t.save()
+        return redirect(url_for("team2", team_id=team_id))
+
+    return render_template("team2.html", team = t, results = results, currentPlayers = currentPlayers)
+
 @app.route('/team/<team_id>', methods=["GET", "POST"])
 def team(team_id):
     team = mongo.db.teams.find_one({"_id": ObjectId(team_id)})

fpl_lib.py

@@ -1,16 +1,65 @@
 from fpl import FPL
-from fpl.utils import get_current_gameweek
+from fpl.utils import get_current_gameweek, position_converter
 import aiohttp
 import asyncio
 from bson.objectid import ObjectId
 
 class Team:
-    def __init__(self, name):
-        self.name = name
+    def __init__(self, title, mongo, sfp=None):
+        self.mongo = mongo
+
+        self.title = title
+        self.gks = []
+        self.gks_n = 2
+
+        self.defs = []
+        self.defs_n = 5
+
+        self.mids = []
+        self.mids_n = 5
+
+        self.fwds = []
+        self.fwds_n = 3
+
+
 
     def set_players(self, players):
         self.players = players
 
+
+    def get_players(self):
+        temp = []
+        if self.gks != None:
+            for o in self.gks:
+                temp.append(self.sfp.get_player(o))
+
+        if self.defs != None:
+            for o in self.defs:
+                temp.append(self.sfp.get_player(o))
+
+        if self.mids != None:
+            for o in self.mids:
+                temp.append(self.sfp.get_player(o))
+
+        if self.fwds != None:
+            for o in self.fwds:
+                temp.append(self.sfp.get_player(o))
+
+        return temp
+
+    def remove_player(self, pid):
+        p = self.sfp.get_player(pid)
+        ps = self.pos_str(p)
+
+        if ps == "Goalkeeper":
+            self.gks.remove(pid)
+        if ps == "Defender":
+            self.defs.remove(pid)
+        if ps == "Midfielder":
+            self.mids.remove(pid)
+        if ps == "Forward":
+            self.fwds.remove(pid)
+
     def show_players_info(self):
         print(f"{self.name} players:")
         for p in self.players:
@@ -20,7 +69,43 @@ class Team:
         print(f"Total Score => {total}")
 
     def manage_squad(self):
-        pass
+        self.select_gk()
+        self.select_def()
+        self.select_mid()
+        self.select_fwd()
+
+    def pos_str(self, player):
+        return position_converter(player.element_type)
+
+    def add_player(self, player):
+        print(f"add player {player}")
+        self.select_gk(player)
+        self.select_def(player)
+        self.select_mid(player)
+        self.select_fwd(player)
+
+    def select_gk(self, player):
+        if len(self.gks) < self.gks_n:
+            print(f"xxx {len(self.gks)}")
+            if self.pos_str(player) == "Goalkeeper":
+                print("Goal Keep")
+                self.gks.append(player.id)
+                print(self.gks)
+
+    def select_def(self, player):
+        if len(self.defs) < self.defs_n:
+            if self.pos_str(player) == "Defender":
+                self.defs.append(player.id)
+
+    def select_mid(self, player):
+        if len(self.mids) < self.mids_n:
+            if self.pos_str(player) == "Midfielder":
+                self.mids.append(player.id)
+
+    def select_fwd(self, player):
+        if len(self.fwds) < self.fwds_n:
+            if self.pos_str(player) == "Forward":
+                self.fwds.append(player.id)
 
     def set_starting(self):
         pass
@@ -28,8 +113,42 @@ class Team:
     def transfer(self):
         pass
 
+    def save(self):
+        return  self.mongo.db.teams.update_one({"_id": self._id}, {'$set':{
+            'title': self.title, 'gks': self.gks, 'defs': self.defs, 'mids': self.mids,
+            'fwds': self.fwds
+        }})
+
+    def create(self):
+        return  self.mongo.db.teams.insert_one({
+            'name': self.name, 'gks': self.gks, 'defs': self.defs, 'mids': self.mids,
+            'fwds': self.fwds
+        })
+
+
+    @classmethod
+    def from_id(cls, oid, mongo=None):
+        doc = mongo.db.teams.find_one({"_id": ObjectId(oid)})
+        c = cls(None, mongo)
+        c.init_db(doc)
+        return c
+
+    def init_db(self,  col):
+        for f in ['title', 'gks', 'defs', 'mids', 'fwds', "_id"]:
+            print(col)
+            setattr(self, f, col.get(f, None))
+
+        if self.gks == None:
+            self.gks = []
+
+        if self.defs == None:
+            self.defs = []
 
+        if self.mids == None:
+            self.mids = []
 
+        if self.fwds == None:
+            self.fwds = []
 
 class Leauge:
     def __init__(self, title, mongo=None):
@@ -97,7 +216,8 @@ class SFL:
         return await self.fpl.get_player(pid)
 
     def search_player(self, name):
-        results = [{'info':  str(x), 'id': x.id } for x in self.players if name in x.web_name.lower()]
+        results = [{'info':  str(x), 'id': x.id } for x in self.players
+                   if (name in x.web_name.lower()) or ( name  in position_converter(x.element_type).lower())]
         return results
 
 

templates/team2.html

@@ -0,0 +1,30 @@
+{% extends "base.html" %}
+{% block sidemenu %}
+<form method="post">
+    Enter Player Name
+    <input type="text" name="name" >
+    <input type="submit" name="searchBtn" value="Search">
+</form>
+    <form  method="post">
+        
+    <ul>
+    {% for r in results %}
+    <li><input type=checkbox name="players" value="{{ r.id }}" />{{ r.info }}</li>
+    {% endfor %}
+    </ul>
+    <input type="submit" name="selectPlayers" value="Select Players">
+    </form>
+{% endblock %}
+{% block content %}
+{{ team }}
+<h1> Current Team: {{ team.title }}</h1>
+    <h2>Players</h2>
+    <form method="post">
+    <ol>
+    {% for p in  currentPlayers %}
+    <li>{{ p }}  GS={{ p.goals_scored }} <input type="checkbox" name="deletes" value="{{ p.id  }}">Delete</li>
+    {% endfor %}
+    </ul>
+    <input type="submit" name="deletePlayer"  value="Delete Players">
+    </form>
+{% endblock %}