fix(queue): add --format=json and --limit flags to queue list #258
@@ -56,14 +56,21 @@ Usage:
|
|||||||
kugetsu queue [subcommand]
|
kugetsu queue [subcommand]
|
||||||
|
|
||||||
Subcommands:
|
Subcommands:
|
||||||
list Show pending tasks (default)
|
list [--limit=N] [--format=json] Show pending tasks (default: 10 items, text format)
|
||||||
stats Show queue statistics
|
stats Show queue statistics
|
||||||
clear Clear all queue items
|
clear Clear all queue items
|
||||||
enqueue <issue-ref> <message> Enqueue a task
|
enqueue <issue-ref> <message> Enqueue a task
|
||||||
help Show this help
|
help Show this help
|
||||||
|
|
||||||
|
Options for list:
|
||||||
|
--limit=N Number of items to return (default: 10)
|
||||||
|
--format=json Output in JSON format with stats
|
||||||
|
|
||||||
Examples:
|
Examples:
|
||||||
kugetsu queue list
|
kugetsu queue list
|
||||||
|
kugetsu queue list --limit=50
|
||||||
|
kugetsu queue list --format=json
|
||||||
|
kugetsu queue list --limit=20 --format=json
|
||||||
kugetsu queue stats
|
kugetsu queue stats
|
||||||
kugetsu queue clear
|
kugetsu queue clear
|
||||||
kugetsu queue enqueue github.com/shoko/kugetsu#14 "fix bug"
|
kugetsu queue enqueue github.com/shoko/kugetsu#14 "fix bug"
|
||||||
@@ -355,19 +362,25 @@ PYEOF
|
|||||||
|
|
||||||
get_pending_tasks() {
|
get_pending_tasks() {
|
||||||
local limit="${1:-10}"
|
local limit="${1:-10}"
|
||||||
|
local format="${2:-text}"
|
||||||
|
|
||||||
if [ ! -d "$QUEUE_ITEMS_DIR" ]; then
|
if [ ! -d "$QUEUE_ITEMS_DIR" ]; then
|
||||||
echo "[]"
|
if [ "$format" = "json" ]; then
|
||||||
|
echo '{"items": [], "total": 0, "pending": 0}'
|
||||||
|
else
|
||||||
|
echo "No pending tasks in queue."
|
||||||
|
fi
|
||||||
return
|
return
|
||||||
fi
|
fi
|
||||||
|
|
||||||
python3 -c "
|
python3 << PYEOF
|
||||||
import json
|
import json
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
queue_dir = os.environ.get('QUEUE_ITEMS_DIR', '')
|
queue_dir = os.environ.get('QUEUE_ITEMS_DIR', '')
|
||||||
limit = int(sys.argv[1]) if len(sys.argv) > 1 else 10
|
limit = int(sys.argv[1]) if len(sys.argv) > 1 else 10
|
||||||
|
format_type = sys.argv[2] if len(sys.argv) > 2 else 'text'
|
||||||
|
|
||||||
items = []
|
items = []
|
||||||
if os.path.isdir(queue_dir):
|
if os.path.isdir(queue_dir):
|
||||||
@@ -379,13 +392,27 @@ if os.path.isdir(queue_dir):
|
|||||||
data = json.load(f)
|
data = json.load(f)
|
||||||
if data.get('state') == 'pending':
|
if data.get('state') == 'pending':
|
||||||
items.append(data)
|
items.append(data)
|
||||||
if len(items) >= limit:
|
|
||||||
break
|
|
||||||
except:
|
except:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
print(json.dumps(items))
|
items.sort(key=lambda x: x.get('pending_since', ''))
|
||||||
" "$limit"
|
|
||||||
|
if format_type == 'json':
|
||||||
|
output = {
|
||||||
|
"items": items[:limit],
|
||||||
|
"total": len(items),
|
||||||
|
"pending": len(items)
|
||||||
|
}
|
||||||
|
print(json.dumps(output))
|
||||||
|
else:
|
||||||
|
if not items:
|
||||||
|
print("No pending tasks in queue.")
|
||||||
|
else:
|
||||||
|
print("Pending tasks:")
|
||||||
|
for t in items[:limit]:
|
||||||
|
msg = t.get('message', '')[:50]
|
||||||
|
print(f" {t.get('id')}: {t.get('issue_ref')} - {msg}...")
|
||||||
|
PYEOF
|
||||||
}
|
}
|
||||||
|
|
||||||
get_queue_stats() {
|
get_queue_stats() {
|
||||||
@@ -920,13 +947,24 @@ cmd_queue() {
|
|||||||
usage_queue
|
usage_queue
|
||||||
;;
|
;;
|
||||||
list)
|
list)
|
||||||
local pending_tasks=$(get_pending_tasks 10)
|
local limit="10"
|
||||||
if [ "$pending_tasks" = "[]" ]; then
|
local format="text"
|
||||||
echo "No pending tasks in queue."
|
|
||||||
else
|
while [ $# -gt 0 ]; do
|
||||||
echo "Pending tasks:"
|
case "$1" in
|
||||||
echo "$pending_tasks" | python3 -c "import sys, json; [print(f\" {t.get('id')}: {t.get('issue_ref')} - {t.get('message', '')[:50]}...\") for t in json.load(sys.stdin)]"
|
--limit=*)
|
||||||
fi
|
limit="${1#*=}"
|
||||||
|
;;
|
||||||
|
--format=json)
|
||||||
|
format="json"
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
shift
|
||||||
|
done
|
||||||
|
|
||||||
|
get_pending_tasks "$limit" "$format"
|
||||||
;;
|
;;
|
||||||
stats)
|
stats)
|
||||||
local stats=$(get_queue_stats)
|
local stats=$(get_queue_stats)
|
||||||
|
|||||||
Reference in New Issue
Block a user