project/monkey.py

   1 # -*- encoding: utf-8 -*-
   2
   3 ################################################################################
   4 #
   5 # PATCH AJAX SELECT FIELD : pour gérer le fait que certaine FK ne sont pas
   6 # forcément mappé sur la PK du modèle, dans notre cas ici c'est le modèle
   7 # Pays qui dispose d'un pk sur son id mais les fk sont mappé sur le champs 'code'
   8 #
   9 ################################################################################
  10 from ajax_select import get_lookup
  11 from django.contrib.admin import site
  12 from django.db import models
  13 from django.http import HttpResponse
  14 import ajax_select.views
  15
  16 def fk_ajax_lookup(request,channel):
  17     """ this view supplies results for both foreign keys and many to many fields """
  18
  19     # it should come in as GET unless global $.ajaxSetup({type:"POST"}) has been set
  20     # in which case we'll support POST
  21     if request.method == "GET":
  22         # we could also insist on an ajax request
  23         if 'q' not in request.GET:
  24             return HttpResponse('')
  25         query = request.GET['q']
  26     else:
  27         if 'q' not in request.POST:
  28             return HttpResponse('') # suspicious
  29         query = request.POST['q']
  30
  31     lookup_channel = get_lookup(channel)
  32
  33     if query:
  34         instances = lookup_channel.get_query(query,request)
  35     else:
  36         instances = []
  37
  38     results = []
  39     for item in instances:
  40         itemf = lookup_channel.format_item(item)
  41         itemf = itemf.replace("\n","").replace("|","&brvbar;")
  42         resultf = lookup_channel.format_result(item)
  43         resultf = resultf.replace("\n","").replace("|","&brvbar;")
  44         fk = getattr(lookup_channel, 'fk_key', None)
  45         if fk is not None:
  46             id = getattr(item, fk)
  47         else:
  48             id = item.pk
  49         results.append( "|".join((unicode(id),itemf,resultf)) )
  50     return HttpResponse("\n".join(results))
  51
  52 ajax_select.views.ajax_lookup = fk_ajax_lookup
  53
  54 from ajax_select.fields import AutoCompleteSelectWidget
  55
  56 def value_from_datadict(self, data, files, name):
  57
  58     got = data.get(name, None)
  59     if got:
  60         try:
  61             return long(got)
  62         except:
  63             return got
  64     else:
  65         return None
  66
  67 AutoCompleteSelectWidget.value_from_datadict = value_from_datadict
  68
  69 from django.conf import settings
  70 from django.core.urlresolvers import reverse
  71 from django.utils.safestring import mark_safe
  72 from django.forms.util import flatatt
  73 from django.template.loader import render_to_string
  74
  75 def patched_render(self, name, value, attrs=None):
  76     """
  77     prevention contre les choses supprimees
  78     """
  79     # precheck existence valeur
  80     lookup = get_lookup(self.channel)
  81     if value:
  82         objs = lookup.get_objects([value])
  83         if len(objs) == 0:
  84             value = ''
  85
  86     # copie de django_ajax_selects-1.1.4-py2.6.egg/ajax_select/fields.py
  87     value = value or ''
  88     final_attrs = self.build_attrs(attrs)
  89     self.html_id = final_attrs.pop('id', name)
  90
  91     lookup = get_lookup(self.channel)
  92     if value:
  93         objs = lookup.get_objects([value])
  94         try:
  95             obj = objs[0]
  96         except IndexError:
  97             raise Exception("%s cannot find object:%s" % (lookup, value))
  98         current_result = mark_safe(lookup.format_item( obj ) )
  99     else:
 100         current_result = ''
 101
 102     context = {
 103             'name': name,
 104             'html_id' : self.html_id,
 105             'lookup_url': reverse('ajax_lookup',kwargs={'channel':self.channel}),
 106             'current_id': value,
 107             'current_result': current_result,
 108             'help_text': self.help_text,
 109             'extra_attrs': mark_safe(flatatt(final_attrs)),
 110             'func_slug': self.html_id.replace("-",""),
 111             'add_link' : self.add_link,
 112             'admin_media_prefix' : settings.ADMIN_MEDIA_PREFIX
 113             }
 114
 115     return mark_safe(render_to_string(('autocompleteselect_%s.html' % self.channel, 'autocompleteselect.html'),context))
 116
 117
 118
 119 AutoCompleteSelectWidget.render = patched_render