57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
from copy import deepcopy
|
|
from z3c.form import button
|
|
from zope.i18nmessageid.message import MessageFactory
|
|
|
|
from plone import api
|
|
from plone.dexterity.browser.add import DefaultAddForm
|
|
from plone.stringinterp.adapters import _recursiveGetMembersFromIds
|
|
|
|
from collective.task import _
|
|
|
|
|
|
class AddInformation(DefaultAddForm):
|
|
"""Custom add information view"""
|
|
|
|
portal_type = "information"
|
|
|
|
@button.buttonAndHandler(_('Add'), name='add')
|
|
def handleAdd(self, action):
|
|
portal = api.portal.get()
|
|
data, errors = self.extractData()
|
|
if errors:
|
|
self.status = self.formErrorsMessage
|
|
return
|
|
|
|
objs = []
|
|
seen = {}
|
|
for responsible in data['responsible']:
|
|
group = api.group.get(responsible)
|
|
if group is not None:
|
|
# responsible is a group, create an Information by user in this group
|
|
groupname = group.getId()
|
|
users = _recursiveGetMembersFromIds(portal, [groupname])
|
|
for user in users:
|
|
username = user.getId()
|
|
if username in seen:
|
|
continue
|
|
_data = deepcopy(data)
|
|
_data['responsible'] = [username]
|
|
obj = self.createAndAdd(_data)
|
|
if obj is not None:
|
|
objs.append(obj)
|
|
seen[username] = True
|
|
else:
|
|
# responsible is a user
|
|
if responsible in seen:
|
|
continue
|
|
_data = deepcopy(data)
|
|
_data['responsible'] = [responsible]
|
|
obj = self.createAndAdd(_data)
|
|
if obj is not None:
|
|
objs.append(obj)
|
|
seen[responsible] = True
|
|
|
|
if objs:
|
|
# mark only as finished if we get the new object
|
|
self._finishedAdd = True
|