This is a fork of:
https://github.com/roglew/puppy
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
63 lines
2.0 KiB
63 lines
2.0 KiB
7 years ago
|
from ..util import confirm
|
||
|
|
||
|
def tag_cmd(client, args):
|
||
|
if len(args) == 0:
|
||
|
raise CommandError("Usage: tag <tag> [reqid1] [reqid2] ...")
|
||
|
if not args[0]:
|
||
|
raise CommandError("Tag cannot be empty")
|
||
|
tag = args[0]
|
||
|
reqids = []
|
||
|
if len(args) > 1:
|
||
|
for reqid in args[1:]:
|
||
|
client.add_tag(reqid, tag)
|
||
|
else:
|
||
|
icr = client.in_context_requests(headers_only=True)
|
||
|
cnt = confirm("You are about to tag {} requests with \"{}\". Continue?".format(len(icr), tag))
|
||
|
if not cnt:
|
||
|
return
|
||
|
for reqh in icr:
|
||
|
reqid = client.prefixed_reqid(reqh)
|
||
|
client.remove_tag(reqid, tag)
|
||
|
|
||
|
def untag_cmd(client, args):
|
||
|
if len(args) == 0:
|
||
|
raise CommandError("Usage: untag <tag> [reqid1] [reqid2] ...")
|
||
|
if not args[0]:
|
||
|
raise CommandError("Tag cannot be empty")
|
||
|
tag = args[0]
|
||
|
reqids = []
|
||
|
if len(args) > 0:
|
||
|
for reqid in args[1:]:
|
||
|
client.remove_tag(reqid, tag)
|
||
|
else:
|
||
|
icr = client.in_context_requests(headers_only=True)
|
||
|
cnt = confirm("You are about to remove the \"{}\" tag from {} requests. Continue?".format(tag, len(icr)))
|
||
|
if not cnt:
|
||
|
return
|
||
|
for reqh in icr:
|
||
|
reqid = client.prefixed_reqid(reqh)
|
||
|
client.add_tag(reqid, tag)
|
||
|
|
||
|
def clrtag_cmd(client, args):
|
||
|
if len(args) == 0:
|
||
|
raise CommandError("Usage: clrtag [reqid1] [reqid2] ...")
|
||
|
reqids = []
|
||
|
if len(args) > 0:
|
||
|
for reqid in args:
|
||
|
client.clear_tag(reqid)
|
||
|
else:
|
||
|
icr = client.in_context_requests(headers_only=True)
|
||
|
cnt = confirm("You are about to clear ALL TAGS from {} requests. Continue?".format(len(icr)))
|
||
|
if not cnt:
|
||
|
return
|
||
|
for reqh in icr:
|
||
|
reqid = client.prefixed_reqid(reqh)
|
||
|
client.clear_tag(reqid)
|
||
|
|
||
|
def load_cmds(cmd):
|
||
|
cmd.set_cmds({
|
||
|
'clrtag': (clrtag_cmd, None),
|
||
|
'untag': (untag_cmd, None),
|
||
|
'tag': (tag_cmd, None),
|
||
|
})
|