Merge pull request #52 from vicenteneto/dev

Adicionando suporte para a pontuaçao dobrada do capitao

Author: vicenteneto

CHANGES

@@ -3,10 +3,18 @@ Changelog
 
 Aqui você pode ver a lista completa de modificações entre cada versão.
 
+Versão 1.4.15
+-------------
+
+Lançada em 15 de abril de 2018
+
+- Adicionando suporte para pontuaçao do capitao
+
+
 Versão 1.4.14
 -------------
 
-Lançada em 1 de abril de 2018
+Lançada em 14 de abril de 2018
 
 - Atualizando projeto exemplo para suportar Python3
 - Corrigindo erro ao visualizar Mercado

cartolafc/api.py

@@ -143,7 +143,7 @@ def time_logado(self):
         url = '{api_url}/auth/time'.format(api_url=self._api_url)
         data = self._request(url)
         clubes = {clube['id']: Clube.from_dict(clube) for clube in data['clubes'].values()}
-        return Time.from_dict(data, clubes=clubes)
+        return Time.from_dict(data, clubes=clubes, capitao=data['capitao_id'])
 
     def clubes(self):
         url = '{api_url}/clubes'.format(api_url=self._api_url)
@@ -248,7 +248,7 @@ def time(self, id=None, nome=None, slug=None, as_json=False):
             return data
 
         clubes = {clube['id']: Clube.from_dict(clube) for clube in data['clubes'].values()}
-        return Time.from_dict(data, clubes=clubes)
+        return Time.from_dict(data, clubes=clubes, capitao=data['capitao_id'])
 
     def time_parcial(self, id=None, nome=None, slug=None, parciais=None):
         if self.mercado().status.id == MERCADO_FECHADO:
@@ -260,6 +260,10 @@ def time_parcial(self, id=None, nome=None, slug=None, parciais=None):
                 tem_parcial = atleta.id in parciais
                 atleta.pontos = parciais[atleta.id].pontos if tem_parcial else 0
                 atleta.scout = parciais[atleta.id].scout if tem_parcial else {}
+
+                if atleta.is_capitao:
+                    atleta.pontos *= 2
+
                 time.pontos += atleta.pontos
 
             return time

cartolafc/models.py

@@ -43,22 +43,23 @@ def __repr__(self):
 class Atleta(BaseModel):
     """ Representa um atleta (jogador ou técnico), e possui informações como o apelido, clube e pontuação obtida """
 
-    def __init__(self, atleta_id, apelido, pontos, scout, posicao_id, clube, status_id=None):
+    def __init__(self, atleta_id, apelido, pontos, scout, posicao_id, clube, status_id=None, is_capitao=None):
         self.id = atleta_id
         self.apelido = apelido
         self.pontos = pontos
         self.scout = scout
         self.posicao = _posicoes[posicao_id]
         self.clube = clube
         self.status = _atleta_status[status_id] if status_id else None
+        self.is_capitao = is_capitao
 
     @classmethod
-    def from_dict(cls, data, clubes, atleta_id=None):
+    def from_dict(cls, data, clubes, atleta_id=None, is_capitao=None):
         atleta_id = atleta_id if atleta_id else data['atleta_id']
         pontos = data['pontos_num'] if 'pontos_num' in data else data['pontuacao']
         clube = clubes[data['clube_id']]
         return cls(atleta_id, data['apelido'], pontos, data['scout'], data['posicao_id'], clube,
-                   data.get('status_id', None))
+                   data.get('status_id', None), is_capitao)
 
 
 class Clube(BaseModel):
@@ -183,9 +184,9 @@ def __init__(self, patrimonio, valor_time, ultima_pontuacao, atletas, info):
         self.info = info
 
     @classmethod
-    def from_dict(cls, data, clubes):
+    def from_dict(cls, data, clubes, capitao):
         data['atletas'].sort(key=lambda a: a['posicao_id'])
-        atletas = [Atleta.from_dict(atleta, clubes) for atleta in data['atletas']]
+        atletas = [Atleta.from_dict(atleta, clubes, is_capitao=atleta['atleta_id'] == capitao) for atleta in data['atletas']]
         info = TimeInfo.from_dict(data['time'])
         return cls(data['patrimonio'], data['valor_time'], data['pontos'], atletas, info)
 

examples/requirements.txt

@@ -1 +1 @@
-Python-CartolaFC==1.4.14
+Python-CartolaFC==1.4.15

setup.py

@@ -14,7 +14,7 @@
 """
 from setuptools import setup
 
-version = '1.4.14'
+version = '1.4.15'
 packages = ['cartolafc']
 install_requires = ['requests']
 python_cartolafc_pkg_data = []

testdata/time.json

@@ -564,6 +564,7 @@
     "temporada_inicial": 2013,
     "simplificado": false
   },
+  "capitao_id": 73800,
   "patrimonio": 0,
   "esquema_id": 3,
   "pontos": 70.02978515625,

testdata/time_logado.json

@@ -621,6 +621,7 @@
       "status": "HABILITADO"
     }
   ],
+  "capitao_id": 38140,
   "total_ligas": 3,
   "total_ligas_matamata": 1
 }