correct missing required sites in power_feeds

This commit is contained in:
ryanmerolle 2021-01-15 14:25:42 -05:00
parent 00fa1793d0
commit 788aeacd9b
1 changed files with 3 additions and 2 deletions

View File

@ -1,6 +1,6 @@
import sys import sys
from dcim.models import Rack, PowerPanel, PowerFeed from dcim.models import Rack, PowerPanel, PowerFeed, Site
from startup_script_utils import * from startup_script_utils import *
from tenancy.models import Tenant from tenancy.models import Tenant
@ -10,7 +10,8 @@ if power_feeds is None:
sys.exit() sys.exit()
required_assocs = { required_assocs = {
'power_panel': (PowerPanel, 'name') 'power_panel': (PowerPanel, 'name'),
'site': (Site, 'name')
} }
optional_assocs = { optional_assocs = {