teszt állományok áthelyezése és szelektálása

tests/archive/create_vehicle_for_org.py.old

@@ -0,0 +1,98 @@
+#!/usr/bin/env python3
+"""
+Script to create a vehicle for Organization 34 and Branch 18fcd55f...
+"""
+import asyncio
+import aiohttp
+import sys
+import os
+sys.path.insert(0, '/app')
+
+from app.db.session import AsyncSessionLocal
+from app.models.identity import User
+from app.core.security import create_tokens
+from sqlalchemy import select
+
+async def get_token_for_user(user_id: int):
+    """Generate JWT token for given user."""
+    async with AsyncSessionLocal() as db:
+        result = await db.execute(select(User).where(User.id == user_id))
+        user = result.scalar_one_or_none()
+        if not user:
+            raise ValueError(f"User {user_id} not found")
+        
+        token_payload = {
+            "sub": str(user.id),
+            "role": user.role.value if hasattr(user.role, 'value') else user.role,
+            "rank": 10,
+            "scope_level": user.scope_level.value if hasattr(user.scope_level, 'value') else (user.scope_level or "individual"),
+            "scope_id": str(user.scope_id) if user.scope_id else str(user.id)
+        }
+        access_token, refresh_token = create_tokens(data=token_payload)
+        return access_token
+
+async def create_vehicle(token: str):
+    """POST /api/v1/assets/vehicles with test data."""
+    url = "http://sf_api:8000/api/v1/assets/vehicles"
+    headers = {
+        "Authorization": f"Bearer {token}",
+        "Content-Type": "application/json"
+    }
+    # Branch ID: 18fcd55f... (need full UUID). Let's find it from database.
+    async with AsyncSessionLocal() as db:
+        from app.models.marketplace.organization import Branch
+        result = await db.execute(select(Branch).where(Branch.organization_id == 34))
+        branches = result.scalars().all()
+        for b in branches:
+            print(f"Branch: {b.id}, name: {b.name}")
+            # Use the first branch with ID starting with 18fcd55f
+            if str(b.id).startswith("18fcd55f"):
+                branch_id = b.id
+                break
+        else:
+            # fallback to first branch
+            if branches:
+                branch_id = branches[0].id
+            else:
+                raise ValueError("No branch found for organization 34")
+    
+    payload = {
+        "license_plate": "ABC-123",
+        "vin": "1HGCM82633A123456",
+        "brand": "Toyota",
+        "model": "Corolla",
+        "vehicle_class": "Passenger Car",
+        "fuel_type": "Petrol",
+        "organization_id": 34,
+        # branch_id is not in schema? Actually branch_id is not in AssetCreate.
+        # The branch assignment likely happens via organization_id and garage logic.
+        # We'll rely on the service to assign to the correct branch.
+    }
+    
+    async with aiohttp.ClientSession() as session:
+        async with session.post(url, headers=headers, json=payload) as resp:
+            print(f"Status: {resp.status}")
+            response_text = await resp.text()
+            print(f"Response: {response_text}")
+            if resp.status == 201:
+                data = await resp.json()
+                print(f"Vehicle created successfully: ID={data.get('id')}")
+                return data
+            else:
+                raise Exception(f"Failed to create vehicle: {resp.status} {response_text}")
+
+async def main():
+    try:
+        token = await get_token_for_user(28)
+        print(f"Token obtained: {token[:30]}...")
+        vehicle = await create_vehicle(token)
+        print("Vehicle creation successful.")
+        print(f"Vehicle ID: {vehicle['id']}")
+        print(f"Status: {vehicle.get('status')}")
+        print(f"Data Enrichment Status: {vehicle.get('data_status')}")
+    except Exception as e:
+        print(f"Error: {e}")
+        sys.exit(1)
+
+if __name__ == "__main__":
+    asyncio.run(main())