Compare commits
2 Commits
fc0a232808
...
b715019845
| Author | SHA1 | Date | |
|---|---|---|---|
|
b715019845 | ||
|
|
3268274f69 |
2
.gitignore
vendored
2
.gitignore
vendored
@@ -1,2 +1,4 @@
|
||||
/.venv/
|
||||
/.idea/
|
||||
/.netbox/netbox
|
||||
/.vscode/
|
||||
|
||||
1
.python-version
Normal file
1
.python-version
Normal file
@@ -0,0 +1 @@
|
||||
3.12
|
||||
@@ -1,324 +0,0 @@
|
||||
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)
|
||||
7
pyproject.toml
Normal file
7
pyproject.toml
Normal file
@@ -0,0 +1,7 @@
|
||||
[project]
|
||||
name = "netbox-scripts"
|
||||
version = "0.1.0"
|
||||
description = "Add your description here"
|
||||
readme = "README.md"
|
||||
requires-python = ">=3.12"
|
||||
dependencies = []
|
||||
12
pyrightconfig.json
Normal file
12
pyrightconfig.json
Normal file
@@ -0,0 +1,12 @@
|
||||
{
|
||||
"extraPaths": ["netbox/netbox"],
|
||||
"typeCheckingMode": "off",
|
||||
"reportMissingImports": "warning",
|
||||
"reportMissingTypeStubs": "none",
|
||||
"reportGeneralTypeIssues": "none",
|
||||
"reportOptionalMemberAccess": "none",
|
||||
"reportOptionalSubscript": "none",
|
||||
"reportArgumentType": "none",
|
||||
"reportAssignmentType": "none",
|
||||
"reportReturnType": "none"
|
||||
}
|
||||
Reference in New Issue
Block a user