Map: basic management of map tab

1 files changed, 6 insertions, 3 deletions

diff --git a/ishtar_common/forms.py b/ishtar_common/forms.py
index 8d0d80497..0bb97b980 100644
--- a/ishtar_common/forms.py
+++ b/ishtar_common/forms.py
@@ -40,8 +40,8 @@ from django.utils.text import slugify
 from django.utils.translation import ugettext_lazy as _
 
 from bootstrap_datepicker.widgets import DatePicker, DATE_FORMAT, DateField
-import models
-import widgets
+from ishtar_common import models
+from ishtar_common import widgets
 from ishtar_common.utils import MultiValueDict
 
 
@@ -661,7 +661,7 @@ def get_form_selection(
         class_name, label, key, model, base_form, get_url,
         not_selected_error=_(u"You should select an item."), new=False,
         new_message=_(u"Add a new item"), get_full_url=None,
-        gallery=False
+        gallery=False, map=False
     ):
     """
     Generate a class selection form
@@ -674,6 +674,7 @@ def get_form_selection(
         new -- can add new items
         new_message -- message of link to add new items
         gallery -- display a gallery
+        map -- display a map
     """
     attrs = {'_main_key': key,
              '_not_selected_error': not_selected_error,
@@ -685,6 +686,8 @@ def get_form_selection(
         widget_kwargs['source_full'] = reverse_lazy(get_full_url)
     if gallery:
         widget_kwargs['gallery'] = True
+    if map and models.get_current_profile().mapping:
+        widget_kwargs['map'] = True
     attrs[key] = forms.IntegerField(
         label="", required=False,
         validators=[models.valid_id(model)],