Refactor

svn path=/myrpki.rototill/setup.py; revision=3015

1 files changed, 76 insertions, 0 deletions

diff --git a/rpkid/rpki/cli.py b/rpkid/rpki/cli.py
new file mode 100644
index 00000000..4960c387
--- /dev/null
+++ b/rpkid/rpki/cli.py
@@ -0,0 +1,76 @@
+"""
+Customizations of Python cmd module.
+
+$Id$
+
+Copyright (C) 2010  Internet Systems Consortium ("ISC")
+
+Permission to use, copy, modify, and distribute this software for any
+purpose with or without fee is hereby granted, provided that the above
+copyright notice and this permission notice appear in all copies.
+
+THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
+REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
+AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
+INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
+LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
+OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
+PERFORMANCE OF THIS SOFTWARE.
+"""
+
+import cmd, glob
+
+try:
+  import readline
+  have_readline = True
+except ImportError:
+  have_readline = False
+
+class Cmd(cmd.Cmd):
+
+  identchars = cmd.IDENTCHARS + "/-."
+  
+  histfile = None
+
+  def __init__(self, argv = None):
+    cmd.Cmd.__init__(self)
+    if argv:
+      self.onecmd(" ".join(argv))
+    else:      
+      self.cmdloop_with_history()
+
+  def do_EOF(self, arg):
+    print
+    return True
+
+  def do_exit(self, arg):
+    return True
+
+  do_quit = do_exit
+
+  def emptyline(self):
+    pass
+
+  def filename_complete(self, text, line, begidx, endidx):
+    return glob.glob(text + "*")
+
+  if have_readline:
+
+    def cmdloop_with_history(self):
+      old_completer_delims = readline.get_completer_delims()
+      if self.histfile is not None:
+        try:
+          readline.read_history_file(self.histfile)
+        except IOError:
+          pass
+      try:
+        readline.set_completer_delims("".join(set(old_completer_delims) - set(self.identchars)))
+        self.cmdloop()
+      finally:
+        if self.histfile is not None and readline.get_current_history_length():
+          readline.write_history_file(self.histfile)
+        readline.set_completer_delims(old_completer_delims)
+
+  else:
+
+    cmdloop_with_history = cmd.Cmd.cmdloop