This commit is contained in:
who
2007-11-13 10:13:37 +01:00
2 changed files with 71 additions and 1 deletions

View File

@ -3,6 +3,12 @@ import threading
import cmd
import sys
import Entry
class LocalMyCmdException(Exception):
def __init__(self, msg):
self.msg = msg
class MyCmd(cmd.Cmd):
@classmethod
@ -19,6 +25,71 @@ class MyCmd(cmd.Cmd):
def do_hello(self, l):
self.stdout.write("DO: %s\n" % l)
def do_list(self, l):
try:
parts = l.split(' ')
if len(parts) != 1:
raise LocalMyCmdException("illegal number of arguments")
x = parts[0]
if x == 'all':
for entry in MyCmd.entries.values():
self.stdout.write(str(entry) + "\n")
elif MyCmd.entries.has_key(x):
self.stdout.write(str(MyCmd.entries[x]) + "\n")
else:
raise LocalMyCmdException("unknown dynid")
except LocalMyCmdException, e:
self.stdout.write("Failure: %s\n" % str(e.msg))
def do_add(self, l):
try:
parts = l.split(' ')
if len(parts) != 3:
raise LocalMyCmdException("illegal number of arguments")
(dynid, name, sharedSecret) = parts
if MyCmd.entries.has_key(dynid):
raise LocalMyCmdException("duplicate dynid")
for entry in MyCmd.entries.values():
if entry.name == name:
raise LocalMyCmdException("duplicate name")
newEntry = Entry.Entry(dynid, sharedSecret, name)
MyCmd.entries[dynid] = newEntry
self.stdout.write("Done\n")
except LocalMyCmdException, e:
self.stdout.write("Failure: %s\n" % str(e.msg))
def do_edit(self, l):
try:
parts = l.split(' ')
if len(parts) != 3:
raise LocalMyCmdException("illegal number of arguments")
(dynid, attr, value) = parts
if not MyCmd.entries.has_key(dynid):
raise LocalMyCmdException("unknown dynid")
if attr == 'sharedSecret':
MyCmd.entries[dynid].sharedSecret = value
elif attr == 'name':
MyCmd.entries[dynid].name = value
else:
raise LocalMyCmdException("unknown attribute to change")
self.stdout.write("Done\n")
except LocalMyCmdException, e:
self.stdout.write("Failure: %s\n" % str(e.msg))
def do_delete(self, l):
try:
parts = l.split(' ')
if len(parts) != 1:
raise LocalMyCmdException("illegal number of arguments")
x = parts[0]
if not MyCmd.entries.has_key(x):
raise LocalMyCmdException("unknown dynid")
del MyCmd.entries[x]
self.stdout.write("Done\n")
except LocalMyCmdException, e:
self.stdout.write("Failure: %s\n" % str(e.msg))
def do_quit(self, l):
self.stdout.write("Bye\n")
return True

View File

@ -17,7 +17,6 @@ import Event
MSG_TIME_CORRIDOR = 5
EVENT_LIFE_TIME = 10
NULL_ADDRESS = '0.0.0.0'
NAME = 'serve'
TTL = 120
NAMESERVER = '127.0.0.1'
TSIGKEY = { "monitoring." : "HYHN8l/dg1+q6QLOURp2qw==" }