浏览代码

caqmadm: Add command line count to prompt.

Ask Solem 15 年之前
父节点
当前提交
1ec30a71cb
共有 1 个文件被更改,包括 7 次插入2 次删除
  1. 7 2
      celery/bin/camqadm.py

+ 7 - 2
celery/bin/camqadm.py

@@ -16,6 +16,7 @@ import shlex
 import pprint
 import readline
 import optparse
+from itertools import count
 
 from amqplib import client_0_8 as amqp
 
@@ -157,10 +158,10 @@ class AMQShell(cmd.Cmd):
     """
     conn = None
     chan = None
-    prompt = "--> "
+    prompt_fmt = "%d> "
     identchars = cmd.IDENTCHARS = "."
     needs_reconnect = False
-
+    counter = count(1).next
 
     builtins = {"exit": "do_exit",
                 "EOF": "do_exit",
@@ -317,6 +318,10 @@ class AMQShell(cmd.Cmd):
         self.chan = self.conn.create_backend().channel
         self.needs_reconnect = False
 
+    @property
+    def prompt(self):
+        return self.prompt_fmt % self.counter()
+
 
 class AMQPAdmin(object):
     """The celery ``camqadm`` utility."""