mangroveorg/datawinners

# coding=utf-8
import codecs
import json
import os
from datawinners.project.couch_view_helper import get_all_projects
from mangrove.form_model.field import field_to_json
from mangrove.form_model.form_model import FormModel
from datawinners.main.database import get_db_manager


def generate_template_data():
    templates = []
    #database which has template specific projects.
    test_dbm = get_db_manager('hni_templates_aoi959205')
    projects = get_all_projects(test_dbm)
    for doc in projects:
        form_model = FormModel.get(test_dbm, doc.id)
        json_obj = {}
        json_obj.update({"name": form_model.name})
        json_obj.update({"language": form_model.activeLanguages[0]})
        json_obj.update({"category": _get_category(form_model)})
        json_obj.update({"form_code": form_model.form_code})
        fields = _remove_entity_field(form_model)
        json_obj.update({"json_fields": [field_to_json(f) for f in fields]})
        json_obj.update({"validators": [validator.to_json() for validator in form_model.validators]})
        templates.append(json_obj)

    file = os.path.dirname(__file__) + '/../datawinners/questionnaire/template_data.json'
    with codecs.open(file, 'w', encoding='utf-8') as outfile:
        json.dump(templates, outfile, ensure_ascii=False, indent=4)


def _get_category(project):
    project_name = project.name
    language = project.activeLanguages[0]
    project_name = project_name.strip()
    project_to_category_map = get_category_mapping_fr() if language=='fr' else get_category_mapping_en()
    for category, project_names in project_to_category_map.iteritems():
        if project_name in [name.lower() for name in project_names]:
            return category


def _remove_entity_field(form_model):
    return [each for each in form_model.fields if not each.code == "eid"]


def get_category_mapping_en():
    map = {}
    map.update({'Health':
                    ['Monthly Client Report', 'Monthly Stock Report', 'Patient Interview',
                     'Weekly Sentinel Site Survey']})
    map.update({'Food Security': ['Waybill Sent', 'Waybill Received']})
    map.update({'Education': ['Student Census', 'Grant Reception', 'Textbook Reception', 'Standardized Test Results',
                              'Early Grade Reading Assessment']})
    map.update({'Early Warning': ['Weekly assessment', 'Fast Onset']})
    map.update({'Agriculture': ['Livestock Census']})
    map.update({'Commercial': ['Invoice']})
    map.update({'Socio-Economic': ['Household Survey']})
    return map

def get_category_mapping_fr():
    map = {}
    map.update({u'Santé':
                    [u'rapport mensuel sur les clients', u'Rapport de stock mensuel', u'Entretien avec la patiente',
                     u'Enquête hebdomadaire auprès des sites sentinelles']})
    map.update({u'Sécurité alimentaire': [u'Lettre de transport envoyée', u'Lettre de transport reçue']})
    map.update({u'Éducation': [u'Recensement des étudiants', u'Réception de subvention', u'Réception de livres scolaires', u'Résultats de test standardisés',
                              u'Evaluation de la lecture en primaire']})
    map.update({u'Alerte précoce': [u'contrôle hebdomadaire', u'catastrophes survenant rapidement']})
    map.update({u'Agriculture': [u'recensement du cheptel']})
    map.update({u'Commercial': [u'Facture']})
    map.update({u'Socio-économique': [u'enquête sur les ménages']})
    return map


generate_template_data()