Skip to content
Snippets Groups Projects

added role column to worker list. Issue #238

Merged Carlos Vega requested to merge feature/238_add_role_column_worker_list into master
1 unresolved thread
3 files
+ 39
1
Compare changes
  • Side-by-side
  • Inline
Files
3
+ 11
1
@@ -7,7 +7,7 @@ from django.db import models
from web.models.constants import GLOBAL_STUDY_ID, COUNTRY_OTHER_ID
from web.models.worker_study_role import STUDY_ROLE_CHOICES, HEALTH_PARTNER_ROLE_CHOICES, \
VOUCHER_PARTNER_ROLE_CHOICES, WORKER_STAFF, WORKER_HEALTH_PARTNER, WORKER_VOUCHER_PARTNER
VOUCHER_PARTNER_ROLE_CHOICES, WORKER_STAFF, WORKER_HEALTH_PARTNER, WORKER_VOUCHER_PARTNER, ROLE_CHOICES
logger = logging.getLogger(__name__)
@@ -161,6 +161,16 @@ class Worker(models.Model):
else:
return False
@property
def role(self):
roles = self.roles.filter(study=GLOBAL_STUDY_ID)
if roles.count() == 0:
return WORKER_STAFF
Please register or sign in to reply
role = roles[0].role
if role not in [role_type for role_type, _ in ROLE_CHOICES]:
raise TypeError("Unknown worker role")
return role
@staticmethod
def get_by_user(the_user):
if isinstance(the_user, User):
Loading