feat: add view for nonreporting centres

dodumosu · dodumosu · commit d7e75fc90696 · 2022-01-12T03:52:04.000+01:00
also, fix issues with centre edit view+template
diff --git a/locations/forms.py b/locations/forms.py
@@ -1,9 +1,13 @@
+import calendar
+
 from django import forms
-from .models import Location
+
+from br.utils import get_report_year_range
+from locations.models import Location
 
 
 def generate_edit_form(location, data=None):
-    state_choices = Location.objects.filter(type__name='state').values_list(
+    state_choices = Location.objects.filter(type__name='State').values_list(
         'id', 'name')
 
     def clean_location_field(form, field_name, location_type):
@@ -64,3 +68,22 @@ class CenterCreationForm(forms.Form):
     name = forms.CharField()
     lga = forms.ModelChoiceField(queryset=Location.objects.filter(
         type__name=u'LGA'))
+
+
+def _get_year_choices():
+    choices = [('', '----- Select year -----')]
+    year_range = get_report_year_range()
+    choices.extend([(yr, yr) for yr in range(year_range[0], year_range[1] + 1)])
+    return choices
+
+
+def _get_month_choices():
+    choices = [('', '----- Select month -----')]
+    choices.extend([(i, calendar.month_abbr[i]) for i in range(1, 13)])
+    return choices
+
+
+class NonReportingCentresFilterForm(forms.Form):
+    location = forms.ModelChoiceField(queryset=Location.objects.filter(type__name__in=['State', 'LGA']), required=False)
+    year = forms.ChoiceField(choices=_get_year_choices, required=False)
+    month = forms.ChoiceField(choices=_get_month_choices, required=False)
diff --git a/locations/templates/locations/center_edit.html b/locations/templates/locations/center_edit.html
@@ -1,5 +1,5 @@
 {% extends 'base/layout.html' %}
-{% load pipeline staticfiles %}
+{% load pipeline staticfiles widget_tweaks %}
 {% block stylesheets %}
 {% stylesheet 'centers' %}
 {% endblock %}
@@ -21,7 +21,7 @@
                 </div>
                 <div class="form-group">
                     <label for="{{ form.state.name }}">{{ form.state.label }}</label>
-                    {{ form.state }}
+                    {% render_field form.state class='form-control' %}
                 </div>
                 <div class="form-group">
                     <label for="{{ form.lga.id_for_label }}">{{ form.lga.label|upper }}</label>
@@ -46,6 +46,7 @@
 </div>
 {% endblock %}
 {% block scripts %}
+<script src="{% static 'base/vendor/jquery/dist/jquery.min.js' %}"></script>
 {% javascript 'centers' %}
 <script type="text/javascript">
 $(document).ready(function() {
diff --git a/locations/templates/locations/non_reporting_centre_list.html b/locations/templates/locations/non_reporting_centre_list.html
@@ -0,0 +1,132 @@
+{% extends 'base/layout.html' %}{% load bootstrap_pagination location_tags static widget_tweaks %}
+{% block title %}Birth Registration Statistics &middot; {{ page_title }}{% endblock %}
+{% block usermenu %}
+{% include 'common/usermenu.html' %}
+{% endblock %}
+{% block stylesheets %}
+{{ block.super }}
+<link rel="stylesheet" type="text/css" href="{% static 'css/select2/select2.min.css' %}">
+<style type="text/css">
+    .select2-container.form-control {
+        display: inline-block;
+    }
+</style>
+{% endblock %}
+{% block masthead %}
+<div class="row">
+    <div class="col-lg-12 title">
+        <ol class="breadcrumb">
+            <li><a href="{% url "br:dashboard" %}">Birth Registration</a></li>
+            <li class="active">{{ page_title }}</li>
+        </ol>
+        <h1 class="page-title">{{ page_title }}</h1>
+    </div>
+</div>
+{% endblock %}
+{% block content %}
+<div class="row">
+    <div class="col-lg-12">
+        <section class="widget">
+            <form class="form-horizontal form-inline" role="form" id="filter_form">
+                <fieldset>
+                    <div class="form-group row">
+                        <div class="col-md-12">
+                            {% render_field filter_form.year class="form-control" %}
+                            {% render_field filter_form.month class="form-control" %}
+                            <button type="submit" class="btn btn-primary" id="filter_button">Filter</button>
+                            <a class="btn btn-inverse" href="{% url 'locations:non_reporting_center_list' %}">Clear</a>
+                            <a class="btn btn-primary-outline" href="{% url 'locations:non_reporting_center_list' %}?export=1&year={{ year }}&month={{ month }}">Export</a>
+                        </div>
+                    </div>
+                </fieldset>
+            </form>
+        </section>
+    </div>
+</div>
+<div class="row">
+    <div class="col-lg-12">
+        <section class="widget">
+            <div class="widget-table">
+                <table class="table table-striped table-bordered table-hover">
+                    <thead>
+                        <tr>
+                            <th>State</th>
+                            <th>LGA</th>
+                            <th>Name</th>
+                            <th>Code</th>
+                            <th>Active?</th>
+                            <th>Last report received</th>
+                            <th></th>
+                        </tr>
+                    </thead>
+                    <tbody>
+                        {% for centre in centres %}
+                        {% with ancestors=centre.get_ancestors %}
+                        <tr>
+                            <td>{{ ancestors|extract_ancestor_name:'State' }}</td>
+                            <td>{{ ancestors|extract_ancestor_name:'LGA' }}</td>
+                            <td>{{ centre.name }}</td>
+                            <td>{{ centre.code }}</td>
+                            {% if centre.active %}
+                            <td class="success"><i class="glyphicon glyphicon-ok"></i></td>
+                            {% else %}
+                            <td class="danger"><i class="glyphicon glyphicon-remove"></i></td>
+                            {% endif %}
+                            <td>{{ centre.latest_birth_report_time|date:'SHORT_DATE_FORMAT' }}</td>
+                            <td><a href="{% url 'locations:center_edit' pk=centre.pk %}">Edit</a></td>
+                        </tr>
+                        {% endwith %}
+                        {% empty %}
+                        <tr>
+                            <td class="table-warning text-xs-center" colspan="7">No non-reporting centers found</td>
+                        </tr>
+                        {% endfor %}
+                    </tbody>
+                </table>
+            </div>
+        </section>
+    </div>
+</div>
+<div class="row">
+    <div class="col-lg-12">
+        <div class="text-xs-center">
+            {% bootstrap_paginate page_obj range=10 show_first_last="true" %}
+        </div>
+    </div>
+</div>
+{% endblock content %}
+{% block scripts %}
+{{ block.super }}
+<script src="{% static "js/select2/select2.min.js" %}"></script>
+<script>
+document.addEventListener('DOMContentLoaded', function () {
+    var endpoint = '{% url "api:location_list_typed" %}?type=lga';
+    $('#{{ filter_form.lga.id_for_label }}').select2({
+        ajax: {
+            url: endpoint,
+            data: function(params) {
+              return {q: params.term};
+            },
+            dataType: 'json',
+            quietMillis: 250,
+            processResults: function(data, params) {
+              return {results: data.results};
+            }
+        },
+        width: '30em',
+        allowClear: true,
+        minimumInputLength: 2,
+        cache: true,
+        templateResult: function(item) {
+            return item.name + ' (' + item.type + ')';
+        },
+        templateSelection: function(item) {
+            if (item.text)
+                return item.text;
+            else
+                return item.name + ' (' + item.type + ')';
+        }
+    });
+});
+</script>
+{% endblock %}
diff --git a/locations/urls.py b/locations/urls.py
@@ -1,10 +1,11 @@
 #!/usr/bin/env python
 # vim: ai ts=4 sts=4 et sw=4
 from django.conf.urls import url
-from locations.views import *
+from locations import views
 
 urlpatterns = [
-    url(r'^center/new/?$', CenterCreationView.as_view(), name='center_add'),
-    url(r'^centers/?$', CenterListView.as_view(), name='center_list'),
-    url(r'^center/(?P<pk>\d+)/?$', CenterUpdateView.as_view(), name='center_edit'),
+    url(r'^center/new/?$', views.CenterCreationView.as_view(), name='center_add'),
+    url(r'^centers/?$', views.CenterListView.as_view(), name='center_list'),
+    url(r'^non-reporting-centers/?$', views.NonReportingCentresView.as_view(), name='non_reporting_center_list'),
+    url(r'^center/(?P<pk>\d+)/?$', views.CenterUpdateView.as_view(), name='center_edit'),
 ]
diff --git a/locations/views.py b/locations/views.py
@@ -1,4 +1,5 @@
 # vim: ai ts=4 sts=4 et sw=4
+import csv
 import json
 
 from django.contrib import messages
@@ -7,17 +8,23 @@
 from django.core.exceptions import ObjectDoesNotExist
 from django.core.urlresolvers import reverse, reverse_lazy
 from django.db import connection
+from django.db.models import (
+    Case, Count, IntegerField, OuterRef, Subquery, When)
 from django.forms.formsets import formset_factory
-from django.http import HttpResponse, HttpResponseRedirect
+from django.http import (
+    HttpResponse, HttpResponseRedirect, StreamingHttpResponse)
 from django.shortcuts import get_object_or_404
+from django.utils.timezone import now
 from django.views.generic import ListView, FormView, TemplateView
 from drf_yasg.utils import swagger_auto_schema
 import pandas as pd
 
 from django.conf import settings
 
+from br.models import BirthRegistration
 from locations import raw_queries
-from locations.forms import generate_edit_form, CenterCreationForm
+from locations.forms import (
+    generate_edit_form, CenterCreationForm, NonReportingCentresFilterForm)
 from locations.filters import CenterFilterSet
 from locations.models import Facility, Location, LocationType
 
@@ -57,11 +64,14 @@ def get_queryset(self):
 
 class CenterUpdateView(
         LoginRequiredMixin, PermissionRequiredMixin, FormView):
+    permission_required = 'locations.edit_location'
     template_name = 'locations/center_edit.html'
 
     def get_context_data(self, **kwargs):
         context = super(CenterUpdateView, self).get_context_data(**kwargs)
 
+        self.object = self.get_object()
+
         context['page_title'] = 'Edit center: {}'.format(self.object.name)
         context[u'location'] = self.object
 
@@ -77,8 +87,7 @@ def form_valid(self, form):
         center = Location.objects.get(pk=form.cleaned_data['id'])
         center.name = form.cleaned_data['name']
         center.code = form.cleaned_data['code']
-        center.parent = Location.objects.get_object_or_404(
-            pk=form.cleaned_data['lga'])
+        center.parent = get_object_or_404(Location, pk=form.cleaned_data['lga'])
         center.active = form.cleaned_data['active']
         center.save()
         center.facilities.update(
@@ -87,14 +96,14 @@ def form_valid(self, form):
 
         return HttpResponseRedirect(self.get_success_url())
 
-    def get_form(self, form_class):
-        return generate_edit_form(self.object)
+    def get_form(self, form_class=None):
+        return generate_edit_form(self.get_object())
 
     def get_queryset(self):
         return Location.objects.filter(type__name='RC')
 
     def get_success_url(self):
-        return reverse('center_list')
+        return reverse('locations:center_list')
 
     def post(self, request, *args, **kwargs):
         form = generate_edit_form(self.get_object(), request.POST)
@@ -195,3 +204,86 @@ def facilities(request):
     facility_dataframe.to_csv(response, encoding='UTF-8', index=False)
 
     return response
+
+
+
+class CSVBuffer(object):
+    def write(self, value):
+        return value
+
+
+def _get_rows(centres):
+    yield ['Centre', 'Code', 'LGA', 'State', 'Active?', 'Last report date']
+
+    for centre in centres:
+        last_report_time = centre.latest_birth_report_time()
+        yield [
+            centre.name,
+            centre.code,
+            centre.parent.name,
+            centre.parent.parent.name,
+            'Yes' if centre.active else 'No',
+            last_report_time.strftime('%d-%m-%Y') if last_report_time else 'N/A'
+        ]
+
+
+class NonReportingCentresView(LoginRequiredMixin, ListView):
+    context_object_name = 'centres'
+    page_title = 'Non-reporting centres'
+    paginate_by = settings.PAGE_SIZE
+    template_name = 'locations/non_reporting_centre_list.html'
+
+    def get(self, request, *args, **kwargs):
+        centres = self.get_queryset()
+
+        year = request.GET.get('year') or None
+        month = request.GET.get('month') or None
+
+        if request.GET.get('export'):
+            writer = csv.writer(CSVBuffer())
+            response = StreamingHttpResponse((writer.writerow(row) for row in _get_rows(centres)), content_type='text/csv')
+            filename = 'non-reporting-centres-{}-{}.csv'.format(
+                year,
+                str(month).zfill(2)
+            )
+            response['Content-Disposition'] = 'attachment; filename={}'.format(
+                filename)
+            
+            return response
+        
+        return super(NonReportingCentresView, self).get(request, *args, **kwargs)
+
+    def get_context_data(self, **kwargs):
+        context = super(NonReportingCentresView, self).get_context_data(**kwargs)
+
+        context['page_title'] = self.page_title
+        context['filter_form'] = self.filter_form
+        context['year'] = self.request.GET.get('year') or now().year
+        context['month'] = self.request.GET.get('month') or ''
+
+        return context
+
+    def get_queryset(self):
+        self.filter_form = NonReportingCentresFilterForm(self.request.GET)
+        if self.filter_form.is_valid():
+            filter_data = self.filter_form.cleaned_data
+        else:
+            filter_data = {}
+
+        filter_kwargs = {}
+        year = filter_data.get('year') or now().year
+        filter_kwargs['birthregistration_records__time__year'] = int(year)
+        month = filter_data.get('month')
+        if month:
+            filter_kwargs['birthregistration_records__time__month'] = int(month)
+
+        if len(filter_kwargs) == 0:
+            centres = Location.objects.none()
+        else:
+            centres = Location.objects.filter(
+                type__name='RC'
+            ).annotate(
+                cnt=Count(Case(When(then=1, **filter_kwargs)))
+            ).filter(cnt=0)
+
+        return centres
diff --git a/unicefng/templates/common/usermenu.html b/unicefng/templates/common/usermenu.html
@@ -2,6 +2,7 @@
 {% if user.is_authenticated %}
 {% if perms.br.change_birthregistration %}<li><a class="dropdown-item" href="{% url 'br:reports_list' %}">BR reports</a></li>{% endif %}
 {% if perms.br.change_birthregistration %}<li><a class="dropdown-item" href="{% url 'locations:center_list' %}">BR centers</a></li>{% endif %}
+{% if perms.br.change_birthregistration %}<li><a class="dropdown-item" href="{% url 'locations:non_reporting_center_list' %}">Non-reporting BR centers</a></li>{% endif %}
 <li><a class="dropdown-item" href="{% url 'br:help' %}">BR help</a></li>
 <div class="dropdown-divider"></div>
 {% if perms.dr.change_deathreport %}<li><a class="dropdown-item" href="{% url 'dr:dr_report_list' %}">DR reports</a></li>{% endif %}
