Use list comprehensions and built-in functions
svn: r14134
This commit is contained in:
		| @@ -160,8 +160,7 @@ class DjangoInterface(object): | |||||||
|         obj_type = ContentType.objects.get_for_model(obj) |         obj_type = ContentType.objects.get_for_model(obj) | ||||||
|         attribute_list = models.Attribute.objects.filter(object_id=obj.id,  |         attribute_list = models.Attribute.objects.filter(object_id=obj.id,  | ||||||
|                                                      object_type=obj_type) |                                                      object_type=obj_type) | ||||||
|         return [self.pack_attribute(attribute) for attribute  |         return map(self.pack_attribute, attribute_list) | ||||||
|                 in attribute_list] |  | ||||||
|  |  | ||||||
|     def get_names(self, person, preferred): |     def get_names(self, person, preferred): | ||||||
|         names = person.name_set.filter(preferred=preferred).order_by("order") |         names = person.name_set.filter(preferred=preferred).order_by("order") | ||||||
| @@ -174,29 +173,19 @@ class DjangoInterface(object): | |||||||
|             return map(self.pack_name, names) |             return map(self.pack_name, names) | ||||||
|       |       | ||||||
|     def get_datamap(self, obj): # obj is source |     def get_datamap(self, obj): # obj is source | ||||||
|         datamap_dict = {} |         return dict([map.key, map.value] for map in obj.datamap_set.all()) | ||||||
|         datamap_list = obj.datamap_set.all() |  | ||||||
|         for datamap in datamap_list: |  | ||||||
|             datamap_dict[datamap.key] = datamap.value |  | ||||||
|         return datamap_dict |  | ||||||
|  |  | ||||||
|     def get_media_list(self, obj): |     def get_media_list(self, obj): | ||||||
|         obj_type = ContentType.objects.get_for_model(obj) |         obj_type = ContentType.objects.get_for_model(obj) | ||||||
|         mediarefs = models.MediaRef.objects.filter(object_id=obj.id,  |         mediarefs = models.MediaRef.objects.filter(object_id=obj.id,  | ||||||
|                                                object_type=obj_type) |                                                object_type=obj_type) | ||||||
|         retval = [] |         return map(self.pack_media_ref, mediarefs) | ||||||
|         for mediaref in mediarefs: |  | ||||||
|             retval.append(self.pack_media_ref(mediaref)) |  | ||||||
|         return retval |  | ||||||
|  |  | ||||||
|     def get_note_list(self, obj): |     def get_note_list(self, obj): | ||||||
|         obj_type = ContentType.objects.get_for_model(obj) |         obj_type = ContentType.objects.get_for_model(obj) | ||||||
|         noterefs = models.NoteRef.objects.filter(object_id=obj.id,  |         noterefs = models.NoteRef.objects.filter(object_id=obj.id,  | ||||||
|                                              object_type=obj_type) |                                              object_type=obj_type) | ||||||
|         retval = [] |         return [noteref.ref_object.handle for noteref in noterefs] | ||||||
|         for noteref in noterefs: |  | ||||||
|             retval.append( noteref.ref_object.handle) |  | ||||||
|         return retval |  | ||||||
|  |  | ||||||
|     def get_repository_ref_list(self, obj): |     def get_repository_ref_list(self, obj): | ||||||
|         obj_type = ContentType.objects.get_for_model(obj) |         obj_type = ContentType.objects.get_for_model(obj) | ||||||
| @@ -209,30 +198,20 @@ class DjangoInterface(object): | |||||||
|  |  | ||||||
|     def get_address_list(self, obj, with_parish): # person or repository |     def get_address_list(self, obj, with_parish): # person or repository | ||||||
|         addresses = obj.address_set.all().order_by("order") |         addresses = obj.address_set.all().order_by("order") | ||||||
|         retval = [] |         return [self.pack_address(address, with_parish) | ||||||
|         count = 1 |                     for address in addresses] | ||||||
|         for address in addresses: |  | ||||||
|             retval.append(self.pack_address(address, with_parish)) |  | ||||||
|             count += 1 |  | ||||||
|         return retval |  | ||||||
|  |  | ||||||
|     def get_child_ref_list(self, family): |     def get_child_ref_list(self, family): | ||||||
|         obj_type = ContentType.objects.get_for_model(family) |         obj_type = ContentType.objects.get_for_model(family) | ||||||
|         childrefs = models.ChildRef.objects.filter(object_id=family.id, \ |         childrefs = models.ChildRef.objects.filter(object_id=family.id, \ | ||||||
|                                                    object_type=obj_type).order_by("order") |                                                    object_type=obj_type).order_by("order") | ||||||
|         retval = [] |         return map(self.pack_child_ref, childrefs) | ||||||
|         for childref in childrefs: |  | ||||||
|             retval.append(self.pack_child_ref(childref)) |  | ||||||
|         return retval |  | ||||||
|  |  | ||||||
|     def get_source_ref_list(self, obj): |     def get_source_ref_list(self, obj): | ||||||
|         obj_type = ContentType.objects.get_for_model(obj) |         obj_type = ContentType.objects.get_for_model(obj) | ||||||
|         sourcerefs = models.SourceRef.objects.filter(object_id=obj.id, \ |         sourcerefs = models.SourceRef.objects.filter(object_id=obj.id, \ | ||||||
|                                   object_type=obj_type).order_by("order") |                                   object_type=obj_type).order_by("order") | ||||||
|         retval = [] |         return map(self.pack_source_ref, sourcerefs) | ||||||
|         for sourceref in sourcerefs: |  | ||||||
|             retval.append(self.pack_source_ref(sourceref)) |  | ||||||
|         return retval |  | ||||||
|  |  | ||||||
|     def get_event_refs(self, obj, order="order"): |     def get_event_refs(self, obj, order="order"): | ||||||
|         obj_type = ContentType.objects.get_for_model(obj) |         obj_type = ContentType.objects.get_for_model(obj) | ||||||
| @@ -244,10 +223,7 @@ class DjangoInterface(object): | |||||||
|         obj_type = ContentType.objects.get_for_model(obj) |         obj_type = ContentType.objects.get_for_model(obj) | ||||||
|         eventrefs = models.EventRef.objects.filter(object_id=obj.id, \ |         eventrefs = models.EventRef.objects.filter(object_id=obj.id, \ | ||||||
|                                   object_type=obj_type).order_by("order") |                                   object_type=obj_type).order_by("order") | ||||||
|         retval = [] |         return map(self.pack_event_ref, eventrefs) | ||||||
|         for eventref in eventrefs: |  | ||||||
|             retval.append(self.pack_event_ref(eventref)) |  | ||||||
|         return retval |  | ||||||
|  |  | ||||||
|     def get_family_list(self, person): # person has families |     def get_family_list(self, person): # person has families | ||||||
|         return [fam.handle for fam in person.families.all()] |         return [fam.handle for fam in person.families.all()] | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user