BulgarianEGNCheck/EGNCheck.py at main · CloudVisionApps/BulgarianEGNCheck · 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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
class EGNCheck:
    def __init__(self):
        self.EGN_REGIONS = {
            "Благоевград": 43, "Бургас": 93, "Варна": 139, "Велико Търново": 169, "Видин": 183,
            "Враца": 217, "Габрово": 233, "Кърджали": 281, "Кюстендил": 301, "Ловеч": 319,
            "Монтана": 341, "Пазарджик": 377, "Перник": 395, "Плевен": 435, "Пловдив": 501,
            "Разград": 527, "Русе": 555, "Силистра": 575, "Сливен": 601, "Смолян": 623,
            "София - град": 721, "София - окръг": 751, "Стара Загора": 789, "Добрич (Толбухин)": 821,
            "Търговище": 843, "Хасково": 871, "Шумен": 903, "Ямбол": 925, "Друг/Неизвестен": 999
        }
        self.EGN_WEIGHTS = [2, 4, 8, 5, 10, 9, 7, 3, 6]
        self.MONTHS_BG = {
            1: "януари", 2: "февруари", 3: "март", 4: "април", 5: "май", 6: "юни",
            7: "юли", 8: "август", 9: "септември", 10: "октомври", 11: "ноември", 12: "декември"
        }

    def check_is_valid(self, egn):
        if len(egn) != 10:
            return False

        year = int(egn[0:2])
        month = int(egn[2:4])
        day = int(egn[4:6])

        if month > 40:
            if not (1 <= (month - 40) <= 12) or not (1 <= day <= 31):
                return False
        elif month > 20:
            if not (1 <= (month - 20) <= 12) or not (1 <= day <= 31):
                return False
        else:
            if not (1 <= month <= 12) or not (1 <= day <= 31):
                return False

        checksum = int(egn[9])
        egn_sum = sum(int(egn[i]) * self.EGN_WEIGHTS[i] for i in range(9))

        valid_checksum = egn_sum % 11
        if valid_checksum == 10:
            valid_checksum = 0

        return checksum == valid_checksum

    def parse_egn(self, egn):
        if not self.check_is_valid(egn):
            return None

        ret = {
            "year": int(egn[0:2]),
            "month": int(egn[2:4]),
            "day": int(egn[4:6]),
            "birthday_text": None,
            "region_num": int(egn[6:9]),
            "sex": int(egn[8]) % 2,
            "sex_text": "жена",
            "region_text": None,
            "birthnumber": None
        }
        if not ret["sex"]:
            ret["sex_text"] = "мъж"

        if ret["month"] > 40:
            ret["month"] -= 40
            ret["year"] += 2000
        elif ret["month"] > 20:
            ret["month"] -= 20
            ret["year"] += 1800
        else:
            ret["year"] += 1900

        ret["birthday_text"] = f"{ret['day']} {self.MONTHS_BG[ret['month']]} {ret['year']} г."

        first_region_num = 0
        for region_name, last_region_num in self.EGN_REGIONS.items():
            if ret["region_num"] >= first_region_num and ret["region_num"] <= last_region_num:
                ret["region_text"] = region_name
                break
            first_region_num = last_region_num + 1

        if int(egn[8]) % 2 != 0:
            ret["region_num"] -= 1

        ret["birthnumber"] = (ret["region_num"] - first_region_num) / 2 + 1

        return ret

    def egn_info(self, egn):
        result = {}

        if not self.check_is_valid(egn):
            result["message"] = f"<b>{egn}</b> невалиден ЕГН"
            return result

        data = self.parse_egn(egn)
        message = f"<b>{egn}</b> е ЕГН на <b>{data['sex_text']}</b>, роден{'' if data['sex'] else 'а'} " \
                  f"на <b>{data['birthday_text']}</b> в регион <b>{data['region_text']}</b> "

        if data["birthnumber"] - 1 > 0:
            message += f"като преди {'нея' if data['sex'] else 'него'} "
            if data["birthnumber"] - 1 > 1:
                message += f"в този ден и регион са се родили <b>{data['birthnumber'] - 1}</b>"
                message += ' момичета' if data['sex'] else ' момчета'
            else:
                message += f"в този ден и регион се е родило <b>1</b>"
                message += ' момиче' if data['sex'] else ' момче'
        else:
            message += f"като е {'била' if data['sex'] else 'билa'} "
            message += f"<b>първото {'момиче' if data['sex'] else 'момче'}</b> "
            message += "родено в този ден и регион"

        data["message"] = message
        return data