Masking Personal Information LeetCode solution in Python

1
class Solution:
2
    def maskPII(self, s: str) -> str:
3
        res = ""
4
        if "@" in s:
5

6
            l = s.split("@")
7

8
            a = l[0]
9
            b = l[1]
10

11
            c = b.split(".")
12

13
            res = (
14
                a[0].lower()
15
                + "*" * 5
16
                + a[-1].lower()
17
                + "@"
18
                + c[0].lower()
19
                + "."
20
                + c[1].lower()
21
            )
22
            return res
23
        else:
24
            l = 0
25
            res = ""
26
            flag = False
27
            f = False
28
            c = 0
29
            for i in range(len(s) - 1, -1, -1):
30
                if (
31
                    s[i] == "("
32
                    or s[i] == ")"
33
                    or s[i] == "-"
34
                    or s[i] == "+"
35
                    or s[i] == " "
36
                ):
37
                    continue
38

39
                if f == True:
40
                    c += 1
41
                    continue
42

43
                if flag == False:
44
                    res = s[i] + res
45
                else:
46
                    res = "*" + res
47

48
                if l == 3 or l == 6:
49
                    if l == 3:
50
                        flag = True
51
                    res = "-" + res
52
                l += 1
53

54
                if len(res) == 12:
55
                    f = True
56
            if c == 1:
57
                res = "+*-" + res
58
            elif c == 2:
59
                res = "+**-" + res
60
            elif c == 3:
61
                res = "+***-" + res
62

63
            return res