python-utils/doc-formatter.py at main · venzellucas/python-utils · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
def validate_cpf(x):
    x = x.zfill(11)
    try:
        mult = list(range(10,1,-1))
        sum_prod = 0
        for i in range(len(mult)):
            sum_prod += int(x[i]) * mult[i]
        assert sum_prod * 10 % 11 %10 == int(x[9])

        mult = list(range(11,1,-1))
        sum_prod = 0
        for i in range(len(mult)):
            sum_prod += int(x[i]) * mult[i]
        assert sum_prod * 10 % 11 %10 == int(x[10])

        return True, x

    except:
        return False, None

def validate_cnpj(x):
    x = x.zfill(14)
    try:
        mult = [5,4,3,2,9,8,7,6,5,4,3,2]
        sum_prod = 0
        for i in range(len(mult)):
            sum_prod += int(x[i]) * mult[i]
        digit = sum_prod % 11
        if digit < 2:
            assert int(x[12]) == 0
        else:
            assert int(x[12]) == 11 - digit

        mult = [6,5,4,3,2,9,8,7,6,5,4,3,2]
        sum_prod = 0
        for i in range(len(mult)):
            sum_prod += int(x[i]) * mult[i]
        digit = sum_prod % 11
        if digit < 2:
            assert int(x[13]) == 0
        else:
            assert int(x[13]) == 11 - digit

        return True, x
    except:
        return False, None


def transform_cpf_cnpj(x):
    is_cnpj, cnpj = validate_cnpj(x)
    if is_cnpj:
        return cnpj
    is_cpf, cpf = validate_cpf(x)
    if is_cpf:
        return cpf
    return None