Changeset 9593 for src/django_gheat/gheat/management/commands

Timestamp:

Aug 28, 2011, 1:34:06 PM (14 years ago)

Author:

rick

Message:

Fix the organization imports

File:

• src/django_gheat/gheat/management/commands/import_nodes.py (modified) (3 diffs)

src/django_gheat/gheat/management/commands/import_nodes.py

@@ -3 +3 @@
 #
 # Script for importing nodes from yaml information generated by gformat.py
+#
+# Example: $ ~/wleiden/node-config/genesis/tools/gformat.py  full-export |
+# ./manage.py import_nodes -
 #
 # Rick van der Zwet <info@rickvanderzwet.nl>
@@ -25 +28 @@
   args = '[<filename>]'
   help = 'Specify - to read from stdin'
+  option_list = BaseCommand.option_list + (
+    make_option('-g', '--debug', dest='debug', default='False', help="Debug", action="store_true"),
+  )
 
   def handle(self, *args, **options):
+    if options['debug']:
+      logger.setLevel(logging.DEBUG)
     if len(args) == 0:
       self.print_help(sys.argv[0],sys.argv[1])
@@ -46 +54 @@
     hosts = yaml.load(fh)
     for host,items in hosts.iteritems():
-      node, created = Node.objects.get_or_create(name=host,
-         latitude=items['latitude'],longitude=items['longitude'])
+      for iface_key in [x for x in items.keys() if x.startswith('iface_')]:
+        if items[iface_key].has_key('ssid'):
+          org = Organization.get_by_ssid(items[iface_key]['ssid'])
+          if org: break
+          logger.error("NO Organization for SSID '%s'", items[iface_key]['ssid'])
+
+      if not org:
+        logger.warning("NO Organization for Node '%s'", host)
+
+      node, created = Node.objects.get_or_create(name=host)
       if created:
-        logger.info("Created Node '%s'" % node))
-    logger.info("Processed %s items" % len(hosts))
+        logger.info("Created Node '%s'", node)
+      node.latitude = items['latitude']
+      node.longitude = items['longitude']
+      node.organization = org
+      node.save()
+    logger.info("Processed %s items", len(hosts))