218 lines
7.5 KiB
Python
218 lines
7.5 KiB
Python
"""Vikunja safe task updater — preserves existing fields when updating tasks.
|
|
|
|
Usage:
|
|
python vikunja_helper.py done 75 # Mark task #75 as done
|
|
python vikunja_helper.py done 71 77 78 # Mark multiple tasks done
|
|
python vikunja_helper.py undone 75 # Mark task #75 as not done
|
|
python vikunja_helper.py comment 75 "text" # Add comment to task #75
|
|
python vikunja_helper.py desc 75 "text" # Set description (appends if exists)
|
|
python vikunja_helper.py create "title" "desc" --labels Backend,Priority:High
|
|
python vikunja_helper.py create "title" "desc" --done --labels Frontend,Priority:Mid
|
|
python vikunja_helper.py label 75 Backend Priority:High # Add labels to task
|
|
python vikunja_helper.py list # List all tasks
|
|
python vikunja_helper.py list todo # List TODO only
|
|
python vikunja_helper.py list done # List DONE only
|
|
"""
|
|
|
|
import sys
|
|
import json
|
|
import urllib.request
|
|
import urllib.error
|
|
import io
|
|
|
|
# Fix Windows console encoding (cp949 → utf-8)
|
|
if sys.stdout.encoding != "utf-8":
|
|
sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding="utf-8", errors="replace")
|
|
|
|
# ============================================================
|
|
# ⚙️ CONFIGURATION — PROJECT_ID만 프로젝트별로 변경하세요
|
|
# ============================================================
|
|
API_BASE = "https://plan.variet.net/api/v1"
|
|
TOKEN = "tk_070f8e0b715e818bb7178c3815ed5389040eddca"
|
|
PROJECT_ID = 8 # gravity_control project
|
|
# ============================================================
|
|
|
|
HEADERS = {
|
|
"Authorization": f"Bearer {TOKEN}",
|
|
"Content-Type": "application/json",
|
|
}
|
|
|
|
# Label name → Vikunja label ID mapping
|
|
# Customize for your project's labels
|
|
LABEL_MAP = {
|
|
"Backend": 1, "Frontend": 2, "Engine": 3, "Infra": 4, "Test": 5,
|
|
"Priority:High": 6, "Priority:Mid": 7, "Priority:Low": 8,
|
|
"Agent": 17, "Tool": 18, "AI/LLM": 19,
|
|
}
|
|
|
|
|
|
def api_get(path: str):
|
|
req = urllib.request.Request(f"{API_BASE}{path}", headers=HEADERS)
|
|
with urllib.request.urlopen(req) as resp:
|
|
return json.loads(resp.read().decode("utf-8"))
|
|
|
|
|
|
def api_post(path: str, data: dict):
|
|
body = json.dumps(data).encode("utf-8")
|
|
req = urllib.request.Request(f"{API_BASE}{path}", data=body, headers=HEADERS, method="POST")
|
|
with urllib.request.urlopen(req) as resp:
|
|
return json.loads(resp.read().decode("utf-8"))
|
|
|
|
|
|
def api_put(path: str, data: dict):
|
|
body = json.dumps(data).encode("utf-8")
|
|
req = urllib.request.Request(f"{API_BASE}{path}", data=body, headers=HEADERS, method="PUT")
|
|
with urllib.request.urlopen(req) as resp:
|
|
return json.loads(resp.read().decode("utf-8"))
|
|
|
|
|
|
def get_task(task_id: int) -> dict:
|
|
return api_get(f"/tasks/{task_id}")
|
|
|
|
|
|
def safe_update_task(task_id: int, updates: dict) -> dict:
|
|
task = get_task(task_id)
|
|
safe_body = {
|
|
"title": task.get("title", ""),
|
|
"description": task.get("description", ""),
|
|
"priority": task.get("priority", 0),
|
|
"done": task.get("done", False),
|
|
}
|
|
safe_body.update(updates)
|
|
return api_post(f"/tasks/{task_id}", safe_body)
|
|
|
|
|
|
def mark_done(task_ids: list):
|
|
for tid in task_ids:
|
|
result = safe_update_task(tid, {"done": True})
|
|
title = result.get("title", "?")
|
|
print(f" ✅ #{tid} → done=True [{title}]")
|
|
|
|
|
|
def mark_undone(task_ids: list):
|
|
for tid in task_ids:
|
|
result = safe_update_task(tid, {"done": False})
|
|
title = result.get("title", "?")
|
|
print(f" ⬜ #{tid} → done=False [{title}]")
|
|
|
|
|
|
def add_comment(task_id: int, comment: str):
|
|
result = api_put(f"/tasks/{task_id}/comments", {"comment": comment})
|
|
print(f" 💬 #{task_id} comment added (id={result.get('id', '?')})")
|
|
|
|
|
|
def set_description(task_id: int, desc: str, append: bool = True):
|
|
task = get_task(task_id)
|
|
existing = task.get("description", "") or ""
|
|
if append and existing:
|
|
new_desc = existing.rstrip() + "\n\n" + desc
|
|
else:
|
|
new_desc = desc
|
|
result = safe_update_task(task_id, {"description": new_desc})
|
|
print(f" 📝 #{task_id} description updated [{result.get('title', '?')}]")
|
|
|
|
|
|
def list_tasks(filter_: str = "all"):
|
|
all_tasks = []
|
|
page = 1
|
|
while True:
|
|
batch = api_get(f"/projects/{PROJECT_ID}/tasks?per_page=50&page={page}")
|
|
if not batch:
|
|
break
|
|
all_tasks.extend(batch)
|
|
if len(batch) < 50:
|
|
break
|
|
page += 1
|
|
|
|
if filter_ == "todo":
|
|
all_tasks = [t for t in all_tasks if not t["done"]]
|
|
elif filter_ == "done":
|
|
all_tasks = [t for t in all_tasks if t["done"]]
|
|
|
|
all_tasks.sort(key=lambda t: t["id"])
|
|
for t in all_tasks:
|
|
status = "✅" if t["done"] else "⬜"
|
|
desc = (t.get("description") or "")[:50].replace("\n", " ")
|
|
labels = ", ".join(l["title"] for l in (t.get("labels") or []))
|
|
print(f" {status} #{t['id']:3d} {t['title'][:40]:<40} [{labels}] {desc}")
|
|
print(f"\n Total: {len(all_tasks)} tasks")
|
|
|
|
|
|
def add_labels(task_id: int, label_names: list):
|
|
for name in label_names:
|
|
label_id = LABEL_MAP.get(name)
|
|
if not label_id:
|
|
print(f" ⚠️ Unknown label '{name}'. Valid: {', '.join(LABEL_MAP.keys())}")
|
|
continue
|
|
try:
|
|
api_put(f"/tasks/{task_id}/labels", {"label_id": label_id})
|
|
print(f" 🏷️ #{task_id} + {name} (id={label_id})")
|
|
except Exception as e:
|
|
if "already" in str(e).lower() or "409" in str(e):
|
|
print(f" 🏷️ #{task_id} already has {name}")
|
|
else:
|
|
print(f" ⚠️ #{task_id} label {name} failed: {e}")
|
|
|
|
|
|
def create_task(title: str, description: str = "", done: bool = False, labels: list = None):
|
|
payload = {"title": title, "description": description}
|
|
result = api_put(f"/projects/{PROJECT_ID}/tasks", payload)
|
|
task_id = result["id"]
|
|
print(f" ✨ #{task_id} created: {result.get('title', '?')}")
|
|
|
|
if labels:
|
|
add_labels(task_id, labels)
|
|
|
|
if done:
|
|
result = safe_update_task(task_id, {"done": True})
|
|
print(f" ✅ #{task_id} → done=True")
|
|
|
|
return result
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) < 2:
|
|
print(__doc__)
|
|
return
|
|
|
|
cmd = sys.argv[1].lower()
|
|
|
|
if cmd == "done":
|
|
ids = [int(x) for x in sys.argv[2:]]
|
|
mark_done(ids)
|
|
elif cmd == "undone":
|
|
ids = [int(x) for x in sys.argv[2:]]
|
|
mark_undone(ids)
|
|
elif cmd == "comment":
|
|
add_comment(int(sys.argv[2]), sys.argv[3])
|
|
elif cmd == "desc":
|
|
set_description(int(sys.argv[2]), sys.argv[3])
|
|
elif cmd == "list":
|
|
f = sys.argv[2] if len(sys.argv) > 2 else "all"
|
|
list_tasks(f)
|
|
elif cmd == "label":
|
|
if len(sys.argv) < 4:
|
|
print("Usage: vikunja_helper.py label TASK_ID Label1 Label2 ...")
|
|
return
|
|
add_labels(int(sys.argv[2]), sys.argv[3:])
|
|
elif cmd == "create":
|
|
title = sys.argv[2] if len(sys.argv) > 2 else ""
|
|
desc = sys.argv[3] if len(sys.argv) > 3 and not sys.argv[3].startswith("--") else ""
|
|
is_done = "--done" in sys.argv
|
|
labels = None
|
|
for i, arg in enumerate(sys.argv):
|
|
if arg == "--labels" and i + 1 < len(sys.argv):
|
|
labels = sys.argv[i + 1].split(",")
|
|
break
|
|
if not title:
|
|
print("Error: title is required")
|
|
return
|
|
create_task(title, desc, done=is_done, labels=labels)
|
|
else:
|
|
print(f"Unknown command: {cmd}")
|
|
print(__doc__)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|