149 lines
4.8 KiB
Python
149 lines
4.8 KiB
Python
from ietf.idtracker.models import Area, IETFWG
|
|
|
|
IETFCHAIR = {'name': u'The IETF Chair', 'address': u'chair@ietf.org'}
|
|
IESG = {'name': u'The IESG', 'address': u'iesg@ietf.org'}
|
|
IAB = {'name': u'The IAB', 'address': u'iab@iab.org'}
|
|
IABCHAIR = {'name': u'The IAB Chair', 'address': u'iab-chair@iab.org'}
|
|
IABEXECUTIVEDIRECTOR = {'name': u'The IAB Executive Director', 'address': u'execd@iab.org'}
|
|
|
|
|
|
class FakePerson(object):
|
|
|
|
def __init__(self, name, address):
|
|
self.name = name
|
|
self.address = address
|
|
|
|
def email(self):
|
|
return (self.name, self.address)
|
|
|
|
|
|
class IETFEntity(object):
|
|
|
|
def __init__(self, name, poc=None, obj=None, cc=None):
|
|
self.name = name
|
|
self.poc = poc
|
|
self.cc = cc
|
|
self.obj = obj
|
|
|
|
def get_poc(self):
|
|
if not isinstance(self.poc, list):
|
|
return [self.poc]
|
|
return self.poc
|
|
|
|
def get_cc(self):
|
|
if not isinstance(self.cc, list):
|
|
return [self.cc]
|
|
return self.cc
|
|
|
|
|
|
class AreaEntity(IETFEntity):
|
|
|
|
def get_poc(self):
|
|
return [i.person for i in self.obj.areadirector_set.all()]
|
|
|
|
def get_cc(self):
|
|
return [FakePerson(**IETFCHAIR)]
|
|
|
|
|
|
class WGEntity(IETFEntity):
|
|
|
|
def get_poc(self):
|
|
return [i.person for i in self.obj.wgchair_set.all()]
|
|
|
|
def get_cc(self):
|
|
result = [i.person for i in self.obj.area_directors()]
|
|
if self.obj.email_address:
|
|
result.append(FakePerson(name ='%s Discussion List' % self.obj.group_acronym.name,
|
|
address = self.obj.email_address))
|
|
return result
|
|
|
|
|
|
class IETFEntityManager(object):
|
|
|
|
def __init__(self, pk=None, name=None, queryset=None, poc=None, cc=None):
|
|
self.pk = pk
|
|
self.name = name
|
|
self.queryset = queryset
|
|
self.poc = poc
|
|
self.cc = cc
|
|
|
|
def get_entity(self, pk=None):
|
|
return IETFEntity(name=self.name, poc=self.poc, cc=self.cc)
|
|
|
|
def get_managed_list(self):
|
|
return [(self.pk, self.name)]
|
|
|
|
|
|
class AreaEntityManager(IETFEntityManager):
|
|
|
|
def __init__(self, pk=None, name=None, queryset=None, poc=None):
|
|
super(AreaEntityManager, self).__init__(pk, name, queryset, poc)
|
|
if self.queryset == None:
|
|
self.queryset = Area.active_areas()
|
|
|
|
def get_managed_list(self):
|
|
return [(u'%s_%s' % (self.pk, i.pk), i.area_acronym.name) for i in self.queryset]
|
|
|
|
def get_entity(self, pk=None):
|
|
if not pk:
|
|
return None
|
|
try:
|
|
obj = self.queryset.get(pk=pk)
|
|
except self.queryset.model.DoesNotExist:
|
|
return None
|
|
return AreaEntity(name=obj.area_acronym.name, obj=obj)
|
|
|
|
|
|
class WGEntityManager(IETFEntityManager):
|
|
|
|
def __init__(self, pk=None, name=None, queryset=None, poc=None):
|
|
super(WGEntityManager, self).__init__(pk, name, queryset, poc)
|
|
if self.queryset == None:
|
|
self.queryset = IETFWG.objects.filter(group_type=1, status=IETFWG.ACTIVE, areagroup__area__status=Area.ACTIVE)
|
|
|
|
def get_managed_list(self):
|
|
return [(u'%s_%s' % (self.pk, i.pk), i.group_acronym.name) for i in self.queryset]
|
|
|
|
def get_entity(self, pk=None):
|
|
if not pk:
|
|
return None
|
|
try:
|
|
obj = self.queryset.get(pk=pk)
|
|
except self.queryset.model.DoesNotExist:
|
|
return None
|
|
return WGEntity(name=obj.group_acronym.name, obj=obj)
|
|
|
|
|
|
class IETFHierarchyManager(object):
|
|
|
|
def __init__(self):
|
|
self.managers = {'ietf': IETFEntityManager(pk='ietf', name=u'The IETF',
|
|
poc=FakePerson(**IETFCHAIR),
|
|
cc=FakePerson(**IESG)),
|
|
'iesg': IETFEntityManager(pk='iesg', name=u'The IESG',
|
|
poc=FakePerson(**IETFCHAIR),
|
|
cc=FakePerson(**IESG)),
|
|
'iab': IETFEntityManager(pk='iab', name=u'The IAB',
|
|
poc=[FakePerson(**IABCHAIR),
|
|
FakePerson(**IABEXECUTIVEDIRECTOR)],
|
|
cc=FakePerson(**IAB)),
|
|
'area': AreaEntityManager(pk='area'),
|
|
'wg': WGEntityManager(pk='wg'),
|
|
}
|
|
|
|
def get_entity_by_key(self, entity_id):
|
|
id_list = entity_id.split('_', 1)
|
|
key = id_list[0]
|
|
pk = None
|
|
if len(id_list)==2:
|
|
pk = id_list[1]
|
|
if key not in self.managers.keys():
|
|
return None
|
|
return self.managers[key].get_entity(pk)
|
|
|
|
def get_all_entities(self):
|
|
entities = []
|
|
for manager in self.managers.values():
|
|
entities += manager.get_managed_list()
|
|
return entities
|