dv/utp_service
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

соннет вайб 1

Browse source
This commit is contained in:
dv 2026-04-08 12:45:04 +03:00
parent 23ee10a28e
commit 6c081d5580
1 changed files with 15 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
backend/app/services/device_discovery.py
View file

@ -488,13 +488,22 @@ async def discover_via_ssh(
return result
# Step 2: SSH into each rack, parse config, upsert devices
# Build mapping of external_id -> Rack ID for device assignment
rack_id_mapping = {}
# Build mapping of external_id -> Rack.id for device assignment.
# If a rack with that name doesn't exist yet in our racks table, create it now.
from app.models.rack import Rack
all_racks = (await db.execute(select(Rack).where(Rack.object_id == obj.id))).scalars().all()
for rack in all_racks:
if rack.external_id:
rack_id_mapping[rack.external_id] = rack.id
rack_id_mapping: dict[str, int] = {}
for ext_id, _, _ in rack_hosts:
existing_rack_row = (await db.execute(
select(Rack).where(Rack.object_id == obj.id, Rack.name == ext_id)
)).scalar_one_or_none()
if existing_rack_row:
rack_id_mapping[ext_id] = existing_rack_row.id
else:
new_rack = Rack(object_id=obj.id, name=ext_id)
db.add(new_rack)
await db.flush() # get the new PK
rack_id_mapping[ext_id] = new_rack.id
logger.info(f"[ssh_discovery] Created rack entry '{ext_id}' (id={new_rack.id})")
for external_id, rack_host, rack_name in rack_hosts:
# Resolve config_path override from the embedded device's connection_params