Rename invirt_exception to InvirtException
[invirt/packages/invirt-base.git] / scripts / invirt-getconf
index d07b215..d56b06f 100755 (executable)
@@ -1,4 +1,4 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 
 """
 invirt-getconf loads an invirt configuration file (either the original YAML
@@ -16,12 +16,6 @@ Examples:
   invirt-getconf hosts.0.ip
 """
 
-from invirt import config
-from sys import argv, exit, stderr, stdout
-from optparse import OptionParser
-
-class invirt_exception(Exception): pass
-
 def main(argv):
     try:
         parser = OptionParser(usage = '%prog [options] key',
@@ -38,6 +32,16 @@ def main(argv):
             raise invirt_exception(__doc__.strip())
         elif args and args[0]:
             components = args[0].split('.')
+import argparse
+import sys
+import yaml
+
+import invirt
+
+
+class InvirtException(Exception):
+    pass
+
         else:
             components = []