#!/usr/bin/python2.5

#

# Copyright 2008 the Melange authors.

#

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

#   http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

"""Views for Organizations.

"""

__authors__ = [

    '"Madhusudan.C.S" <madhusudancs@gmail.com>',

    '"Augie Fackler" <durin42@gmail.com>',

    '"Sverre Rabbelier" <sverre@rabbelier.nl>',

    '"Lennard de Rijk" <ljvderijk@gmail.com>',

  ]

import itertools

from django import forms

from django.utils import simplejson

from django.utils.translation import ugettext

from soc.logic import cleaning

from soc.logic import dicts

from soc.logic import accounts

from soc.logic.helper import timeline as timeline_helper

from soc.logic.models import mentor as mentor_logic

from soc.logic.models import organization as org_logic

from soc.logic.models import org_admin as org_admin_logic

from soc.logic.models import org_app as org_app_logic

from soc.logic.models import user as user_logic

from soc.views import helper

from soc.views import out_of_band

from soc.views.helper import access

from soc.views.helper import decorators

from soc.views.helper import dynaform

from soc.views.helper import lists

from soc.views.helper import redirects

from soc.views.helper import widgets

from soc.views.models import group

import soc.models.organization

import soc.logic.models.organization

class View(group.View):

  """View methods for the Organization model.

  """

  DEF_ACCEPTED_PROJECTS_MSG_FMT = ugettext("These projects have"

      " been accepted into %s. You can learn more about"

      " each project by visiting the links below.")

  def __init__(self, params=None):

    """Defines the fields and methods required for the base View class

    to provide the user with list, public, create, edit and delete views.

    Params:

      original_params: a dict with params for this View

    """

    from soc.views.models import program as program_view

    rights = access.Checker(params)

    rights['any_access'] = ['allow']

    rights['show'] = ['allow']

    rights['create'] = ['checkIsDeveloper']

    rights['edit'] = [('checkHasActiveRoleForKeyFieldsAsScope',

                           org_admin_logic.logic,),

                      ('checkGroupIsActiveForLinkId', org_logic.logic)]

    rights['delete'] = ['checkIsDeveloper']

    rights['home'] = ['allow']

    rights['public_list'] = ['allow']

    rights['apply_mentor'] = ['checkIsUser']

    rights['list_requests'] = [('checkHasActiveRoleForKeyFieldsAsScope',

                                org_admin_logic.logic)]

    rights['list_roles'] = [('checkHasActiveRoleForKeyFieldsAsScope',

                             org_admin_logic.logic)]

    rights['applicant'] = [('checkIsApplicationAccepted',

                            org_app_logic.logic)]

    rights['list_proposals'] = [('checkHasAny', [

        [('checkHasActiveRoleForKeyFieldsAsScope', [org_admin_logic.logic]),

         ('checkHasActiveRoleForKeyFieldsAsScope', [mentor_logic.logic])]

        ])]

    new_params = {}

    new_params['logic'] = soc.logic.models.organization.logic

    new_params['rights'] = rights

    new_params['scope_view'] = program_view

    new_params['scope_redirect'] = redirects.getCreateRedirect

    new_params['name'] = "Organization"

    new_params['url_name'] = "org"

    new_params['document_prefix'] = "org"

    new_params['sidebar_grouping'] = 'Organizations'

    new_params['public_template'] = 'soc/organization/public.html'

    new_params['list_row'] = 'soc/organization/list/row.html'

    new_params['list_heading'] = 'soc/organization/list/heading.html'

    new_params['home_template'] = 'soc/organization/home.html'

    new_params['application_logic'] = org_app_logic

    new_params['group_applicant_url'] = True

    new_params['sans_link_id_public_list'] = True

    new_params['extra_dynaexclude'] = ['slots', 'slots_calculated',

                                       'nr_applications', 'nr_mentors']

    patterns = []

    patterns += [

        (r'^%(url_name)s/(?P<access_type>apply_mentor)/%(scope)s$',

        '%(module_package)s.%(module_name)s.apply_mentor',

        "List of all %(name_plural)s you can apply to"),

        (r'^%(url_name)s/(?P<access_type>list_proposals)/%(key_fields)s$',

        '%(module_package)s.%(module_name)s.list_proposals',

        "List of all Student Proposals for this %(name)s"),

        ]

    new_params['extra_django_patterns'] = patterns

    new_params['create_dynafields'] = [

        {'name': 'link_id',

         'base': forms.fields.CharField,

         'label': 'Organization Link ID',

         },

        ]

    new_params['create_extra_dynaproperties'] = {

        'scope_path': forms.CharField(widget=forms.HiddenInput,

                                   required=True),

        'description': forms.fields.CharField(

            widget=helper.widgets.FullTinyMCE(

                attrs={'rows': 25, 'cols': 100})),

        'contrib_template': forms.fields.CharField(

            widget=helper.widgets.FullTinyMCE(

                attrs={'rows': 25, 'cols': 100})),

        'clean_description': cleaning.clean_html_content('description'),

        'clean_contrib_template': cleaning.clean_html_content(

            'contrib_template'),

        'clean_ideas': cleaning.clean_url('ideas'),

        'clean': cleaning.validate_new_group('link_id', 'scope_path',

            soc.logic.models.organization, org_app_logic)

        }

    new_params['edit_extra_dynaproperties'] = {

        'clean': cleaning.clean_refs(new_params, ['home_link_id'])

        }

    params = dicts.merge(params, new_params)

    super(View, self).__init__(params=params)

    # create and store the special form for applicants

    updated_fields = {

        'link_id': forms.CharField(widget=widgets.ReadOnlyInput(),

            required=False),

        'clean_link_id': cleaning.clean_link_id('link_id')

        }

    applicant_create_form = dynaform.extendDynaForm(

        dynaform = self._params['create_form'],

        dynaproperties = updated_fields)

    self._params['applicant_create_form'] = applicant_create_form

  @decorators.merge_params

  @decorators.check_access

  def applyMentor(self, request, access_type,

                  page_name=None, params=None, **kwargs):

    """Shows a list of all organizations and you can choose one to

       apply to become a mentor.

    Args:

      request: the standard Django HTTP request object

      access_type : the name of the access type which should be checked

      page_name: the page name displayed in templates as page and header title

      params: a dict with params for this View

      kwargs: the Key Fields for the specified entity

    """

    list_params = params.copy()

    list_params['list_action'] = (redirects.getRequestRedirectForRole, 'mentor')

    list_params['list_description'] = ugettext('Choose an Organization which '

        'you want to become a Mentor for.')

    filter = {'scope_path': kwargs['scope_path'],

              'status' : 'active'}

    return self.list(request, access_type,

        page_name, params=list_params, filter=filter)

  @decorators.merge_params

  @decorators.check_access

  def listProposals(self, request, access_type,

                    page_name=None, params=None, **kwargs):

    """Lists all proposals for the organization given in kwargs.

    For params see base.View.public().

    """

    from soc.logic.models.ranker_root import logic as ranker_root_logic

    from soc.logic.models.student_proposal import logic as sp_logic

    from soc.models import student_proposal

    from soc.views.helper import list_info as list_info_helper

    from soc.views.models import student_proposal as student_proposal_view

    try:

      org_entity = self._logic.getFromKeyFieldsOr404(kwargs)

    except out_of_band.Error, error:

      return helper.responses.errorResponse(

          error, request, template=params['error_public'])

    context = {}

    context['entity'] = org_entity

    list_params = student_proposal_view.view.getParams().copy()

    list_params['list_template'] = 'soc/student_proposal/list_for_org.html'

    list_params['list_key_order'] = [

         'title', 'abstract', 'content', 'additional_info', 'mentor',

         'possible_mentors', 'score', 'status', 'created_on',

         'last_modified_on']

    ranked_params = list_params.copy()# ranked proposals

    ranked_params['list_row'] = ('soc/%(module_name)s/list/'

        'detailed_row.html' % list_params)

    ranked_params['list_heading'] = ('soc/%(module_name)s/list/'

        'detailed_heading.html' % list_params)

    ranked_params['list_description'] = '%s already under review sent to %s' % (

        ranked_params['name_plural'], org_entity.name)

    ranked_params['list_action'] = (redirects.getReviewRedirect, ranked_params)

    # TODO(ljvderijk) once sorting with IN operator is fixed, 

    # make this list show more

    filter = {'org': org_entity,

              'status': 'pending'}

    # order by descending score

    order = ['-score']

    prop_list = lists.getListContent(

        request, ranked_params, filter, order=order, idx=0)

    proposals = prop_list['data']

    # get a list of scores

    scores = [[proposal.score] for proposal in proposals]

    # retrieve the ranker

    fields = {'link_id': student_proposal.DEF_RANKER_NAME,

              'scope': org_entity}

    ranker_root = ranker_root_logic.getForFields(fields, unique=True)

    ranker = ranker_root_logic.getRootFromEntity(ranker_root)

    # retrieve the ranks for these scores

    ranks = [rank+1 for rank in ranker.FindRanks(scores)]

    # link the proposals to the rank

    ranking = dict([i for i in itertools.izip(proposals, ranks)])

    assigned_proposals = []

    # only when the program allows allocations 

    # to be seen we should color the list

    if org_entity.scope.allocations_visible:

      assigned_proposals = sp_logic.getProposalsToBeAcceptedForOrg(org_entity)

      # show the amount of slots assigned on the webpage

      context['slots_visible'] = True

    ranking_keys = dict([(k.key(), v) for k, v in ranking.iteritems()])

    proposal_keys = [i.key() for i in assigned_proposals]

    # update the prop_list with the ranking and coloring information

    prop_list['info'] = (list_info_helper.getStudentProposalInfo(ranking_keys,

        proposal_keys), None)

    # check if the current user is a mentor

    user_entity = user_logic.logic.getForCurrentAccount()

    fields = {'user': user_entity,

        'scope': org_entity,}

    mentor_entity = mentor_logic.logic.getForFields(fields, unique=True)

    if mentor_entity:

      mp_params = list_params.copy() # proposals mentored by current user

      description = ugettext('List of %s sent to %s you are mentoring') % (

          mp_params['name_plural'], org_entity.name)

      mp_params['list_description'] = description

      mp_params['list_action'] = (redirects.getReviewRedirect, mp_params)

      filter = {'org': org_entity,

                'mentor': mentor_entity,

                'status': 'pending'}

      mp_list = lists.getListContent(

          request, mp_params, filter, idx=1, need_content=True)

    new_params = list_params.copy() # new proposals

    new_params['list_description'] = 'List of new %s sent to %s ' % (

        new_params['name_plural'], org_entity.name)

    new_params['list_action'] = (redirects.getReviewRedirect, new_params)

    filter = {'org': org_entity,

              'status': 'new'}

    contents = []

    new_list = lists.getListContent(

        request, new_params, filter, idx=2, need_content=True)

    ap_params = list_params.copy() # accepted proposals

    description = ugettext('List of accepted %s sent to %s ') % (

        ap_params['name_plural'], org_entity.name)

    ap_params['list_description'] = description

    ap_params['list_action'] = (redirects.getReviewRedirect, ap_params)

    filter = {'org': org_entity,

              'status': 'accepted'}

    ap_list = lists.getListContent(

        request, ap_params, filter, idx=3, need_content=True)

    rp_params = list_params.copy() # rejected proposals

    description = ugettext('List of rejected %s sent to %s ') % (

        rp_params['name_plural'], org_entity.name)

    rp_params['list_description'] = description

    rp_params['list_action'] = (redirects.getReviewRedirect, rp_params)

    filter = {'org': org_entity,

              'status': 'rejected'}

    rp_list = lists.getListContent(

        request, rp_params, filter, idx=4, need_content=True)

    ip_params = list_params.copy() # ineligible proposals

    description = ugettext('List of ineligible %s sent to %s ') % (

        ip_params['name_plural'], org_entity.name)

    ip_params['list_description'] = description

    ip_params['list_action'] = (redirects.getReviewRedirect, ip_params)

    filter = {'org': org_entity,

              'status': 'invalid'}

    ip_list = lists.getListContent(

        request, ip_params, filter, idx=5, need_content=True)

    # fill contents with all the needed lists

    if new_list != None:

      contents.append(new_list)

    contents.append(prop_list)

    if mentor_entity and mp_list != None:

      contents.append(mp_list)

    if ap_list != None:

      contents.append(ap_list)

    if rp_list != None:

      contents.append(rp_list)

    if ip_list != None:

      contents.append(ip_list)

    # call the _list method from base to display the list

    return self._list(request, list_params, contents, page_name, context)

  @decorators.merge_params

  @decorators.check_access

  def listPublic(self, request, access_type, page_name=None,

                 params=None, filter=None, **kwargs):

    """See base.View.list.

    """

    account = accounts.getCurrentAccount()

    user = user_logic.logic.getForAccount(account) if account else None

    try:

      rights = self._params['rights']

      rights.setCurrentUser(account, user)

      rights.checkIsHost()

      is_host = True

    except out_of_band.Error:

      is_host = False

    params = params.copy()

    if is_host:

      params['list_action'] = (redirects.getAdminRedirect, params)

    else:

      params['list_action'] = (redirects.getPublicRedirect, params)

    new_filter = {}

    new_filter['scope_path'] = kwargs['scope_path']

    new_filter['status'] = 'active'

    filter = dicts.merge(filter, new_filter)

    content = lists.getListContent(request, params, filter)

    contents = [content]

    return self._list(request, params, contents, page_name)

  def _getMapData(self, filter=None):

    """Constructs the JSON object required to generate 

       Google Maps on organization home page.

    Args:

      filter: a dict for the properties that the entities should have

    Returns: 

      A JSON object containing map data.

    """

    from soc.logic.models.student_project import logic as student_project_logic

    from soc.views.models import student_project as student_project_view

    sp_params = student_project_view.view.getParams().copy()

    people = {}

    projects = {}

    # get all the student_project entities for the given filter

    student_project_entities = student_project_logic.getForFields(

        filter=filter)

    # Construct a dictionary of mentors and students. For each mentor construct

    # a list of 3-tuples containing student name, project title and url.

    # And for each student a list of 3 tuples containing mentor name, project

    # title and url. Only students and mentors who have agreed to publish their

    # locations will be in the dictionary.

    for entity in student_project_entities:

      project_key_name = entity.key().id_or_name()

      project_redirect = redirects.getPublicRedirect(entity, sp_params)

      student_entity = entity.student

      student_key_name = student_entity.key().id_or_name()

      mentor_entity = entity.mentor

      mentor_key_name = mentor_entity.key().id_or_name()

      # store the project data in the projects dictionary

      projects[project_key_name] = {'title': entity.title,

                                    'redirect': project_redirect,

                                    'student_key': student_key_name,

                                    'student_name': student_entity.name(),

                                    'mentor_key': mentor_key_name,

                                    'mentor_name': mentor_entity.name()}

      if mentor_entity.publish_location:

        if mentor_key_name not in people:

          # we have not stored the information of this mentor yet

          people[mentor_key_name] = {

              'type': 'mentor',

              'name': mentor_entity.name(),

              'lat': mentor_entity.latitude,

              'projects': []

              }

        # add this project to the mentor's list

        people[mentor_key_name]['projects'].append(project_key_name)

      if student_entity.publish_location:

        if student_key_name not in people: