Adicionada constante CPF_LENGHT para melhor leitura do código

brazilian-utils · Oct 29, 2024 · 831aead · 831aead
1 parent dba3ad4
commit 831aead
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/brutils/cpf.py b/brutils/cpf.py
@@ -271,13 +271,13 @@ def remove_symbols_cpf(cpf: str) -> str:
         >>> remove_symbols_cpf('000.111.222-333')
         None
     """
+    CPF_LENGHT = 11
 
-    result = None
-    cpf_no_symbols = ""
-    if isinstance(cpf, str) and cpf != "":
-        cpf_no_symbols = "".join([char for char in cpf if char not in ".-"])
+    if not cpf:
+        return None
 
-    if len(cpf_no_symbols) == 11:
-        result = cpf_no_symbols
+    if isinstance(cpf, str) and cpf != "":
+        cpf_no_symbols = "".join([char for char in cpf if char.isalnum()])
 
-    return result
+        if len(cpf_no_symbols) == CPF_LENGHT:
+            return cpf_no_symbols