Add parameter to force capitalization of mixed case strings

Author: derek73

docs/release_log.rst

@@ -4,6 +4,7 @@ Release Log
     - Remove "CONSTANTS.suffixes", replaced by "suffix_acronyms" and "suffix_not_acronyms" (#49)
     - Add "du" to prefixes
     - Add "sheikh" variations to titles
+    - Add parameter to force capitalization of mixed case strings
 * 0.3.16 - March 24, 2016
     - Clarify LGPL licence version (#47)
     - Skip pickle tests if pickle not installed (#48)

nameparser/parser.py

@@ -544,8 +544,8 @@ def parse_pieces(self, parts, additional_parts_count=0):
                 raise TypeError("Name parts must be strings. Got {0}".format(type(part)))
             output += [x.strip(' ,') for x in part.split(' ')]
 
-        # If there's periods, check if it's titles without spaces and add spaces
-        # so they get picked up later as titles.
+        # If part contains periods, check if it's multiple titles or suffixes together without spaces
+        # if so, add the new part with periods to the constants so they get parsed correctly later
         for part in output:
             # if this part has a period not at the beginning or end
             if self.C.regexes.period_not_at_end.match(part):
@@ -684,12 +684,15 @@ def cap_piece(self, piece):
         replacement = lambda m: self.cap_word(m.group(0))
         return self.C.regexes.word.sub(replacement, piece)
 
-    def capitalize(self):
+    def capitalize(self, force=False):
         """
         The HumanName class can try to guess the correct capitalization 
-        of name entered in all upper or lower case. It will not adjust 
-        the case of names entered in mixed case.
+        of name entered in all upper or lower case. By default, it will not adjust 
+        the case of names entered in mixed case. To run capitalization on all names
+        pass the parameter `force=True`.
         
+        :param bool force: force capitalization of strings that include mixed case
+
         **Usage**
         
         .. doctest:: capitalize
@@ -703,10 +706,13 @@ def capitalize(self):
             >>> name.capitalize()
             >>> str(name) 
             'Shirley Maclaine'
+            >>> name.capitalize(force=True)
+            >>> str(name) 
+            'Shirley MacLaine'
         
         """
         name = u(self)
-        if not (name == name.upper() or name == name.lower()):
+        if not force and not (name == name.upper() or name == name.lower()):
             return
         self.title_list  = self.cap_piece(self.title ).split(' ')
         self.first_list  = self.cap_piece(self.first ).split(' ')

tests.py

@@ -1791,6 +1791,11 @@ def test_no_change_to_mixed_chase(self):
         hn.capitalize()
         self.m(str(hn), 'Shirley Maclaine', hn)
 
+    def test_force_capitalization(self):
+        hn = HumanName('Shirley Maclaine')
+        hn.capitalize(force=True)
+        self.m(str(hn), 'Shirley MacLaine', hn)
+
     def test_capitalize_diacritics(self):
         hn = HumanName('matthëus schmidt')
         hn.capitalize()