mangroveorg/datawinners

# vim: ai ts=4 sts=4 et sw=4 encoding=utf-8
from datawinners.project.couch_view_helper import get_all_projects, remove_poll_questionnaires
from datawinners.settings import CRS_ORG_ID
from django.utils.translation import ugettext_lazy as _
from datawinners.main.database import get_database_manager
from datawinners.utils import get_organization
from mangrove.datastore.user_permission import get_questionnaires_for_user


def default_get_project_info(dbm, x):
    return x


def get_all_project_for_user(user, get_project_info_function=default_get_project_info, dbm=None):
    if user.get_profile().reporter:
        return [get_project_info_function(dbm, row['value'])
                          for row in get_all_projects(get_database_manager(user), user.get_profile().reporter_id)
                          if row['value'].get('is_poll',False) is False]

    
    if user.is_ngo_admin() or user.is_extended_user():
        questionnaires = [get_project_info_function(dbm, row['value']) for row in get_all_projects(get_database_manager(user))]
        return questionnaires

    questionnaires_as_datasender = [get_project_info_function(dbm, row['value']) for row in get_all_projects(get_database_manager(user), user.get_profile().reporter_id)]
    questionnaires_as_pm = get_questionnaires_for_user(user.id, get_database_manager(user),
                                                       project_info_function=get_project_info_function)
    for q_ds in questionnaires_as_datasender:
        already_exists = [True for q_pm in questionnaires_as_pm if q_pm.get('_id') == q_ds.get('project_id', q_ds.get('_id'))]
        if True not in already_exists:
           questionnaires_as_pm.append(q_ds)
    return questionnaires_as_pm



def get_visibility_settings_for(user):
    if user.get_profile().reporter:
        return "disable_link_for_reporter", "none"
    return "", ""


def get_page_heading(request):
    organization = get_organization(request)
    if request.user.get_profile().reporter:
        return "Data Submission"
    if organization.is_pro_sms:
        return "Questionnaires & Polls"
    else:
        return "Questionnaires"


def link(report_name, language):
    locale = '&__locale=fr_CA' if language == 'fr' else ''
    return '/birt-viewer/frameset?__report=crs/' + report_name + '.rptdesign' + locale


def get_reports_list(org_id, language):
    if org_id == CRS_ORG_ID:
        return [
            {'link': link('waybill_sent_and_received', language), 'name': _('Way Bill Sent vs Received Report'),
             'desc': _('Way bill sent vs received description')},
            {'link': link('CSR', language), 'name': _('CSR'), 'desc': _('Warehouse Commodity Status Report')},
            {'link': link('LSR', language), 'name': _('LSR'), 'desc': _('LSR description')},
            {'link': link('MSL', language), 'name': _('MSL'), 'desc': _('MSL description')},
            {'link': link('NCSR', language), 'name': _('CSR National'), 'desc': _('National Commodity Status Report')},
            {'link': link('recipient_status_report', language), 'name': _('Recipient Status Report'),
             'desc': _('Recipient Status Report description')},
            {'link': link('return_report', language), 'name': _('Return Report'),
             'desc': _('Return Report description')},
            {'link': link('billOfLadingVsWayBillPort', language), 'name': _('Bill Of Lading vs Way Bill Port'),
             'desc': _('Bill Of Lading vs Way Bill Port description')},
        ]
    return []