Managing Projects

import requests
import os

# API key authentication (Developer API)
api_key = os.environ.get("PROMPTGUARD_API_KEY")
base_url = "https://api.promptguard.co/api/v1/projects"

headers = {
    "X-API-Key": api_key,
    "Content-Type": "application/json"
}

# CREATE - Create a new project
def create_project(name, description=None, use_case="default", strictness_level="moderate"):
    payload = {
        "name": name,
        "description": description,
        "use_case": use_case,
        "strictness_level": strictness_level
    }

    response = requests.post(base_url, headers=headers, json=payload)

    if response.status_code == 201:
        project = response.json()
        print(f"Created project: {project['id']}")
        return project
    elif response.status_code == 400:
        error = response.json()
        print(f"Error: {error.get('detail', 'Invalid request')}")
    elif response.status_code == 401:
        print("Error: Authentication required. Please log in.")
    else:
        print(f"Error: HTTP {response.status_code}")
    return None

# READ - List all projects
def list_projects():
    response = requests.get(base_url, headers=headers)

    if response.status_code == 200:
        projects = response.json()
        print(f"Found {len(projects)} projects:")
        for project in projects:
            print(f"  - {project['name']} ({project['id']})")
        return projects
    else:
        print(f"Error: HTTP {response.status_code}")
        return []

# READ - Get single project
def get_project(project_id):
    url = f"{base_url}/{project_id}"
    response = requests.get(url, headers=headers)

    if response.status_code == 200:
        project = response.json()
        print(f"Project: {project['name']}")
        print(f"  Description: {project.get('description', 'N/A')}")
        print(f"  Use Case: {project.get('use_case', 'default')}")
        print(f"  Strictness: {project.get('strictness_level', 'moderate')}")
        return project
    elif response.status_code == 404:
        print(f"Error: Project {project_id} not found")
    else:
        print(f"Error: HTTP {response.status_code}")
    return None

# Note: Update endpoint is not available in Developer API
# To update projects, use the dashboard or delete and recreate

# DELETE - Delete project
def delete_project(project_id):
    url = f"{base_url}/{project_id}"
    response = requests.delete(url, headers=headers)

    if response.status_code == 200:
        print(f"Deleted project {project_id}")
        return True
    elif response.status_code == 404:
        print(f"Error: Project {project_id} not found")
    elif response.status_code == 400:
        error = response.json()
        if "last project" in error.get('detail', '').lower():
            print("Error: Cannot delete your last project. Create another project first.")
        else:
            print(f"Error: {error.get('detail', 'Cannot delete project')}")
    else:
        print(f"Error: HTTP {response.status_code}")
    return False

# Example usage
if __name__ == "__main__":
    # Create a new project
    project = create_project(
        name="Staging Environment",
        description="Pre-production testing",
        use_case="default",
        strictness_level="moderate"
    )

    if project:
        project_id = project['id']

        # Get project details
        get_project(project_id)

        # Note: Update is not available in Developer API
        # Use dashboard or delete and recreate if needed

        # List all projects
        list_projects()

        # Note: Uncomment to delete (be careful!)
        # delete_project(project_id)