Init Repo

2026-02-28 03:22:04 +08:00
commit de59b57ee7
883 changed files with 156857 additions and 0 deletions
@@ -0,0 +1,112 @@
+"""
+Test script for project settings endpoints
+"""
+
+import requests
+import json
+
+BASE_URL = "http://localhost:8000"
+
+def test_project_settings():
+    """Test project settings endpoints"""
+    
+    # First, login to get a token
+    print("1. Logging in...")
+    login_response = requests.post(
+        f"{BASE_URL}/auth/login",
+        data={
+            "username": "admin@example.com",
+            "password": "admin123"
+        }
+    )
+    
+    if login_response.status_code != 200:
+        print(f"❌ Login failed: {login_response.status_code}")
+        print(login_response.text)
+        return
+    
+    token = login_response.json()["access_token"]
+    headers = {"Authorization": f"Bearer {token}"}
+    print("✓ Login successful")
+    
+    # Get projects
+    print("\n2. Getting projects...")
+    projects_response = requests.get(f"{BASE_URL}/projects/", headers=headers)
+    
+    if projects_response.status_code != 200:
+        print(f"❌ Failed to get projects: {projects_response.status_code}")
+        return
+    
+    projects = projects_response.json()
+    if not projects:
+        print("❌ No projects found")
+        return
+    
+    project_id = projects[0]["id"]
+    print(f"✓ Found project: {projects[0]['name']} (ID: {project_id})")
+    
+    # Get project settings
+    print("\n3. Getting project settings...")
+    settings_response = requests.get(
+        f"{BASE_URL}/projects/{project_id}/settings",
+        headers=headers
+    )
+    
+    if settings_response.status_code != 200:
+        print(f"❌ Failed to get settings: {settings_response.status_code}")
+        print(settings_response.text)
+        return
+    
+    settings = settings_response.json()
+    print("✓ Current settings:")
+    print(json.dumps(settings, indent=2))
+    
+    # Update project settings
+    print("\n4. Updating project settings...")
+    new_settings = {
+        "upload_data_location": "/mnt/projects/test-project/uploads",
+        "asset_task_templates": {
+            "characters": ["modeling", "surfacing", "rigging"],
+            "props": ["modeling", "surfacing"],
+            "sets": ["modeling", "surfacing"],
+            "vehicles": ["modeling", "surfacing", "rigging"]
+        },
+        "shot_task_templates": ["layout", "animation", "lighting", "compositing"]
+    }
+    
+    update_response = requests.put(
+        f"{BASE_URL}/projects/{project_id}/settings",
+        headers=headers,
+        json=new_settings
+    )
+    
+    if update_response.status_code != 200:
+        print(f"❌ Failed to update settings: {update_response.status_code}")
+        print(update_response.text)
+        return
+    
+    updated_settings = update_response.json()
+    print("✓ Settings updated:")
+    print(json.dumps(updated_settings, indent=2))
+    
+    # Verify the update
+    print("\n5. Verifying update...")
+    verify_response = requests.get(
+        f"{BASE_URL}/projects/{project_id}/settings",
+        headers=headers
+    )
+    
+    if verify_response.status_code != 200:
+        print(f"❌ Failed to verify: {verify_response.status_code}")
+        return
+    
+    verified_settings = verify_response.json()
+    print("✓ Verified settings:")
+    print(json.dumps(verified_settings, indent=2))
+    
+    print("\n✅ All tests passed!")
+
+if __name__ == "__main__":
+    print("Testing Project Settings API")
+    print("=" * 50)
+    test_project_settings()