Added possibility to include custom regex as allowed characters (un33k#36)

vrbaskiz · un33k · commit 750c737b37a7 · 2017-03-31T11:47:10.000-04:00
Added possibility to allow custom regex as allowed characters.
diff --git a/slugify/slugify.py b/slugify/slugify.py
@@ -71,7 +71,7 @@ def smart_truncate(string, max_length=0, word_boundaries=False, separator=' ', s
 
 
 def slugify(text, entities=True, decimal=True, hexadecimal=True, max_length=0, word_boundary=False,
-            separator='-', save_order=False, stopwords=()):
+            separator='-', save_order=False, stopwords=(), allowed_characters=None):
     """
     Make a slug from the given text.
     :param text (str): initial text
@@ -131,7 +131,11 @@ def slugify(text, entities=True, decimal=True, hexadecimal=True, max_length=0, w
 
     # replace unwanted characters
     text = NUMBERS_PATTERN.sub('', text)
-    text = ALLOWED_CHARS_PATTERN.sub('-', text)
+    print(text)
+    if allowed_characters:
+        text = re.sub(allowed_characters, '-', text)
+    else:
+        text = ALLOWED_CHARS_PATTERN.sub('-', text)
 
     # remove redundant -
     text = DUPLICATE_DASH_PATTERN.sub('-', text).strip('-')
diff --git a/test.py b/test.py
@@ -167,6 +167,11 @@ def test_numbers_and_symbols(self):
         r = slugify(txt)
         self.assertEqual(r, '1000-reasons-you-are-1')
 
+    def test_underscore(self):
+        txt = "___This is a test___"
+        r = slugify(txt, allowed_characters=r'[^-a-z0-9_]+')
+        self.assertEqual(r, "___this-is-a-test___")
+
 
 class TestUtils(unittest.TestCase):
 
