onecmd()
次の簡単な例に示すように、メソッドをオーバーライドできる可能性があります。このonecmd()
メソッドは基本的に元のメソッドのコピーですが、引数を関数に渡す前にcmd.py
への呼び出しを追加します。urlparse.parse_qs()
import cmd
import urlparse
class myCmd(cmd.Cmd):
def onecmd(self, line):
"""Mostly ripped from Python's cmd.py"""
cmd, arg, line = self.parseline(line)
arg = urlparse.parse_qs(arg) # <- added line
if not line:
return self.emptyline()
if cmd is None:
return self.default(line)
self.lastcmd = line
if cmd == '':
return self.default(line)
else:
try:
func = getattr(self, 'do_' + cmd)
except AttributeError:
return self.default(line)
return func(arg)
def do_foo(self, arg)
print arg
my_cmd = myCmd()
my_cmd.cmdloop()
サンプル出力:
(Cmd) foo
{}
(Cmd) foo a b c
{}
(Cmd) foo a=b&c=d
{'a': ['b'], 'c': ['d']}
これはあなたが達成しようとしていることですか?
cmd.Cmd
クラスデコレータを使用してサブクラスを変更し、基本的do_*
にそのクラスのすべてのメソッドにデコレータ関数を適用する別の潜在的なソリューションを次に示し
ます。
import cmd
import urlparse
import types
# function decorator to add parse_qs to individual functions
def parse_qs_f(f):
def f2(self, arg):
return f(self, urlparse.parse_qs(arg))
return f2
# class decorator to iterate over all attributes of a class and apply
# the parse_qs_f decorator to all do_* methods
def parse_qs(cls):
for attr_name in dir(cls):
attr = getattr(cls, attr_name)
if attr_name.startswith('do_') and type(attr) == types.MethodType:
setattr(cls, attr_name, parse_qs_f(attr))
return cls
@parse_qs
class myCmd(cmd.Cmd):
def do_foo(self, args):
print args
my_cmd = myCmd()
my_cmd.cmdloop()
私はすぐにこれをまとめて、意図したとおりに機能しているように見えますが、落とし穴やこのソリューションをどのように改善できるかについての提案を受け入れています。