def _convert_state_person(self, ingest_state_person: StatePerson) \ -> entities.StatePerson: """Converts an ingest_info proto StatePerson to a persistence entity.""" state_person_builder = entities.StatePerson.builder() state_person.copy_fields_to_builder( state_person_builder, ingest_state_person, self.metadata) converted_aliases = [state_alias.convert(self.aliases[alias_id], self.metadata) for alias_id in ingest_state_person.state_alias_ids] state_person_builder.aliases = converted_aliases converted_races = [ state_person_race.convert(self.person_races[race_id], self.metadata) for race_id in ingest_state_person.state_person_race_ids ] state_person_builder.races = converted_races converted_ethnicities = [ state_person_ethnicity.convert( self.person_ethnicities[ethnicity_id], self.metadata) for ethnicity_id in ingest_state_person.state_person_ethnicity_ids ] state_person_builder.ethnicities = converted_ethnicities converted_assessments = [ self._convert_assessment(self.assessments[assessment_id]) for assessment_id in ingest_state_person.state_assessment_ids ] state_person_builder.assessments = converted_assessments converted_external_ids = [ state_person_external_id.convert( self.person_external_ids[external_id], self.metadata) for external_id in ingest_state_person.state_person_external_ids_ids ] state_person_builder.external_ids = converted_external_ids converted_sentence_groups = [ self._convert_sentence_group( self.sentence_groups[sentence_group_id]) for sentence_group_id in ingest_state_person.state_sentence_group_ids ] state_person_builder.sentence_groups = converted_sentence_groups return state_person_builder.build()
def testParseStatePersonEthnicity(self) -> None: # Arrange ingest_person_ethnicity = ingest_info_pb2.StatePersonEthnicity( ethnicity="HISPANIC", state_code="US_ND", state_person_ethnicity_id="123", ) # Act result = state_person_ethnicity.convert(ingest_person_ethnicity, _EMPTY_METADATA) # Assert expected_result = entities.StatePersonEthnicity( ethnicity=Ethnicity.HISPANIC, ethnicity_raw_text="HISPANIC", state_code="US_ND", ) self.assertEqual(result, expected_result)
def _convert_person(self, ingest_person: StatePerson) -> entities.StatePerson: """Converts an ingest_info proto StatePerson to a persistence entity.""" state_person_builder = entities.StatePerson.builder() state_person.copy_fields_to_builder(state_person_builder, ingest_person, self.metadata) converted_aliases = [ state_alias.convert(self.aliases[alias_id], self.metadata) for alias_id in ingest_person.state_alias_ids ] state_person_builder.aliases = converted_aliases converted_races = [ state_person_race.convert(self.person_races[race_id], self.metadata) for race_id in ingest_person.state_person_race_ids ] state_person_builder.races = converted_races converted_ethnicities = [ state_person_ethnicity.convert( self.person_ethnicities[ethnicity_id], self.metadata) for ethnicity_id in ingest_person.state_person_ethnicity_ids ] state_person_builder.ethnicities = converted_ethnicities converted_assessments = [ self._convert_assessment(self.assessments[assessment_id]) for assessment_id in ingest_person.state_assessment_ids ] state_person_builder.assessments = converted_assessments converted_program_assignments = [ self._convert_program_assignment( self.program_assignments[assignment_id]) for assignment_id in ingest_person.state_program_assignment_ids ] state_person_builder.program_assignments = converted_program_assignments converted_external_ids = [ state_person_external_id.convert( self.person_external_ids[external_id], self.metadata) for external_id in ingest_person.state_person_external_ids_ids ] state_person_builder.external_ids = converted_external_ids converted_sentence_groups = [ self._convert_sentence_group( self.sentence_groups[sentence_group_id]) for sentence_group_id in ingest_person.state_sentence_group_ids ] state_person_builder.sentence_groups = converted_sentence_groups if ingest_person.supervising_officer_id: converted_supervising_officer = state_agent.convert( self.agents[ingest_person.supervising_officer_id], self.metadata) state_person_builder.supervising_officer = converted_supervising_officer return state_person_builder.build(StatePersonFactory.deserialize)