Content tags

curl -X GET 'https://api.wolfia.com/v1/tags/' \
  -H "X-API-Key: wolfia-api-YOUR_KEY_HERE"

[
  {
    "id": "a1b2c3d4-e5f6-7890-abcd-ef1234567890",
    "name": "SOC 2",
    "description": "SOC 2 compliance questionnaires",
    "is_ai_generated": false,
    "status": "ACTIVE"
  },
  {
    "id": "b2c3d4e5-f6a7-8901-bcde-f12345678901",
    "name": "ISO 27001",
    "description": "ISO 27001 security questionnaires",
    "is_ai_generated": true,
    "status": "DRAFT"
  }
]

import os

import requests

API_KEY = os.environ['WOLFIA_API_KEY']
BASE_URL = 'https://api.wolfia.com/v1'


def upload_with_tags(file_path, company_name, tag_names):
    """Upload a questionnaire with matching tags applied."""
    headers = {'X-API-Key': API_KEY}

    # Fetch available tags
    tags_resp = requests.get(
        f'{BASE_URL}/tags/',
        headers=headers,
        params={'status': 'ACTIVE'},
        timeout=10,
    )
    tags_resp.raise_for_status()
    all_tags = tags_resp.json()

    # Match requested tag names (case-insensitive)
    tag_ids = [
        t['id'] for t in all_tags
        if t['name'].lower() in [n.lower() for n in tag_names]
    ]

    # Upload the questionnaire with tags
    with open(file_path, 'rb') as f:
        upload_resp = requests.post(
            f'{BASE_URL}/questionnaire/upload/file',
            headers=headers,
            files={'file': (os.path.basename(file_path), f, 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')},
            data={
                'company_name': company_name,
                'tag_ids': tag_ids,
            },
            timeout=60,
        )
    upload_resp.raise_for_status()
    return upload_resp.json()


if __name__ == '__main__':
    result = upload_with_tags('questionnaire.xlsx', 'Acme Corp', ['SOC 2', 'ISO 27001'])
    print(f"Uploaded: {result['questionnaire_id']}")