Sampler field

src/bika/aquaculture/extenders/batch.py

@@ -13,6 +13,7 @@
 from bika.aquaculture.config import _
 from bika.aquaculture.config import is_installed
 from bika.aquaculture.vocabularies import BATCH_PRIORITY
+from bika.aquaculture.vocabularies import getUsers
 from bika.aquaculture.vocabularies import get_countries
 from bika.aquaculture.interfaces import IBikaAquacultureLayer
 from bika.extras.extenders.fields import ExtStringField
@@ -147,6 +148,18 @@
     )
 )
 
+sampler_field = ExtStringField(
+    "Sampler",
+    required=False,
+    mode="rw",
+    write_permission=FieldEditContact,
+    vocabulary=getUsers(None, ["Sampler"]),
+    widget=SelectionWidget(
+        label=_("Sampler"),
+        format='select',
+    )
+)
+
 
 @implementer(ISchemaExtender, IBrowserLayerAwareExtender)
 class BatchSchemaExtender(object):
@@ -160,6 +173,7 @@ class BatchSchemaExtender(object):
         country_of_origin_field,
         pooling_info_field,
         destination_country_field,
+        sampler_field,
         payment_method_field,
         batch_priority_field,
     ]
@@ -185,6 +199,7 @@ def fiddle(self, schema):
         """
         """
         if is_installed():
+            schema['Sampler'].vocabulary = getUsers(self.context, ["Sampler"])
             schema['ClientBatchID'].widget.label = "Case Number"
             schema['BatchLabels'].widget.label = "Case Labels"
             schema['title'].widget.description = "If no Title value is entered, the Case ID will be used."

src/bika/aquaculture/vocabularies.py

@@ -1,8 +1,35 @@
 # -*- coding: utf-8 -*-
 
+from Products.Archetypes.public import DisplayList
+from zope.interface import provider
+from zope.schema.interfaces import IVocabularyFactory
+
 from bika.aquaculture.config import _
+from bika.lims import api
 from senaite.core.api import geo
 
+
+@provider(IVocabularyFactory)
+def getUsers(context, roles, allow_empty=True):
+    """ Present a DisplayList containing users in the specified
+        list of roles
+    """
+    pairs = allow_empty and [['', '']] or []
+    if not context:
+        return DisplayList(pairs)
+
+    mtool = api.get_tool('portal_membership')
+    users = mtool.searchForMembers(roles=roles)
+    for user in users:
+        uid = user.getId()
+        fullname = user.getProperty('fullname')
+        if not fullname:
+            fullname = uid
+        pairs.append((uid, fullname))
+    pairs.sort(lambda x, y: cmp(x[1], y[1]))
+    return DisplayList(pairs)
+
+
 BATCH_PRIORITY = [
     (_("routine"), _("Routine")),
     (_("rush"), _("Rush")),