Coverage for id_contexts/sa_id_book.py : 61%

""" 

---------------------------------------------------------------------- 

Authors: Jan-Justin van Tonder 

---------------------------------------------------------------------- 

This file contains the logic for South African ID book (post 1994) 

context. 

---------------------------------------------------------------------- 

""" 

from id_contexts.id_context import FieldType, LineType 

from id_contexts.sa_id import SAID 

from hutts_utils.hutts_logger import logger 

from fuzzywuzzy import fuzz 

class SAIDBook(SAID): 

""" 

A class that represents an ID context for a South African ID book (post 1994). 

It supplies some of the concrete information, such as the match contexts, to the classes higher up in inheritance 

hierarchy and implements abstract methods defined by its parent. 

""" 

def __init__(self): 

""" 

Initialises the SAIDBook object. 

""" 

# Logging for debugging purposes. 

logger.debug('Initialising %s...' % type(self).__name__) 

# Specify initial list of contexts for string image_processing when populating 

# the ID information dictionary to send as output. 

match_contexts = [{ 

'field': 'identity_number', 

'find': 'id no', 

'field_type': FieldType.NUMERIC_ONLY, 

'line_type': LineType.TITLED_ADJACENT, 

'multi_line': False 

}, { 

'field': 'surname', 

'find': 'surname', 

'field_type': FieldType.TEXT_ONLY, 

'to_uppercase': False, 

'line_type': LineType.TITLED_NEWLINE, 

'multi_line': True, 

'multi_line_end': 'forenames' 

}, { 

'field': 'names', 

'find': 'forenames', 

'field_type': FieldType.TEXT_ONLY, 

'to_uppercase': False, 

'line_type': LineType.TITLED_NEWLINE, 

'multi_line': True, 

'multi_line_end': 'country of birth' 

}, { 

'field': 'sex', 

'find': 'sex', 

'field_type': FieldType.TEXT_ONLY, 

'to_uppercase': False, 

'line_type': LineType.TITLED_NEWLINE, 

'multi_line': False 

}, { 

'field': 'date_of_birth', 

'find': 'date of birth', 

'field_type': FieldType.DATE_HYPHENATED, 

'to_uppercase': False, 

'line_type': LineType.TITLED_NEWLINE, 

'multi_line': False 

}, { 

'field': 'country_of_birth', 

'find': 'country of birth', 

'field_type': FieldType.TEXT_ONLY, 

'to_uppercase': False, 

'line_type': LineType.TITLED_NEWLINE, 

'multi_line': False 

}, { 

'field': 'status', 

'find': 'sacitizen', 

'field_type': FieldType.TEXT_ONLY, 

'to_uppercase': False, 

'line_type': LineType.UNTITLED_ADJACENT, 

'multi_line': False 

}] 

# Initialise parent 

SAID.__init__(self, match_contexts) 

def _get_idiosyncratic_match(self, match_context, id_string_list, current_index): 

""" 

Identifies and returns matches that are specific to the current ID context. 

Authors: 

Jan-Justin van Tonder 

Args: 

match_context (dict): A dictionary object that provides context for the information that is to be extracted. 

id_string_list (list): An ID string that has been broken down into a list of individual lines. 

current_index (int): The current index within the ID string list. 

Returns: 

(str): A string containing the match value of a context-specific case. 

(None): Used to indicate that no special case was identified. 

""" 

# Check for the status special case. 

if match_context['field'] == 'status': 

citizen_match_ratio = fuzz.token_set_ratio(id_string_list[current_index], 'sacitizen') 

non_citizen_match_ratio = fuzz.token_set_ratio(id_string_list[current_index], 'nonsacitizen') 

match = 'Citizen' if citizen_match_ratio > non_citizen_match_ratio else 'Non Citizen' 

return match 

# Otherwise return an empty string to indicate that a special case was not identified. 

return None