codesphere-python/examples/create_workspace_with_landscape.py at 87e53f6a474cf3b3b2258f50e954e00f995642b4 · Datata1/codesphere-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
import asyncio
import time

from codesphere import CodesphereSDK
from codesphere.resources.workspace import WorkspaceCreate
from codesphere.resources.workspace.landscape import (
    PipelineStage,
    PipelineState,
    ProfileBuilder,
)
from codesphere.resources.workspace.logs import LogStage

TEAM_ID = 123


async def main():
    async with CodesphereSDK() as sdk:
        plans = await sdk.metadata.list_plans()
        plan = next((p for p in plans if p.title == "Micro" and not p.deprecated), None)
        if not plan:
            raise ValueError("Micro plan not found")

        workspace_name = f"pipeline-demo-{int(time.time())}"

        print(f"Creating workspace '{workspace_name}'...")
        workspace = await sdk.workspaces.create(
            WorkspaceCreate(plan_id=plan.id, team_id=TEAM_ID, name=workspace_name)
        )
        print(f"✓ Workspace created (ID: {workspace.id})")

        print("Waiting for workspace to start...")
        await workspace.wait_until_running(timeout=300.0, poll_interval=5.0)
        print("✓ Workspace is running\n")

        profile = (
            ProfileBuilder()
            .prepare()
            .add_step("echo 'Installing dependencies...' && sleep 2")
            .add_step("echo 'Setup complete!' && sleep 1")
            .done()
            .add_reactive_service("web")
            .plan(plan.id)
            .add_step(
                'for i in $(seq 1 20); do echo "[$i] Processing request..."; sleep 1; done'
            )
            .add_port(3000, public=True)
            .add_path("/", port=3000)
            .replicas(1)
            .done()
            .build()
        )

        print("Deploying landscape profile...")
        await workspace.landscape.save_profile("production", profile)
        await workspace.landscape.deploy(profile="production")
        print("✓ Profile deployed\n")

        print("--- Prepare Stage ---")
        await workspace.landscape.start_stage(
            PipelineStage.PREPARE, profile="production"
        )
        prepare_status = await workspace.landscape.wait_for_stage(
            PipelineStage.PREPARE, timeout=60.0
        )

        for status in prepare_status:
            icon = "✓" if status.state == PipelineState.SUCCESS else "✗"
            print(f"{icon} {status.server}: {status.state.value}")

        print("\nPrepare logs:")
        for step in range(len(prepare_status[0].steps)):
            logs = await workspace.logs.collect(
                stage=LogStage.PREPARE, step=step, timeout=5.0
            )
            for entry in logs:
                if entry.get_text():
                    print(f"  {entry.get_text().strip()}")

        print("\n--- Run Stage ---")
        await workspace.landscape.start_stage(PipelineStage.RUN, profile="production")
        print("Started run stage\n")

        print("Streaming logs from 'web' service (using context manager):")
        count = 0
        async with workspace.logs.open_server_stream(step=0, server="web") as stream:
            async for entry in stream:
                if entry.get_text():
                    print(f"  {entry.get_text().strip()}")
                    count += 1

        print(f"\n✓ Stream ended ({count} log entries)")


if __name__ == "__main__":
    asyncio.run(main())