def copy_fields_to_builder(
new: entities.StateCharge.Builder,
proto: StateCharge,
metadata: IngestMetadata) -> None:
"""Mutates the provided |charge_builder| by converting an ingest_info proto
StateCharge.
Note: This will not copy children into the Builder!
"""
enum_fields = {
'status': ChargeStatus,
'classification_type': StateChargeClassificationType,
}
enum_mappings = EnumMappings(proto, enum_fields, metadata.enum_overrides)
# Enum values
new.status = enum_mappings.get(ChargeStatus,
default=ChargeStatus.PRESENT_WITHOUT_INFO)
new.status_raw_text = fn(normalize, 'status', proto)
new.classification_type = \
enum_mappings.get(StateChargeClassificationType)
new.classification_type_raw_text = \
fn(normalize, 'classification_type', proto)
# 1-to-1 mappings
new.external_id = fn(parse_external_id, 'state_charge_id', proto)
new.offense_date = fn(parse_date, 'offense_date', proto)
new.date_charged = fn(parse_date, 'date_charged', proto)
new.state_code = parse_region_code_with_override(
proto, 'state_code', metadata)
new.county_code = fn(normalize, 'county_code', proto)
new.statute = fn(normalize, 'statute', proto)
new.ncic_code = fn(normalize, 'ncic_code', proto)
new.description = fn(normalize, 'description', proto)
if new.description is None and new.ncic_code is not None:
ncic_description = ncic.get_description(new.ncic_code)
if ncic_description:
new.description = normalize(ncic_description)
new.attempted = fn(parse_bool, 'attempted', proto)
if new.classification_type is None:
new.classification_type = \
StateChargeClassificationType.find_in_string(new.description)
new.classification_subtype = \
fn(normalize, 'classification_subtype', proto)
new.counts = fn(parse_int, 'counts', proto)
new.charge_notes = fn(normalize, 'charge_notes', proto)
new.is_controlling = fn(parse_bool, 'is_controlling', proto)
new.charging_entity = fn(normalize, 'charging_entity', proto)
def test_get_description_not_found(self):
self.assertIsNone(ncic.get_description("9999"))
def test_get_description(self):
description = ncic.get_description("1313")
self.assertEqual(description, "SIMPLE ASSAULT")