Autodetect suitable client in Doctor's edit view (#108)

* Autodetect the client a Doctor can be assigned to

* Update Changelog

Author: xispa

CHANGES.rst

@@ -6,8 +6,6 @@ Changelog
 
 **Added**
 
-- #102 Add Primary Referrer column in Doctor listings
-- #85 Allow client contact to list/add/edit Batches from its own Client
 - #102 Add Primary Referrer column in Doctor listings
 - #85 Allow client contact to list/add/edit Cases from its own Client
 - #83 Allow client contact to create and edit Doctors
@@ -21,6 +19,7 @@ Changelog
 
 **Fixed**
 
+- #108 Autodetect suitable client in Doctor's edit view
 - #109 Traceback `bika_widgets/remarks does not exist for`
 - #107 DateTime validator fires even if DateOfBirth is non-mandatory
 - #106 Fields from Patient's additional tabs are not displayed

bika/health/content/doctor.py

@@ -94,7 +94,24 @@ def get_clients_vocabulary(self):
             # Allow to change the client if there are no ARs associated
             client = self.getPrimaryReferrer()
             if not client:
-                return DisplayList([('', '')])
+                # Maybe all Batches and ARs assigned to this Doctor belong to
+                # the same Client.. If so, just assign this client by default
+                client_uids = map(lambda ar: ar.getClientUID,
+                                  self.getAnalysisRequests())
+                client_uids = list(set(client_uids))
+                if len(client_uids) > 1:
+                    # More than one client assigned!
+                    return DisplayList([('', '')])
+                clients = map(lambda batch: batch.getClient(),
+                              self.getBatches(full_objects=True))
+                client_uids += map(lambda client: api.get_uid(client), clients)
+                client_uids = list(set(client_uids))
+                if len(client_uids) > 1:
+                    # More than one client assigned!
+                    return DisplayList([('', '')])
+
+                client = api.get_object_by_uid(client_uids[0])
+
             return DisplayList([(api.get_uid(client), client.Title())])
 
         # If the current user is a client contact, do not display other clients