Import representatives module

Note: it **must** be kept loosely coupled with other modules in this
repository. It was merged here just to remove the burden of having
several repositories to manage: volunteer time is precious.

.coveragerc

@@ -2,3 +2,7 @@
 omit =
 	src/representatives_positions/contrib/*
 	src/representatives_recommendations/contrib/*
+	src/representatives/tests/*
+	src/representatives/migrations/*
+	src/representatives/contrib/francedata/tests/*
+	src/representatives/contrib/parltrack/tests/*

src/representatives/admin.py

+# coding: utf-8
+
+# This file is part of compotista.
+#
+# compotista is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as
+# published by the Free Software Foundation, either version 3 of
+# the License, or any later version.
+#
+# compotista is distributed in the hope that it will
+# be useful, but WITHOUT ANY WARRANTY; without even the implied
+# warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
+# See the GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU General Affero Public
+# License along with django-representatives.
+# If not, see <http://www.gnu.org/licenses/>.
+#
+# Copyright (C) 2013 Laurent Peuch <cortex@worlddomination.be>
+# Copyright (C) 2015 Arnaud Fabre <af@laquadrature.net>
+
+from django.contrib import admin
+
+from .models import (Address, Constituency, Country, Email, Group, Mandate,
+                     Phone, Representative, WebSite)
+
+
+class EmailInline(admin.TabularInline):
+    model = Email
+    extra = 0
+
+
+class WebsiteInline(admin.TabularInline):
+    model = WebSite
+    extra = 0
+
+
+class AdressInline(admin.StackedInline):
+    model = Address
+    extra = 0
+
+
+class PhoneInline(admin.TabularInline):
+    model = Phone
+    extra = 0
+
+
+class MandateInline(admin.StackedInline):
+    model = Mandate
+    extra = 0
+
+
+class RepresentativeAdmin(admin.ModelAdmin):
+    list_display = ('id', 'full_name', 'gender', 'birth_place')
+    search_fields = ('first_name', 'last_name', 'birth_place')
+    list_filter = ('gender', )
+    inlines = [
+        PhoneInline,
+        EmailInline,
+        WebsiteInline,
+        AdressInline,
+        MandateInline
+    ]
+
+
+class GroupAdmin(admin.ModelAdmin):
+    list_display = ('id', 'name', 'abbreviation', 'kind')
+    list_filter = ('kind',)
+
+
+class MandateAdmin(admin.ModelAdmin):
+    list_display = (
+        'id',
+        'representative',
+        'group',
+        'role',
+        'constituency',
+        'begin_date',
+        'end_date')
+    search_fields = ('representative', 'group', 'constituency')
+
+
+admin.site.register(Representative, RepresentativeAdmin)
+admin.site.register(Country)
+admin.site.register(Mandate, MandateAdmin)
+admin.site.register(Group, GroupAdmin)
+admin.site.register(Constituency)

src/representatives/api.py

+from django.db import models
+
+from rest_framework import (
+    filters,
+    pagination,
+    renderers,
+    viewsets,
+)
+
+from rql_filter.backend import RQLFilterBackend
+
+from representatives.serializers import (
+    ChamberSerializer,
+    ConstituencySerializer,
+    CountrySerializer,
+    GroupSerializer,
+    MandateSerializer,
+    RepresentativeDetailSerializer,
+    RepresentativeSerializer,
+)
+
+from .models import (
+    Address,
+    Chamber,
+    Constituency,
+    Country,
+    Group,
+    Mandate,
+    Phone,
+    Representative,
+)
+
+
+class DefaultWebPagination(pagination.PageNumberPagination):
+    default_web_page_size = 10
+
+    def get_page_size(self, request):
+        web = isinstance(request.accepted_renderer,
+                         renderers.BrowsableAPIRenderer)
+        size = pagination.PageNumberPagination.get_page_size(self, request)
+
+        if web and not size:
+            return self.default_web_page_size
+
+        return size
+
+
+class RepresentativeViewSet(viewsets.ReadOnlyModelViewSet):
+    """
+    API endpoint that allows representatives to be viewed.
+    """
+    queryset = Representative.objects.all()
+    filter_backends = (
+        filters.DjangoFilterBackend,
+        filters.SearchFilter,
+        filters.OrderingFilter,
+        RQLFilterBackend
+    )
+    filter_fields = {
+        'active': ['exact'],
+        'slug': ['exact', 'icontains'],
+        'id': ['exact'],
+        'first_name': ['exact', 'icontains'],
+        'last_name': ['exact', 'icontains'],
+        'full_name': ['exact', 'icontains'],
+        'gender': ['exact'],
+        'birth_place': ['exact'],
+        'birth_date': ['exact', 'gte', 'lte'],
+    }
+    search_fields = ('first_name', 'last_name', 'slug')
+    ordering_fields = ('id', 'birth_date', 'last_name', 'full_name')
+    pagination_class = DefaultWebPagination
+
+    def get_queryset(self):
+        qs = super(RepresentativeViewSet, self).get_queryset()
+        qs = qs.prefetch_related(
+            'email_set',
+            'website_set',
+            models.Prefetch(
+                'address_set',
+                queryset=Address.objects.select_related('country')
+            ),
+            models.Prefetch(
+                'phone_set',
+                queryset=Phone.objects.select_related('address__country')
+            ),
+            'mandates',
+        )
+        return qs
+
+    def list(self, request):
+        self.serializer_class = RepresentativeSerializer
+        return super(RepresentativeViewSet, self).list(request)
+
+    def retrieve(self, request, pk=None):
+        self.serializer_class = RepresentativeDetailSerializer
+        return super(RepresentativeViewSet, self).retrieve(request, pk)
+
+
+class MandateViewSet(viewsets.ReadOnlyModelViewSet):
+    """
+    API endpoint that allows mandates to be viewed.
+    """
+    pagination_class = DefaultWebPagination
+    queryset = Mandate.objects.select_related('representative')
+    serializer_class = MandateSerializer
+
+    filter_backends = (
+        filters.DjangoFilterBackend,
+        filters.SearchFilter,
+        filters.OrderingFilter,
+        RQLFilterBackend
+    )
+    filter_fields = {
+        'id': ['exact'],
+        'group__name': ['exact', 'icontains'],
+        'group__abbreviation': ['exact'],
+    }
+    search_fields = ('group__name', 'group__abbreviation')
+
+
+class ConstituencyViewSet(viewsets.ReadOnlyModelViewSet):
+    pagination_class = DefaultWebPagination
+    queryset = Constituency.objects.all()
+    serializer_class = ConstituencySerializer
+
+    filter_backends = (
+        RQLFilterBackend,
+    )
+
+
+class GroupViewSet(viewsets.ReadOnlyModelViewSet):
+    pagination_class = DefaultWebPagination
+    queryset = Group.objects.all()
+    serializer_class = GroupSerializer
+
+    filter_backends = (
+        RQLFilterBackend,
+    )
+
+
+class ChamberViewSet(viewsets.ReadOnlyModelViewSet):
+    pagination_class = DefaultWebPagination
+    queryset = Chamber.objects.all()
+    serializer_class = ChamberSerializer
+
+    filter_backends = (
+        RQLFilterBackend,
+    )
+
+
+class CountryViewSet(viewsets.ReadOnlyModelViewSet):
+    pagination_class = DefaultWebPagination
+    queryset = Country.objects.all()
+    serializer_class = CountrySerializer
+
+    filter_backends = (
+        RQLFilterBackend,
+    )

src/representatives/contrib/francedata/import_representatives.py

+# coding: utf-8
+
+import logging
+import re
+import sys
+from datetime import datetime
+
+import django.dispatch
+import ijson
+import django
+from django.apps import apps
+from django.db import transaction
+from django.utils import timezone
+from django.utils.text import slugify
+
+from representatives.models import (Country, Mandate, Email, Address, WebSite,
+                                    Representative, Constituency, Phone, Group,
+                                    Chamber)
+from variants import FranceDataVariants
+
+
+logger = logging.getLogger(__name__)
+
+representative_pre_import = django.dispatch.Signal(
+    providing_args=['representative_data'])
+
+
+def _get_mdef_item(mdef, item, json, default=None):
+    if item in mdef:
+        try:
+            return mdef[item] % json
+        except:
+            return default
+
+    if '%s_path' % item in mdef:
+        return _get_path(json, mdef['%s_path' % item])
+
+    if '%s_fn' % item in mdef:
+        return mdef['%s_fn' % item](json)
+
+    return default
+
+
+def _parse_date(date):
+    return datetime.strptime(date, "%Y-%m-%d").date()
+
+
+def _create_mandate(representative, group, constituency, role='',
+                    begin_date=None, end_date=None):
+    mandate, _ = Mandate.objects.get_or_create(
+        representative=representative,
+        group=group,
+        constituency=constituency,
+        role=role,
+        begin_date=begin_date,
+        end_date=end_date
+    )
+
+    if _:
+        logger.debug('Created mandate %s', mandate.pk)
+
+
+def _get_path(dict_, path):
+    '''
+    Get value at specific path in dictionary. Path is specified by slash-
+    separated string, eg _get_path(foo, 'bar/baz') returns foo['bar']['baz']
+    '''
+    cur = dict_
+    for part in path.split('/'):
+        cur = cur[part]
+    return cur
+
+
+class GenericImporter(object):
+
+    def pre_import(self):
+        self.import_start_datetime = timezone.now()
+
+    def touch_model(self, model, **data):
+        '''
+        This method create or look up a model with the given data
+        it saves the given model if it exists, updating its
+        updated field
+        '''
+
+        instance, created = model.objects.get_or_create(**data)
+
+        if not created:
+            if instance.updated < self.import_start_datetime:
+                instance.save()     # Updates updated field
+
+        return (instance, created)
+
+
+def ensure_chambers():
+    """
+    Ensures chambers are created
+    """
+    france = Country.objects.get(name="France")
+    for key in ('AN', 'SEN'):
+        variant = FranceDataVariants[key]
+        Chamber.objects.get_or_create(name=variant['chamber'],
+                                      abbreviation=variant['abbreviation'],
+                                      country=france)
+
+
+class FranceDataImporter(GenericImporter):
+    url = 'http://francedata.future/data/parlementaires.json'
+
+    def parse_date(self, date):
+        return _parse_date(date)
+
+    def __init__(self, variant):
+        self.france = Country.objects.get(name="France")
+        self.variant = FranceDataVariants[variant]
+        self.chamber = Chamber.objects.get(name=self.variant['chamber'])
+        self.ch_constituency, _ = Constituency.objects.get_or_create(
+            name=self.variant['chamber'], country=self.france)
+
+    @transaction.atomic
+    def manage_rep(self, rep_json):
+        '''
+        Import a rep as a representative from the json dict fetched from
+        FranceData (which comes from nosdeputes.fr)
+        '''
+
+        # Some versions of memopol will connect to this and skip inactive reps.
+        responses = representative_pre_import.send(sender=self,
+                representative_data=rep_json)
+
+        for receiver, response in responses:
+            if response is False:
+                logger.debug(
+                    'Skipping MEP %s', rep_json['nom'])
+                return
+
+        changed = False
+        slug = slugify('%s-%s' % (
+            rep_json['nom'] if 'nom' in rep_json
+            else rep_json['prenom'] + " " + rep_json['nom_de_famille'],
+            _parse_date(rep_json["date_naissance"])
+        ))
+
+        try:
+            representative = Representative.objects.get(slug=slug)
+        except Representative.DoesNotExist:
+            representative = Representative(slug=slug)
+            changed = True
+
+        if rep_json['num_circo'] == 'non disponible':
+            rep_json['num_circo'] = 'nd'
+
+        # Save representative attributes
+        self.import_representative_details(representative, rep_json, changed)
+
+        self.add_mandates(representative, rep_json)
+
+        self.add_contacts(representative, rep_json)
+
+        logger.debug('Imported MEP %s', unicode(representative))
+
+        return representative
+
+    def import_representative_details(self, representative, rep_json, changed):
+        active = True
+        if rep_json.get("ancien_depute", 0) == 1:
+            active = False
+        if rep_json.get("ancien_senateur", 0) == 1:
+            active = False
+        if representative.active != active:
+            representative.active = active
+            changed = True
+
+        if rep_json.get("date_naissance"):
+            birth_date = _parse_date(rep_json["date_naissance"])
+            if representative.birth_date != birth_date:
+                representative.birth_date = birth_date
+                changed = True
+
+        if rep_json.get("lieu_naissance"):
+            birth_place = rep_json["lieu_naissance"]
+            if representative.birth_place != birth_place:
+                representative.birth_place = birth_place
+                changed = True
+
+        photo = rep_json['photo_url']
+        if representative.photo != photo:
+            representative.photo = photo
+            changed = True
+
+        first_name = rep_json['prenom']
+        if representative.first_name != first_name:
+            representative.first_name = first_name
+            changed = True
+
+        last_name = rep_json['nom_de_famille']
+        if representative.last_name != last_name:
+            representative.last_name = last_name
+            changed = True
+
+        full_name = rep_json["nom"]
+        if representative.full_name != full_name:
+            representative.full_name = full_name
+            changed = True
+
+        gender_convertion_dict = {u"F": 1, u"H": 2}
+        if 'sexe' in rep_json:
+            gender = gender_convertion_dict.get(rep_json['sexe'], 0)
+        else:
+            gender = 0
+        if representative.gender != gender:
+            representative.gender = gender
+            changed = True
+
+        if changed:
+            representative.save()
+
+    def add_mandates(self, representative, rep_json):
+        '''
+        Create mandates from rep data based on variant configuration
+        '''
+
+        # Mandate in country group for party constituency
+        if rep_json.get('parti_ratt_financier'):
+            constituency, _ = Constituency.objects.get_or_create(
+                name=rep_json.get('parti_ratt_financier'), country=self.france)
+
+            group, _ = self.touch_model(model=Group,
+                                        abbreviation=self.france.code,
+                                        kind='country',
+                                        name=self.france.name)
+
+            _create_mandate(representative, group, constituency, 'membre')
+
+        # Configurable mandates
+        for mdef in self.variant['mandates']:
+            if mdef.get('chamber', False):
+                chamber = self.chamber
+            else:
+                chamber = None
+
+            if 'from' in mdef:
+                elems = mdef['from'](rep_json)
+            else:
+                elems = [rep_json]
+
+            for elem in elems:
+                name = _get_mdef_item(mdef, 'name', elem, '')
+                abbr = _get_mdef_item(mdef, 'abbr', elem, '')
+
+                group, _ = self.touch_model(model=Group,
+                                            abbreviation=abbr,
+                                            kind=mdef['kind'],
+                                            chamber=chamber,
+                                            name=name)
+
+                role = _get_mdef_item(mdef, 'role', elem, 'membre')
+                start = _get_mdef_item(mdef, 'start', elem, None)
+                if start is not None:
+                    start = _parse_date(start)
+                end = _get_mdef_item(mdef, 'end', elem, None)
+                if end is not None:
+                    end = _parse_date(end)
+
+                _create_mandate(representative, group, self.ch_constituency,
+                                role, start, end)
+
+                logger.debug(
+                    '%s => %s: %s of "%s" (%s) %s-%s' % (rep_json['slug'],
+                    mdef['kind'], role, name, abbr, start, end))
+
+    def add_contacts(self, representative, rep_json):
+        # Chamber page
+        changed = False
+        try:
+            site = WebSite.objects.get(kind=self.variant['abbreviation'],
+                                       representative=representative)
+        except WebSite.DoesNotExist:
+            site = WebSite(kind=self.variant['abbreviation'],
+                           representative=representative)
+            changed = True
+
+        if site.url != rep_json[self.variant['chamber_url_field']]:
+            site.url = rep_json[self.variant['chamber_url_field']]
+            changed = True
+
+        if changed:
+            site.save()
+
+        # Websites
+        websites = rep_json.get('sites_web', [])
+        for site in websites:
+            if re.search(r'facebook\.com', site['site']):
+                self.touch_model(model=WebSite,
+                                 url=site['site'],
+                                 kind='facebook',
+                                 representative=representative
+                                 )
+            elif not re.search(r'twitter\.com', site['site']):
+                self.touch_model(model=WebSite,
+                                 url=site['site'],
+                                 representative=representative
+                                 )
+
+        # Twitter
+        if rep_json.get('twitter'):
+            tid = rep_json.get('twitter')
+            self.touch_model(model=WebSite,
+                             representative=representative,
+                             kind='twitter',
+                             url='http://twitter.com/%s' % tid
+                             )
+
+        # E-mails
+        emails = rep_json.get('emails', [])