From 788aeacd9b4377fac4f751efeaacf4ea02c646c7 Mon Sep 17 00:00:00 2001
From: ryanmerolle <ryan.merolle@gmail.com>
Date: Fri, 15 Jan 2021 14:25:42 -0500
Subject: [PATCH] correct missing required sites in power_feeds

---
 startup_scripts/340_power_feeds.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/startup_scripts/340_power_feeds.py b/startup_scripts/340_power_feeds.py
index fc7ee38..8660da5 100644
--- a/startup_scripts/340_power_feeds.py
+++ b/startup_scripts/340_power_feeds.py
@@ -1,6 +1,6 @@
 import sys
 
-from dcim.models import Rack, PowerPanel, PowerFeed
+from dcim.models import Rack, PowerPanel, PowerFeed, Site
 from startup_script_utils import *
 from tenancy.models import Tenant
 
@@ -10,7 +10,8 @@ if power_feeds is None:
   sys.exit()
 
 required_assocs = {
-  'power_panel': (PowerPanel, 'name')
+  'power_panel': (PowerPanel, 'name'),
+  'site': (Site, 'name')
 }
 
 optional_assocs = {