Enhance uuid_field function to sanitize output by replacing '-' and '_' with 'A' and 'B'

Author: gmorales96

cuenca_validations/types/helpers.py

@@ -12,7 +12,9 @@
 
 def uuid_field(prefix: str = '') -> Callable[[], str]:
     def base64_uuid_func() -> str:
-        return prefix + urlsafe_b64encode(uuid.uuid4().bytes).decode()[:-2]
+        base64_str = urlsafe_b64encode(uuid.uuid4().bytes).decode()[:-2]
+        sanitized = base64_str.replace('-', 'A').replace('_', 'B')
+        return prefix + sanitized
 
     return base64_uuid_func
 

cuenca_validations/version.py

@@ -1 +1 @@
-__version__ = '2.1.21'
+__version__ = '2.1.22'

tests/test_helpers.py

@@ -27,3 +27,10 @@ def test_multiple_generators():
     assert uuid1.startswith('CA')
     assert uuid2.startswith('TR')
     assert uuid1 != uuid2
+
+
+def test_sanitized_uuid():
+    generator = uuid_field('US')
+    uuid_str = generator()
+    assert "-" not in uuid_str
+    assert "-" not in uuid_str