Add team records view.

Author: SRJ9

driver27/models.py

@@ -215,6 +215,15 @@ def stats_rank(self, **filters):
         rank = sorted(rank, key=lambda x: x[0], reverse=True)
         return rank
 
+    def team_stats_rank(self, **filters):
+        teams = self.teams.all()
+        rank = []
+        for team in teams:
+            team_season = TeamSeason.objects.get(season=self, team=team)
+            rank.append((team_season.get_stats(**filters), team))
+        rank = sorted(rank, key=lambda x: x[0], reverse=True)
+        return rank
+
     def points_rank(self, scoring_code=None):
         contenders = self.contenders()
         scoring = self.get_scoring(scoring_code)
@@ -442,13 +451,25 @@ def check_delete_seat_restriction(team, season):
         return bool(seats_count)
         # return {'team': _('Seats with %(team)s exists in this season. Delete seats before.' % {'team': team})}
 
+    def get_results(self, limit_races=None):
+        results = Result.objects.filter(race__season=self.season, seat__team=self.team)
+        if isinstance(limit_races, int):
+            results = results.filter(race__round__lte=limit_races)
+        results = results.order_by('race__round')
+        return results
+
     def get_points(self):
-        results = Result.objects.filter(race__season=self.season, seat__team=self.team) \
-            .exclude(wildcard=True) \
-            .order_by('race__round')
+        results = self.get_results().exclude(wildcard=True).order_by('race__round')
         points_list = [result.points for result in results.all() if result.points is not None]
         return sum(points_list)
 
+    def get_filtered_results(self, **filters):
+        results = self.get_results()
+        return results.filter(**filters)
+
+    def get_stats(self, **filters):
+        return self.get_filtered_results(**filters).count()
+
     # def delete(self, *args, **kwargs):
     #     team = self.team
     #     season = self.season

driver27/record_filters.py

@@ -6,6 +6,16 @@
         'label': _('Pole'),
         'filter': {'qualifying__exact': 1}
      },
+    {
+        'code': 'FIRST-ROW',
+        'label': _('First row'),
+        'filter': {'qualifying__gte': 1, 'qualifying__lte': 2}
+    },
+    {
+        'code': 'COMEBACK-TO-TEN',
+        'label': _('Comeback to 10 firsts'),
+        'filter': {'qualifying__gte': 11, 'finish__gte': 1, 'finish__lte': 10}
+    },
     {
         'code': 'WIN',
         'label': _('Win'),
@@ -40,6 +50,10 @@
         'code': 'TOP5',
         'label': _('Top 5'),
         'filter': {'finish__gte': 1, 'finish__lte': 5}
+    },    {
+        'code': 'TOP10',
+        'label': _('Top 10'),
+        'filter': {'finish__gte': 1, 'finish__lte': 10}
     },
 
 ]

driver27/templates/driver27/record-team-view.html

@@ -0,0 +1,51 @@
+{% extends 'driver27/base.html' %}
+{% load i18n %}
+{% block dr27_menu %}
+    {% include 'driver27/season-menu-pre.html' %}
+    <li class="nav-item">
+        <a class="nav-link" href="{% url 'season-team-record' season.competition.slug season.year 'POLE' %}">{%  trans 'Pole record' %}</a>
+    </li>
+    <li class="nav-item">
+        <a class="nav-link" href="{% url 'season-team-record' season.competition.slug season.year 'WIN' %}">{%  trans 'Win record' %}</a>
+    </li>
+        <li class="nav-item">
+        <a class="nav-link" href="{% url 'season-team-record' season.competition.slug season.year 'FASTEST' %}">{%  trans 'Fastest lap record' %}</a>
+    </li>
+{% endblock %}
+{% block dr27_content %}
+    {{ block.super }}
+{% load champion_filter i18n %}
+    <form method="POST">
+    {% csrf_token %}
+        <select id="record" name="record" onchange="change_record()">
+            {% for record_code in record_codes %}
+            <option value="{% url 'season-team-record' season.competition.slug season.year record_code.code %}"
+            {% if record_code.code == record %}selected{% endif %}
+            >{{ record_code.label }}</option>
+            {% endfor %}
+        </select>
+    </form>
+<table class="table">
+    <tr>
+        <th>{%  trans 'Count' %}</th>
+        <th>{%  trans 'Team' %}</th>
+    </tr>
+    {% for count,team in rank %}
+        <tr>
+        <td>{{ count }}</td>
+        <td>{{ team }}</td>
+        </tr>
+    {% endfor %}
+
+    </table>
+{% endblock %}
+
+{% block bootstrap3_extra_script %}
+    {{ block.super }}
+    <script type="text/javascript">
+        function change_record(){
+            var record = $('#record').val();
+            location.href = record;
+        }
+    </script>
+{% endblock %}

driver27/urls/season.py

@@ -11,5 +11,6 @@
     url(r'^race$', views.race_list, name='season-race-list'),
     url(r'^race/(?P<race_id>\d+)$', views.race_view, name='season-race-view'),
     url(r'^race/(?P<race_id>\d+)/(?P<rank>[-\w\d]+)$', views.race_view, name='season-race-rank'),
-    url(r'^record/(?P<record>[-\w\d]+)$', views.record_view, name='season-record')
+    url(r'^record/(?P<record>[-\w\d]+)$', views.record_view, name='season-record'),
+    url(r'^team-record/(?P<record>[-\w\d]+)$', views.record_team_view, name='season-team-record')
 ]

driver27/views.py

@@ -164,3 +164,20 @@ def record_view(request, competition_slug, year, record):
                }
     tpl = 'driver27/record-view.html'
     return render(request, tpl, context)
+
+
+def record_team_view(request, competition_slug, year, record):
+    season = get_season(competition_slug, year)
+    record_config = get_record_config(record)
+    rank = season.team_stats_rank(**record_config['filter']) if record_config else None
+    title = _('%(record_label)s Team Record, %(season)s') \
+            % {'record_label': record_config['label'], 'season': season}
+    context = {'rank': rank,
+               'season': season,
+               'title': title,
+               'record': record,
+               'record_codes': DR27_RECORDS_FILTER
+               }
+    tpl = 'driver27/record-team-view.html'
+    return render(request, tpl, context)
+