add script order

.gitignore

@@ -1 +1,2 @@
-/.idea
+/.venv/
+/.idea/

add_tenant_with_prefix.py

@@ -0,0 +1,324 @@
+import re
+
+from tenancy.models import Tenant, TenantGroup, Contact, ContactRole, ContactAssignment
+from ipam.models import Prefix, VRF, Role
+from dcim.models import DeviceType, DeviceRole, Device
+from extras.scripts import Script, StringVar, IPNetworkVar, ChoiceVar, ObjectVar
+from django.core.exceptions import ValidationError
+from extras.models import CustomField
+from utilities.exceptions import AbortScript
+
+
+def slugify(text: str):
+    CYRILLIC_TO_LATIN = {
+        "а": "a",
+        "б": "b",
+        "в": "v",
+        "г": "g",
+        "д": "d",
+        "е": "e",
+        "ё": "e",
+        "ж": "zh",
+        "з": "z",
+        "и": "i",
+        "й": "i",
+        "к": "k",
+        "л": "l",
+        "м": "m",
+        "н": "n",
+        "о": "o",
+        "п": "p",
+        "р": "r",
+        "с": "s",
+        "т": "t",
+        "у": "u",
+        "ф": "f",
+        "х": "h",
+        "ц": "c",
+        "ч": "ch",
+        "ш": "sh",
+        "щ": "shch",
+        "ъ": "",
+        "ы": "y",
+        "ь": "",
+        "э": "e",
+        "ю": "yu",
+        "я": "ya",
+        "А": "A",
+        "Б": "B",
+        "В": "V",
+        "Г": "G",
+        "Д": "D",
+        "Е": "E",
+        "Ё": "E",
+        "Ж": "ZH",
+        "З": "Z",
+        "И": "I",
+        "Й": "I",
+        "К": "K",
+        "Л": "L",
+        "М": "M",
+        "Н": "N",
+        "О": "O",
+        "П": "P",
+        "Р": "R",
+        "С": "S",
+        "Т": "T",
+        "У": "U",
+        "Ф": "F",
+        "Х": "H",
+        "Ц": "C",
+        "Ч": "CH",
+        "Ш": "SH",
+        "Щ": "SHCH",
+        "Ъ": "",
+        "Ы": "Y",
+        "Ь": "",
+        "Э": "E",
+        "Ю": "YU",
+        "Я": "YA",
+    }
+    text = text.replace("ООО", "").lstrip()
+    text = "".join(CYRILLIC_TO_LATIN.get(c, c) for c in text)
+    text = text.lower()
+    text = re.sub(r"\s+", "_", text)
+    text = re.sub(r"[^a-z0-9_-]", "", text)
+    text = re.sub(r"[-_]{2,}", "_", text).strip("_-")
+
+    return text
+
+
+class CreateTenant(Script):
+    class Meta:
+        name = "Создание Оператора"
+        description = "Создание оператора и связанного префикса"
+        scheduling_enabled = False
+        fieldsets = (
+            (
+                "Оператор",
+                (
+                    "tenant_name",
+                    "tenant_group",
+                    "tenant_connection_type",
+                    "tenant_description",
+                ),
+            ),
+            (
+                "Префикс",
+                ("prefix_cidr", "prefix_vrf", "prefix_role", "prefix_description"),
+            ),
+            ("Документы", ("tenant_contract", "tenant_prefix_task", "tenant_tr")),
+            ("Контакты", ("contacts_fio", "contacts_phone", "contacts_email")),
+        )
+
+    connection_types = CustomField.objects.get(name="connection_type_tenant")
+
+    tenant_name = StringVar(label="Имя оператора")
+    tenant_group = ObjectVar(
+        label="Группа оператора", model=TenantGroup, required=False
+    )
+    tenant_connection_type = ChoiceVar(
+        label="Тип подключения", choices=connection_types.choices, required=True
+    )
+    tenant_description = StringVar(label="Описание оператора", required=False)
+    tenant_contract = StringVar(
+        label="Договор",
+        description="Номер договора на разработку комплекса",
+        required=False,
+    )
+    tenant_tr = StringVar(
+        label="Проект/ТР", description="Пример: DOCXXXXX", required=False
+    )
+    tenant_prefix_task = StringVar(
+        label="Задача",
+        description="URL или номер задачи из redmine, jira, CRM",
+        required=False,
+    )
+
+    prefix_cidr = IPNetworkVar(
+        label="Префикс для оператора (CIDR)",
+        description="Указывается с маской подсети /xx",
+    )
+    prefix_vrf = ObjectVar(label="VRF проекта", model=VRF, required=True)
+    prefix_role = ObjectVar(
+        label="Роль",
+        description="Роль для малых операторов IPMI-SSH-MGMT",
+        model=Role,
+        required=True,
+    )
+    prefix_description = StringVar(label="Описание префикса", required=False)
+
+    contacts_fio = StringVar(
+        label="ФИО",
+        description="Указывается в формате Иванов Иван Иванович, остальные значения игнорируются при отсутсвии ФИО",
+        required=False,
+    )
+    contacts_phone = StringVar(label="Номер тел.", required=False)
+    contacts_email = StringVar(label="E-mail", required=False)
+    contacts_role = ObjectVar(label="Роль контакта", model=ContactRole, required=False)
+
+    def check_tenant_exists(self, data):
+        if Tenant.objects.filter(name=data["tenant_name"]).exists():
+            raise ValidationError(
+                f"Оператор с именем '{data['tenant_name']}' уже существует"
+            )
+
+    def check_prefix_exists(self, data):
+        if Prefix.objects.filter(prefix=data["prefix_cidr"]).exists():
+            self.log_warning(
+                f"Префикс {data['prefix_cidr']} уже существует, оператор будет создан без префикса"
+            )
+            return False
+        return True
+
+    def run(self, data, commit):
+        try:
+            self.check_tenant_exists(data)
+            validate_prefix = self.check_prefix_exists(data)
+
+            tenant_slug = slugify(data["tenant_name"])
+            tenant_data = {
+                "name": data["tenant_name"],
+                "slug": tenant_slug,
+                "group": data["tenant_group"],
+            }
+
+            if data.get("tenant_description"):
+                tenant_data["description"] = data["tenant_description"]
+
+            tenant = Tenant(**tenant_data)
+            tenant.custom_field_data["dogovor"] = data["tenant_contract"]
+            tenant.custom_field_data["connection_type_tenant"] = data[
+                "tenant_connection_type"
+            ]
+            tenant.custom_field_data["link_on_docs"] = data["tenant_tr"]
+            tenant.full_clean()
+            if commit:
+                tenant.save()
+                self.log_success(
+                    f"Создан новый оператор: {tenant.name} - {tenant.slug}"
+                )
+                if validate_prefix:
+                    prefix = Prefix(
+                        prefix=data["prefix_cidr"],
+                        tenant=tenant,
+                        description=data.get("prefix_description") or "",
+                        vrf=data["prefix_vrf"],
+                        role=data["prefix_role"],
+                    )
+                    prefix.custom_field_data["Project"] = data["tenant_tr"]
+                    prefix.custom_field_data["zadacha"] = data["tenant_prefix_task"]
+                    prefix.full_clean()
+                    prefix.save()
+                    self.log_success(
+                        f"Создан префикс {prefix.prefix} для оператора: {tenant.name}"
+                    )
+                if data.get("contacts_fio"):
+                    contact = Contact(
+                        name=data["contacts_fio"],
+                        phone=data["contacts_phone"],
+                        email=data["contacts_email"],
+                    )
+                    try:
+                        contact.full_clean()
+                        contact.save()
+                        self.log_success(f"Создан контакт: {contact.name}")
+                    except Exception as e:
+                        self.log_failure(f"Ошибка при создании контакта: {e}")
+                    try:
+                        role = data.get("contacts_role") or ContactRole.objects.first()
+                        assignment = ContactAssignment(
+                            contact=contact,
+                            role=role,
+                            object=tenant,
+                        )
+                        assignment.full_clean()
+                        assignment.save()
+                        self.log_success(
+                            f"Связан контакт {contact.name} с оператором: {tenant.name}"
+                        )
+                    except Exception as e:
+                        self.log_failure(
+                            f"Ошибка при связывании контакта с оператором: {e}"
+                        )
+
+            else:
+                self.log_info(
+                    f"Тестовый режим: Будет создан оператор: {tenant.name} (slug: {tenant.slug})"
+                )
+                self.log_info(
+                    f"Тестовый режим: Будет создан префикс: {data['prefix_cidr']}"
+                )
+                try:
+                    tenant.full_clean()
+                    prefix = Prefix(
+                        prefix=data["prefix_cidr"],
+                        tenant=tenant,
+                        description=data.get("prefix_description") or "",
+                    )
+                    prefix.full_clean()
+                    contact = Contact(
+                        name=data["contacts_fio"],
+                        phone=data["contacts_phone"],
+                        email=data["contacts_email"],
+                    )
+                    contact.full_clean()
+                    role = data.get("contacts_role") or ContactRole.objects.first()
+                    assignment = ContactAssignment(
+                        contact=contact,
+                        role=role,
+                        object=tenant,
+                    )
+                    assignment.full_clean()
+                    self.log_success("Все данные валидны")
+                except ValidationError as e:
+                    self.log_failure(f"Ошибка валидации: {e}")
+        except ValidationError as e:
+            raise AbortScript(f"Ошибка валидации: {e}")
+        except Exception as e:
+            self.log_failure(f"Ошибка при выполнении скрипта: {e}")
+            if commit:
+                self.log_failure("Изменения отменены из-за ошибки")
+            raise
+
+
+class CreateDevice(Script):
+    class Meta:
+        name = "Создание устройства"
+        description = "Создание устройства и привязка к оператору"
+        scheduling_enabled = False
+        fieldsets = (
+            (
+                "Устройство",
+                (
+                    "device_name",
+                    "device_type",
+                    "device_role",
+                    "device_description",
+                    "tenant_group",
+                    "device_tenant",
+                ),
+            ),
+        )
+
+    device_name = StringVar(label="Имя устройства")
+    device_role = ObjectVar(label="Роль устройства", model=DeviceRole, required=True)
+    device_type = ObjectVar(label="Тип устройства", model=DeviceType, required=True)
+    device_description = StringVar(label="Описание устройства", required=False)
+    tenant_group = ObjectVar(
+        label="Группа оператора", model=TenantGroup, required=False
+    )
+    device_tenant = ObjectVar(
+        label="Оператор",
+        model=Tenant,
+        required=True,
+        query_params={
+            "group_id": "$tenant_group",
+        },
+    )
+
+    def run(self, data, commit):
+        pass
+
+
+script_order = (CreateTenant, CreateDevice)

check_vchassis_stack_ports.py

@@ -1,75 +0,0 @@
-import re
-
-from dcim.models.devices import Device
-from extras.models.tags import Tag
-from extras.scripts import ObjectVar, Script
-from dcim.models import VirtualChassis, Interface
-from netbox.choices import ColorChoices
-from utilities.exceptions import AbortScript
-
-TAGS = "sync:vchas_interface"
-
-
-class Vchassis_checker(Script):
-    IFACE_RE = re.compile(
-        r"^(?P<prefix>[A-Za-z]+)\s?(?:(?P<slot>\d+)/)?(?P<rest>\d+/\d+)$"
-    )
-
-    class Meta(Script.Meta):
-        commit_default = False
-        scheduling_enabled = False
-        name = "Переопределение интерфейсов"
-        description = "Переопределение интерфейсов у стекированных коммутаторов"
-
-    virtual_chassis = ObjectVar(
-        model=VirtualChassis,
-        description="Виртуальное шасси без тегов",
-        query_params={"tag__n": [TAGS.replace(":", "_")]},
-    )
-
-    def run(self, data, commit):
-        if Tag.objects.filter(name=TAGS).exists():
-            tag = Tag.objects.get(name=TAGS)
-        else:
-            self.log_warning(f"Отсутствует tag {TAGS}")
-            tag_data = {
-                "color": ColorChoices.COLOR_LIGHT_BLUE,
-                "name": TAGS,
-                "slug": TAGS.replace(":", "_"),
-                "description": "тег для автоматического назначения на синхронизированные интерфейсы vchassis",
-            }
-            tag = Tag(**tag_data)
-            if commit:
-                tag.full_clean()
-                tag.save()
-                self.log_success(
-                    f"Создан тег: {TAGS}",
-                    obj=tag,
-                )
-        vc = data["virtual_chassis"]
-        devices: list[Device] = list(vc.members.all().order_by("vc_position"))
-        if not vc._state.adding and vc.master and vc.master not in devices:
-            raise AbortScript(f"В стеке {vc.name} отсутствует мастер")
-        device_ids = [d.pk for d in devices]
-        interfaces_qs = Interface.objects.filter(
-            device_id__in=device_ids
-        ).select_related("device")
-
-        to_update = []
-        for iface in interfaces_qs:
-            m = self.IFACE_RE.match(iface.name)
-            if m:
-                prefix = m.group("prefix")
-                slot = m.group("slot")
-                rest = m.group("rest")
-                iface.name = f"{prefix}{iface.device.vc_position}/{rest}"
-                to_update.append(iface)
-        if commit:
-            Interface.objects.bulk_update(to_update, fields=["name"], batch_size=100)
-            vc.tags.add(tag)
-            for iface in to_update:
-                iface.tags.add(tag)
-
-            self.log_success(f"Обновлено интерфейсов: {len(to_update)}")
-        else:
-            self.log_success(f"Будет обновлено интерфейсов: {len(to_update)}")