diff --git a/backend/src/core/beautifier/beautifier_tzb.py b/backend/src/core/beautifier/beautifier_tzb.py index 587b3b6..c6576d0 100644 --- a/backend/src/core/beautifier/beautifier_tzb.py +++ b/backend/src/core/beautifier/beautifier_tzb.py @@ -64,12 +64,12 @@ def parse_row(self, row: Series) -> Dict[str, str]: return { "phone_number": str(row["num"]).replace(" ", ""), "region": row["REGION"], - "operator": row["OPERATOR"], + "operator": row["Оператор сотовой связи"], "Пол": row["Пол"], "Возраст": row["Возраст"], "iSayMail": row["iSayMail"], "Reward": row["Reward"], - "Source": row["Source"], + "Source": row["Source"] } def check_if_region_is_ignored(self, tailored_row: Dict[str, str]) -> bool: @@ -101,8 +101,8 @@ def make_tailored_row(self, parsed_row: Dict[str, str]) -> Dict[str, str]: "TimeDifference": self.get_time_difference(region), "Region": region_code, "Operator": operator_code, - "CallIntervalBegin": interval["begin"], - "CallIntervalEnd": interval["end"], + "CallIntervalBegin": "10:00:00", + "CallIntervalEnd": "22:00:00", "Group": self.get_tzb_group(region, operator, parsed_row["Source"]), "CHECK": self.get_external_id(phone_number), "Mark": self.get_tzb_mark(region_code, operator_code), diff --git a/backend/src/tests/beautifier/fixtures.py b/backend/src/tests/beautifier/fixtures.py index 4e92a56..a58b0ee 100644 --- a/backend/src/tests/beautifier/fixtures.py +++ b/backend/src/tests/beautifier/fixtures.py @@ -10,6 +10,7 @@ def row_simple() -> Series: "IdRegion": 83, "IdOper": 28, "REGION": "Свердловская обл.", + "Оператор сотовой связи": 'ООО "ЕКАТЕРИНБУРГ-2000"', "OPERATOR": 'ООО "ЕКАТЕРИНБУРГ-2000"', "Пол": "Ж", "Возраст": 30, @@ -28,6 +29,7 @@ def row_simple_source_two() -> Series: "IdRegion": 83, "IdOper": 28, "REGION": "Свердловская обл.", + "Оператор сотовой связи": 'ООО "ЕКАТЕРИНБУРГ-2000"', "OPERATOR": 'ООО "ЕКАТЕРИНБУРГ-2000"', "Пол": "Ж", "Возраст": 30, @@ -46,6 +48,7 @@ def row_for_nonexistent_region() -> Series: "IdRegion": 83, "IdOper": 28, "REGION": "Волчий край", + "Оператор сотовой связи": 'ООО "ЕКАТЕРИНБУРГ-2000"', "OPERATOR": 'ООО "ЕКАТЕРИНБУРГ-2000"', "Пол": "Ж", "Возраст": 30, @@ -64,6 +67,7 @@ def row_for_chukotka() -> Series: "IdRegion": 104, "IdOper": 78, "REGION": "Чукотский АО", + "Оператор сотовой связи": 'ПАО "Вымпел-Коммуникации"', "OPERATOR": 'ПАО "Вымпел-Коммуникации"', "Пол": "Ж", "Возраст": 30, @@ -82,6 +86,7 @@ def row_for_motiv() -> Series: "IdRegion": 83, "IdOper": 28, "REGION": "Курганская обл.", + "Оператор сотовой связи": 'ООО "ЕКАТЕРИНБУРГ-2000"', "OPERATOR": 'ООО "ЕКАТЕРИНБУРГ-2000"', "Пол": "Ж", "Возраст": 30, @@ -100,6 +105,7 @@ def row_for_yota_moscow() -> Series: "IdRegion": 83, "IdOper": 28, "REGION": "г. Москва * Московская область", + "Оператор сотовой связи": 'ООО "Скартел"', "OPERATOR": 'ООО "Скартел"', "Пол": "Ж", "Возраст": 30, @@ -118,6 +124,7 @@ def row_for_yota_krasnodar_kray() -> Series: "IdRegion": 83, "IdOper": 28, "REGION": "Краснодарский край", + "Оператор сотовой связи": 'ООО "Скартел"', "OPERATOR": 'ООО "Скартел"', "Пол": "Ж", "Возраст": 30, @@ -136,6 +143,7 @@ def row_for_sim_telecom() -> Series: "IdRegion": 83, "IdOper": 28, "REGION": "Краснодарский край", + "Оператор сотовой связи": 'ООО "СИМ ТЕЛЕКОМ"', "OPERATOR": 'ООО "СИМ ТЕЛЕКОМ"', "Пол": "Ж", "Возраст": 30, diff --git a/backend/src/tests/beautifier/test_beautifier_tzb.py b/backend/src/tests/beautifier/test_beautifier_tzb.py index dcb63e4..2b4fab3 100644 --- a/backend/src/tests/beautifier/test_beautifier_tzb.py +++ b/backend/src/tests/beautifier/test_beautifier_tzb.py @@ -188,8 +188,8 @@ def test_make_tailored_row_for_tzb(row_simple): "TimeDifference": "UTC +5", "Region": 12, "Operator": 7, - "CallIntervalBegin": "08:00:00", - "CallIntervalEnd": "20:00:00", + "CallIntervalBegin": "10:00:00", + "CallIntervalEnd": "22:00:00", "Group": "Свердловская область_Мотив_iSay", "CHECK": "9001979228", "Mark": "12_7", diff --git a/backend/src/utils/config_storage/config.json b/backend/src/utils/config_storage/config.json index 84091cd..abdc260 100644 --- a/backend/src/utils/config_storage/config.json +++ b/backend/src/utils/config_storage/config.json @@ -421,7 +421,16 @@ "Мотив": "Мотив", "Другой оператор": "Другие", "СберМобайл": "Другие", - "Теле-2": "Теле-2" + "Теле-2": "Теле-2", + "UNKNOWN": "Неизвестный", + "MTS": "МТС", + "MEGAFON": "МегаФон", + "BEELINE": "Билайн", + "TELE2": "Теле-2", + "YOTA": "Йота", + "OTHER": "Другие", + "MOTIV": "Мотив", + "TINKOFF": "Другие" }, "operator_codes": { "МТС": 1,