PersistentSelect2MultipleSelect

Complexity

Total Complexity

1

Size/Duplication

Total Lines	29
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

1 Method

Rating	Name	Duplication	Size	Complexity
B	__init__()	0	28	1

# -*- coding: utf-8 -*-
import tw2.core as twc
import tw2.jqplugins.select2 as twsel
from tw2.core.validation import Validator


select2_persistence_js = twc.JSSource(src='''
    // This function returns the value of the specified GET parameter.
    // The code has been found at the following url:
    // http://stackoverflow.com/questions/5448545/how-to-retrieve-get-parameters-from-javascript
    // Big Kudos to you mister Jonah.
    var params = function() {
        function urldecode(str) {
            return decodeURIComponent((str+'').replace(/\+/g, '%20'));
        }

        function transformToAssocArray( prmstr ) {
            var params = {};
            var prmarr = prmstr.split("&");
            for ( var i = 0; i < prmarr.length; i++) {
                var tmparr = prmarr[i].split("=");
                params[tmparr[0]] = urldecode(tmparr[1]);
            }
            return params;
        }

        var prmstr = window.location.search.substr(1);
        return prmstr != null && prmstr != "" ? transformToAssocArray(prmstr) : {};
    }();
''')


class PersistentSelect2MultipleSelect(twsel.Select2MultipleSelectField):
    def __init__(self, **kwargs):
        super(PersistentSelect2MultipleSelect, self).__init__(**kwargs)
        self.resources = [select2_persistence_js]
        self.ondemand = True
        self.validator = self.validator or Validator()

        self.options = []

        self.opts = self.opts.copy()
        self.opts['initSelection'] = twc.js_callback(
            '''
            function(element, callback) {
                var init_data = [];

                param = params['%(param)s'];

                if(typeof param !== "undefined" && param) {
                    $.each(param.split(','), function(i, v) {
                        var elem = {};
                        elem.id = v;
                        elem.text = v;
                        init_data.push(elem);
                    });
                }

                callback(init_data);
            }
            ''' % dict(param=self.name)
        )


class PersistentSelect2SingleSelect(twsel.Select2SingleSelectField):
    def __init__(self, **kwargs):
        super(PersistentSelect2SingleSelect, self).__init__(**kwargs)
        self.resources = [select2_persistence_js]

        self.ondemand = True
        self.validator = self.validator or Validator()

        self.js_dict_entries = ['%s: "%s"' % (key, value)
                                for key, value in self.options]
        self.js_opts_dict = '{%s}' % ', '.join(self.js_dict_entries)

        self.opts = self.opts.copy()
        self.opts['initSelection'] = twc.js_callback(
            '''
            function (element, callback) {
                var init_data;
                var list_opts = %(options)s;
                var param = params['%(param)s'];

                if(typeof param !== "undefined" && param) {
                    var elem = {};
                    elem.id = param;

                    if(param in list_opts) {
                        elem.text = list_opts[param];
                    }

                    init_data = elem;
                }

                callback(init_data);
            }
            ''' % dict(options=self.js_opts_dict, param=self.name)
        )


class GeocompleteField(twsel.Select2AjaxSingleSelectField):
    def __init__(self, **kwargs):
        super(GeocompleteField, self).__init__(**kwargs)
        self.resources = [
            select2_persistence_js,
            twc.JSSource(
                src='''
                function format_name(name, complement, postal_code, country) {
                    var display_name = name;
                    display_name += ' ';

                    if(complement) {
                        display_name += complement;
                    }

                    display_name += ' - ';
                    display_name += postal_code;
                    display_name += ', ';
                    display_name += country;

                    return display_name.toUpperCase();
                }
                '''
            )
        ]
        self.options = []
        self.ondemand = True
        self.validator = self.validator or Validator()

        self.opts.copy()
        self.opts['minimumInputLength'] = 1
        self.opts['maximumInputLength'] = 125
        self.opts['allowClear'] = True
        self.opts['dropdownAutoWidth'] = True

        self.opts['initSelection'] = twc.js_callback(
            """
            function (element, callback) {
                var init_data;

                param = params['%(name)s'];

                if(typeof param !== "undefined" && param) {
                    var elem = {};
                    param_dict = JSON.parse(param);
                    elem.id = param;

                    name = format_name(
                        param_dict['name'],
                        param_dict['complement'],
                        param_dict['postal_code'],
                        param_dict['country']
                    );

                    elem.name = name;
                    elem.value = name;

                    init_data = elem;
                }

                callback(init_data);
            }
            """ % dict(name=self.name)
        )
        self.opts['escapeMarkup'] = twc.js_callback(
            """
            function(markup) {
                return markup;
            }
            """
        )
        self.opts['formatResult'] = twc.js_callback(
            """
            function(location) {
                var markup = '<option value="' + location.value + '">'
                    + location.name
                    + '</option>';
                return markup;
            }
            """
        )
        self.opts['formatSelection'] = twc.js_callback(
            """
            function(location) {
                if(typeof location !== "undefined") {
                    return location.value || location.text;
                }
            }
            """
        )

        self.opts['ajax'] = dict(
            url='/geocomplete',
            dataType='json',
            type='POST',
            quietMillis=100,
            cache=True,
            data=twc.js_callback(
                """
                function(term) {
                    return {address: term};
                }
                """
            ),
            results=twc.js_callback(
                """
                function(data) {
                    var results = [];

                    if ('results' in data) {
                        $.each(data['results'], function(i, v) {
                            var o = {};
                            o.id = JSON.stringify(v);
                            name = format_name(
                                v['name'],
                                v['complement'],
                                v['postal_code'],
                                v['country']
                            );
                            o.name = name;
                            o.value = name;
                            results.push(o);
                        });
                    }

                    return {results: results};
                }
                """
            )
        )

    def prepare(self):
        # Temporary workaround a bug in Toscawidget2 Select2 widget prepare,
        # which would make the persistence persistence bug (due to bugous values
        # introduced in $().select2("val", ...)).
        value = None
        try:
            value = self.value
            delattr(self, 'value')
        except AttributeError:
            pass

        super(GeocompleteField, self).prepare()

        if value:
            setattr(self, 'value', value)