1
|
|
|
# -*- coding: utf-8 -*- |
2
|
|
|
# |
3
|
|
|
# This file is part of SENAITE.CORE. |
4
|
|
|
# |
5
|
|
|
# SENAITE.CORE is free software: you can redistribute it and/or modify it under |
6
|
|
|
# the terms of the GNU General Public License as published by the Free Software |
7
|
|
|
# Foundation, version 2. |
8
|
|
|
# |
9
|
|
|
# This program is distributed in the hope that it will be useful, but WITHOUT |
10
|
|
|
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS |
11
|
|
|
# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more |
12
|
|
|
# details. |
13
|
|
|
# |
14
|
|
|
# You should have received a copy of the GNU General Public License along with |
15
|
|
|
# this program; if not, write to the Free Software Foundation, Inc., 51 |
16
|
|
|
# Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
17
|
|
|
# |
18
|
|
|
# Copyright 2018-2021 by it's authors. |
19
|
|
|
# Some rights reserved, see README and LICENSE. |
20
|
|
|
|
21
|
|
|
from AccessControl import ClassSecurityInfo |
22
|
|
|
from Products.Archetypes.Registry import registerWidget |
23
|
|
|
from Products.Archetypes.Widget import TypesWidget |
24
|
|
|
from Products.CMFCore.utils import getToolByName |
25
|
|
|
from senaite.core.api import geo |
26
|
|
|
|
27
|
|
|
|
28
|
|
|
class AddressWidget(TypesWidget): |
29
|
|
|
_properties = TypesWidget._properties.copy() |
30
|
|
|
_properties.update({ |
31
|
|
|
'macro': "bika_widgets/addresswidget", |
32
|
|
|
'helper_js': ("bika_widgets/addresswidget.js",), |
33
|
|
|
'helper_css': ("bika_widgets/addresswidget.css",), |
34
|
|
|
'showLegend': True, |
35
|
|
|
'showDistrict': True, |
36
|
|
|
'showCopyFrom': True, |
37
|
|
|
'showCity': True, |
38
|
|
|
'showPostalCode': True, |
39
|
|
|
'showAddress': True, |
40
|
|
|
}) |
41
|
|
|
|
42
|
|
|
security = ClassSecurityInfo() |
43
|
|
|
|
44
|
|
|
# The values in the form/field are always |
45
|
|
|
# Country Name, State Name, District Name. |
46
|
|
|
|
47
|
|
|
def getCountries(self): |
48
|
|
|
countries = geo.get_countries() |
49
|
|
|
items = map(lambda item: (item.alpha_2, item.name), countries) |
50
|
|
|
return self.to_utf8(items) |
51
|
|
|
|
52
|
|
|
def getDefaultCountry(self): |
53
|
|
|
portal = getToolByName(self, 'portal_url').getPortalObject() |
54
|
|
|
bs = portal._getOb('bika_setup') |
55
|
|
|
return bs.getDefaultCountry() |
56
|
|
|
|
57
|
|
|
def getStates(self, country): |
58
|
|
|
items = [] |
59
|
|
|
if not country: |
60
|
|
|
return items |
61
|
|
|
|
62
|
|
|
# first-level subdivisions of the country (states??) |
63
|
|
|
items = geo.get_subdivisions(country, default=[]) |
64
|
|
|
items = map(lambda sub: [sub.country_code, sub.code, sub.name], items) |
65
|
|
|
return self.to_utf8(items) |
66
|
|
|
|
67
|
|
|
def getDistricts(self, country, state): |
68
|
|
|
items = [] |
69
|
|
|
if not country or not state: |
70
|
|
|
return items |
71
|
|
|
|
72
|
|
|
# first-level subdivisions (districts?) of this subdivision (state?) |
73
|
|
|
state_obj = geo.get_subdivision(state, parent=country, default=None) |
74
|
|
|
items = geo.get_subdivisions(state_obj, default=[]) |
75
|
|
|
items = map(lambda sub: [sub.country_code, sub.code, sub.name], items) |
76
|
|
|
return self.to_utf8(items) |
77
|
|
|
|
78
|
|
|
def to_utf8(self, value): |
79
|
|
|
if isinstance(value, unicode): |
|
|
|
|
80
|
|
|
return value.encode("utf-8") |
81
|
|
|
elif isinstance(value, list): |
82
|
|
|
return map(self.to_utf8, value) |
83
|
|
|
return value |
84
|
|
|
|
85
|
|
|
|
86
|
|
|
registerWidget(AddressWidget, |
87
|
|
|
title = 'Address Widget', |
88
|
|
|
description = ('Simple address widget with country/state lookups'), |
89
|
|
|
) |
90
|
|
|
|