# 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', []) for email in emails: mail = email['email'] self.touch_model( model=Email, representative=representative, kind=('official' if mail.endswith(self.variant['mail_domain']) else 'other'), email=mail) # Official address off_name = self.variant['off_name'] official_addr, _ = self.touch_model(model=Address, representative=representative, country=self.france, city=self.variant['off_city'], street=self.variant['off_street'], number=self.variant['off_number'], postcode=self.variant['off_code'], kind='official', name=off_name ) # Addresses & phone numbers addresses = rep_json.get('adresses', []) for item in addresses: addr = None if 'geo' in item: props = item['geo'].get('properties', {}) name = '' if item['adresse'].lower().startswith('permanence'): name = 'Permanence' addr, _ = self.touch_model(model=Address, representative=representative, country=self.france, city=props.get('city', ''), street=props.get('street', ''), number=props.get('housenumber', ''), postcode=props.get('postcode', ''), kind='', name=name ) elif item['adresse'].lower().startswith(off_name.lower()): addr = official_addr if 'tel' in item: self.touch_model(model=Phone, address=addr, representative=representative, kind='', number=item['tel'] ) def main(stream=None): if not apps.ready: django.setup() ensure_chambers() an_importer = FranceDataImporter('AN') GenericImporter.pre_import(an_importer) sen_importer = FranceDataImporter('SEN') GenericImporter.pre_import(sen_importer) for data in ijson.items(stream or sys.stdin, ''): for rep in data: try: if rep['chambre'] == 'AN': an_importer.manage_rep(rep) elif rep['chambre'] == 'SEN': sen_importer.manage_rep(rep) except Exception: logger.exception('error trying to import rep %s', str(rep))