123 lines
4.3 KiB
Python
123 lines
4.3 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Upload one or more files as attachments to a Jira Cloud issue via REST API v3.
|
|
|
|
Config: from jira_upload.env (current dir or script dir) or env vars.
|
|
Usage:
|
|
python upload_attachment.py --issue ISSUE_KEY --file path1 [--file path2 ...]
|
|
python upload_attachment.py --config /path/to/jira_upload.env --issue KEY --file path1
|
|
"""
|
|
|
|
import argparse
|
|
import base64
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
from typing import Dict, Optional
|
|
|
|
try:
|
|
import requests
|
|
except ImportError:
|
|
print("ERROR: 'requests' is required. Run: pip install requests", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
CONFIG_FILENAME = "jira_upload.env"
|
|
|
|
|
|
def _find_config_file(explicit_path: Optional[str], script_dir: Path) -> Optional[Path]:
|
|
if explicit_path:
|
|
p = Path(explicit_path)
|
|
return p if p.exists() and p.is_file() else None
|
|
cwd_file = Path.cwd() / CONFIG_FILENAME
|
|
if cwd_file.exists():
|
|
return cwd_file
|
|
script_dir_file = script_dir / CONFIG_FILENAME
|
|
if script_dir_file.exists():
|
|
return script_dir_file
|
|
return None
|
|
|
|
|
|
def load_config(config_path: Optional[str]) -> Dict[str, str]:
|
|
"""Load KEY=VALUE from jira_upload.env. Env vars override file values."""
|
|
script_dir = Path(__file__).resolve().parent
|
|
path = _find_config_file(config_path, script_dir)
|
|
out = {}
|
|
if path:
|
|
with open(path, "r", encoding="utf-8") as f:
|
|
for line in f:
|
|
line = line.strip()
|
|
if not line or line.startswith("#"):
|
|
continue
|
|
if "=" in line:
|
|
k, _, v = line.partition("=")
|
|
out[k.strip()] = v.strip().strip('"').strip("'")
|
|
for key in ("JIRA_BASE_URL", "JIRA_EMAIL", "JIRA_API_TOKEN"):
|
|
if key in os.environ:
|
|
out[key] = os.environ[key]
|
|
return out
|
|
|
|
|
|
def main() -> int:
|
|
parser = argparse.ArgumentParser(description="Upload files as attachments to a Jira issue")
|
|
parser.add_argument("--issue", required=True, help="Jira issue key (e.g. G3SF-123)")
|
|
parser.add_argument("--file", action="append", required=True, dest="files", help="Path to file to upload (can be repeated)")
|
|
parser.add_argument("--config", default=None, help=f"Path to config file (default: {CONFIG_FILENAME} in cwd or script dir)")
|
|
args = parser.parse_args()
|
|
|
|
cfg = load_config(args.config)
|
|
base_url = (cfg.get("JIRA_BASE_URL") or "").rstrip("/")
|
|
email = cfg.get("JIRA_EMAIL")
|
|
token = cfg.get("JIRA_API_TOKEN")
|
|
|
|
if not base_url or not email or not token:
|
|
print("ERROR: Set JIRA_BASE_URL, JIRA_EMAIL, JIRA_API_TOKEN in config file or environment.", file=sys.stderr)
|
|
print(f" Config file: {CONFIG_FILENAME} (in current dir or script dir), or use --config PATH", file=sys.stderr)
|
|
return 1
|
|
|
|
issue_key = args.issue.strip()
|
|
files_to_upload = []
|
|
for p in args.files:
|
|
path = Path(p)
|
|
if not path.exists():
|
|
print(f"ERROR: File not found: {path}", file=sys.stderr)
|
|
return 1
|
|
if not path.is_file():
|
|
print(f"ERROR: Not a file: {path}", file=sys.stderr)
|
|
return 1
|
|
files_to_upload.append(path)
|
|
|
|
url = f"{base_url}/rest/api/3/issue/{issue_key}/attachments"
|
|
auth_str = base64.b64encode(f"{email}:{token}".encode()).decode()
|
|
headers = {
|
|
"Authorization": f"Basic {auth_str}",
|
|
"X-Atlassian-Token": "no-check",
|
|
}
|
|
|
|
for path in files_to_upload:
|
|
with open(path, "rb") as f:
|
|
files = {"file": (path.name, f)}
|
|
try:
|
|
resp = requests.post(url, headers=headers, files=files, timeout=60)
|
|
except requests.RequestException as e:
|
|
print(f"ERROR: Request failed for {path.name}: {e}", file=sys.stderr)
|
|
return 1
|
|
|
|
if resp.status_code not in (200, 201):
|
|
print(f"ERROR: Upload failed for {path.name}: HTTP {resp.status_code}", file=sys.stderr)
|
|
if resp.text:
|
|
print(resp.text[:500], file=sys.stderr)
|
|
return 1
|
|
|
|
data = resp.json()
|
|
if isinstance(data, list) and data:
|
|
for att in data:
|
|
print(f"OK: {att.get('filename', path.name)} (id={att.get('id')}, size={att.get('size', '?')})")
|
|
else:
|
|
print(f"OK: {path.name}")
|
|
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|