ovsdb/ovsdb-dot.in

   1 #! @PYTHON@
   2
   3 from datetime import date
   4 import getopt
   5 import os
   6 import re
   7 import sys
   8
   9 argv0 = sys.argv[0]
  10
  11 def printEdge(tableName, baseType, label):
  12     if baseType.refTable:
  13         options = {}
  14         options['label'] = '"%s"' % label
  15         if baseType.refType == 'weak':
  16             options['constraint'] = 'false'
  17         print "\t%s -> %s [%s];" % (
  18             tableName,
  19             baseType.refTable,
  20             ', '.join(['%s=%s' % (k,v) for k,v in options.items()]))
  21
  22 def schemaToDot(schemaFile):
  23     schema = DbSchema.fromJson(ovs.json.from_file(schemaFile))
  24
  25     print "digraph %s {" % schema.name
  26     for tableName, table in schema.tables.iteritems():
  27         print '\tsize="6.5,4";'
  28         print '\tmargin="0";'
  29         print "\tnode [shape=box];"
  30         print "\t%s;" % tableName
  31         for columnName, column in table.columns.iteritems():
  32             if column.type.value:
  33                 printEdge(tableName, column.type.key, "%s key" % columnName)
  34                 printEdge(tableName, column.type.value, "%s value" % columnName)
  35             else:
  36                 printEdge(tableName, column.type.key, columnName)
  37     print "}";
  38
  39 def usage():
  40     print """\
  41 %(argv0)s: compiles ovsdb schemas to graphviz format
  42 Prints a .dot file that "dot" can render to an entity-relationship diagram
  43 usage: %(argv0)s [OPTIONS] SCHEMA
  44 where SCHEMA is an OVSDB schema in JSON format
  45
  46 The following options are also available:
  47   -h, --help                  display this help message
  48   -V, --version               display version information\
  49 """ % {'argv0': argv0}
  50     sys.exit(0)
  51
  52 if __name__ == "__main__":
  53     try:
  54         try:
  55             options, args = getopt.gnu_getopt(sys.argv[1:], 'hV',
  56                                               ['help', 'version'])
  57         except getopt.GetoptError, geo:
  58             sys.stderr.write("%s: %s\n" % (argv0, geo.msg))
  59             sys.exit(1)
  60
  61         for key, value in options:
  62             if key in ['-h', '--help']:
  63                 usage()
  64             elif key in ['-V', '--version']:
  65                 print "ovsdb-dot (Open vSwitch) @VERSION@"
  66             else:
  67                 sys.exit(0)
  68
  69         if len(args) != 1:
  70             sys.stderr.write("%s: exactly 1 non-option argument required "
  71                              "(use --help for help)\n" % argv0)
  72             sys.exit(1)
  73
  74         schemaToDot(args[0])
  75
  76     except Error, e:
  77         sys.stderr.write("%s: %s\n" % (argv0, e.msg))
  78         sys.exit(1)
  79
  80 # Local variables:
  81 # mode: python
  82 # End: