def extract_suffix_name(self, name, suffixes):
"""Separate name and suffix
Parameters
----------
name : str
suffixes : list
list of strings specifying name suffixes to extract
Returns
-------
name_tuple : tuple, length 2
[0] name without suffix : str
[1] suffix : str
"""
suffix = [w for w in name.split(" ") if w in suffixes]
assert len(suffix) < 2, 'Too many suffixes found {}'.format(suffix)
if suffix and (len(suffix) != 1 or suffix != name.split(" ")[0]):
name = re.sub(r'(^{0}\s)|(\s{0}$)|(\s{0}\s)'.format(suffix[0]),
' ', name)
name = string_strip(name, no_sep=False)
suffix = suffix[0].replace(' ', '')
else:
suffix = ''
if not name:
name = suffix
suffix = ''
name_tuple = (string_strip(name, no_sep=False), suffix)
return name_tuple
def extract_middle_initial(self,
name,
mi_pattern=r'(\s[A-Z]\s)|(\s[A-Z]$)',
not_pattern=r'^((DE LA)\s[A-Z]($|\s))',
avoid_suffixes=[]):
"""Separate name and suffix
Parameters
----------
name : str
mi_pattern : str
Regex pattern for identifying middle initial
not_pattern : str
Regex pattern to avoid identifying middle initial
avoid_suffixes : list
list of strings to avoid as middle initials
Returns
-------
name_tuple : tuple, length 2
[0] name without middle initial : str
[1] middle initial : str
"""
mi = re.search(mi_pattern, name)
mi = mi.group() if mi and len(name) > 3 else ''
if (mi and not ((not_pattern and re.search(not_pattern, name))
or mi[-1] in avoid_suffixes)):
name = re.sub(mi_pattern, ' ', name)
mi = mi.replace(' ', '')
else:
mi = ''
return string_strip(name, no_sep=False), mi
def clean(self):
"""Clean name
Returns
----------
name_dict : dict
Dictionary of all name parts created in __init__ after cleaning
"""
if self.middle_name != '' and self.middle_name[0] == '(':
self.second_name = re.sub(r'\(|\)', '', self.middle_name)
elif self.middle_name == '':
pass
else:
self.middle_initial = self.middle_name[0]
self.extract_parts()
self.first_name = string_strip(self.first_name, no_sep=False)
self.first_name_NS = string_strip(self.first_name, no_sep=True)
self.last_name = string_strip(self.last_name, no_sep=False)
self.last_name_NS = string_strip(self.last_name, no_sep=True)
name_dict = self.__dict__
return name_dict
def clean_human_name(name):
cdict = {
'first_name': '',
'last_name': '',
'middle_initial': '',
'middle_initial2': '',
'suffix_name': '',
'middle_name': ''
}
name = string_strip(name)
np = name.split(' ')
def add_mi(m):
nonlocal cdict
if cdict['middle_initial'] == '':
cdict['middle_initial'] = m
elif cdict['middle_initial2'] == '':
cdict['middle_initial2'] = m
else:
print(m)
raise
return None
def clean(x):
nonlocal cdict
if len(x) == 2:
cdict['first_name'] = x[0]
cdict['last_name'] = x[1]
elif len(x) == 3:
if len(x[0]) == 1 and len(x[1]) == 1:
cdict['first_name'] = ' '.join(x[:2])
cdict['last_name'] = x[2]
elif x[2] in sns + ['V', 'I']:
cdict['first_name'] = x[0]
cdict['last_name'] = x[1]
cdict['suffix_name'] = x[2]
elif (x[1] + x[2] in lns or (x[1] in sln and len(x[2]) >= 3)
or '-MC' in x[1]):
cdict['first_name'] = x[0]
cdict['last_name'] = ' '.join(x[1:])
elif len(x[1]) == 1:
cdict['first_name'] = x[0]
cdict['last_name'] = x[2]
add_mi(x[1])
elif (len(x[0]) <= 3 and len(x[1]) >= 3) or x[1] in fem:
cdict['first_name'] = ' '.join(x[:2])
cdict['last_name'] = x[2]
else:
cdict['first_name'] = x[0]
if use_middle_names:
cdict['middle_name'] = x[1]
add_mi(x[1][0])
cdict['last_name'] = x[2]
else:
cdict['last_name'] = ' '.join(x[1:])
elif len(x) > 3:
if x[-1] in sns:
cdict['suffix_name'] = x[-1]
x = x[:-1]
return clean(x)
elif (len(x[1]) == 1 and len(x[2]) == 1
and cdict['middle_initial'] == ''):
add_mi(x[1])
add_mi(x[2])
x = x[:1] + x[3:]
clean(x)
elif len(x[0]) == 1 and len(x[1]) >= 1:
x[1] = ' '.join(x[:2])
x = x[1:]
return clean(x)
elif len(x[1]) == 1 and cdict['middle_initial'] == '':
add_mi(x[1])
x = x[:1] + x[2:]
clean(x)
elif x[2] in sns and len(x[3]) == 1:
cdict['suffix_name'] = x[2]
add_mi(x[3])
x = x[:2]
return clean(x)
elif re.match('DE L[O|A]?. ', ' '.join(x[-3:])):
x[-3] = ' '.join(x[-3:])
x = x[:-2]
return clean(x)
elif re.match('L*', x[0]) or (len(x[0]) == 1
and len(x[1]) == 1):
x[1] = ' '.join(x[:2])
x = x[1:]
return clean(x)
elif x[-2] + x[-1] in lns or (x[-2] in sln and len(x[-1]) > 3):
x[-2] = ' '.join(x[-2:])
x = x[:-1]
return clean(x)
else:
print(x)
raise
else:
print(x)
raise
return None
if len(np) > 1:
clean(np)
assert all([k in ' '.join(cdict.values()) for k in name]),\
print(cdict, name)
cdict['first_name_NS'] = string_strip(cdict['first_name'],
no_sep=True)
cdict['last_name_NS'] = string_strip(cdict['last_name'],
no_sep=True)
return cdict
def __init__(self,
full_name='',
human_name='',
first_name='',
last_name='',
middle_initial='',
middle_name=''):
"""Takes in name or name parts and performs standardization/cleaning
Creates object for single name/name parts and performs standardization
and cleaning operations on name, regardless of initial name format.
Parameters
----------
full_name : str
Name in 'Last, First' format
human_name : str
Name in 'First Last' format
first_name : str
last_name : str
middle_initial : str
middle_name : str
"""
self.full_name = full_name
self.human_name = human_name
self.first_name = first_name
self.first_name_NS = ''
self.last_name = last_name
self.last_name_NS = ''
self.middle_name = middle_name
self.middle_initial = middle_initial
self.middle_initial2 = ''
self.suffix_name = ''
self.second_name = ''
if full_name:
if not re.search('[a-zA-Z]', self.full_name):
self.full_name = ','
self.last_name, self.first_name = self.full_name.rsplit(',', 1)
elif human_name:
# Not perfect, needs work, see tests
self.human_name = re.sub(r'MC\s', 'MC',
self.human_name).replace('.', ' ')
name_dict = HumanName(self.human_name).as_dict()
if name_dict['middle']:
if name_dict['middle'] in ['O', 'MC']:
# Change later, over counts number of O' ____ names
name_dict['last'] = ' '.join(
[name_dict['middle'], name_dict['last']])
else:
name_dict['first'] = ' '.join(
[name_dict['first'], name_dict['middle']])
if name_dict['suffix']:
name_dict['last'] = ' '.join(
[name_dict['last'], name_dict['suffix']])
self.last_name = name_dict['last']
self.first_name = name_dict['first']
self.first_name = string_strip(self.first_name, no_sep=False)
self.last_name = string_strip(self.last_name, no_sep=False)
