def default_merge_flat_fields(
*, new_entity: DatabaseEntity, old_entity: DatabaseEntity) -> DatabaseEntity:
"""Merges all set non-relationship fields on the |new_entity| onto the |old_entity|. Returns the newly merged
entity."""
for child_field_name in get_set_entity_field_names(new_entity, EntityFieldType.FLAT_FIELD):
if child_field_name == old_entity.get_class_id_name():
continue
# Do not overwrite with default status
if child_field_name == 'status' and new_entity.has_default_status():
continue
old_entity.set_field(child_field_name, new_entity.get_field(child_field_name))
return old_entity
def _base_entity_match(
a: DatabaseEntity,
b: DatabaseEntity,
skip_fields: Set[str],
allow_null_mismatch: bool = False
) -> bool:
"""Returns whether two objects of the same type are an entity match.
Args:
a: The first entity to match
b: The second entity to match
skip_fields: A list of names of fields that should be ignored when determining if two objects match based on
flat fields.
allow_null_mismatch: Allow for two objects to still match if one has a null value in a field where the other's
is nonnull.
"""
# Placeholders never match
if is_placeholder(a) or is_placeholder(b):
return False
# Compare external ids if one is present
if a.get_external_id() or b.get_external_id():
return a.get_external_id() == b.get_external_id()
# Compare all flat fields of the two entities
all_set_flat_field_names = \
get_set_entity_field_names(a, EntityFieldType.FLAT_FIELD) | \
get_set_entity_field_names(b, EntityFieldType.FLAT_FIELD)
for field_name in all_set_flat_field_names:
# Skip primary key
if field_name == a.get_class_id_name() or field_name in skip_fields:
continue
a_field = a.get_field(field_name)
b_field = b.get_field(field_name)
if allow_null_mismatch and (a_field is None or b_field is None):
# Do not disqualify a match if one of the fields is null
continue
if a_field != b_field:
return False
return True
def convert_to_placeholder(entity: DatabaseEntity):
for field_name in get_set_entity_field_names(entity, EntityFieldType.FLAT_FIELD):
if field_name == entity.get_class_id_name():
continue
if field_name == 'state_code':
continue
if field_name == 'status':
entity.set_field(field_name, enum_canonical_strings.present_without_info)
continue
if field_name == 'incarceration_type':
entity.set_field(field_name, StateIncarcerationType.STATE_PRISON.value)
continue
if field_name == 'court_type':
entity.set_field(field_name, StateCourtType.PRESENT_WITHOUT_INFO.value)
continue
if field_name == 'agent_type':
entity.set_field(field_name, StateAgentType.PRESENT_WITHOUT_INFO.value)
continue
entity.clear_field(field_name)
def _base_entity_match(a: DatabaseEntity, b: DatabaseEntity) -> bool:
# Placeholders never match
if is_placeholder(a) or is_placeholder(b):
return False
# Compare external ids if one is present
if a.get_external_id() or b.get_external_id():
return a.get_external_id() == b.get_external_id()
# Compare all flat fields of the two entities
all_set_flat_field_names = \
get_set_entity_field_names(a, EntityFieldType.FLAT_FIELD) | \
get_set_entity_field_names(b, EntityFieldType.FLAT_FIELD)
for field_name in all_set_flat_field_names:
# Skip primary key
if field_name == a.get_class_id_name():
continue
a_field = a.get_field(field_name)
b_field = b.get_field(field_name)
if a_field != b_field:
return False
return True