initial commit

main.py

@@ -0,0 +1,75 @@
+import typer
+from pydantic import BaseModel
+from stringcase import camelcase
+from specklepy.transports.memory import MemoryTransport
+from specklepy.transports.server import ServerTransport
+from specklepy.api.operations import receive
+from specklepy.api.client import SpeckleClient
+import random
+
+from flatten import flatten_base
+from make_comment import make_comment
+
+
+class SpeckleProjectData(BaseModel):
+    """Values of the project / model that triggered the run of this function."""
+
+    project_id: str
+    model_id: str
+    version_id: str
+    speckle_server_url: str
+
+    class Config:
+        alias_generator = camelcase
+
+
+class FunctionInputs(BaseModel):
+    """
+    These are function author defined values, automate will make sure to supply them.
+    """
+
+    comment_text: str
+
+    class Config:
+        alias_generator = camelcase
+
+
+def main(speckle_project_data: str, function_inputs: str, speckle_token: str):
+    project_data = SpeckleProjectData.parse_raw(speckle_project_data)
+    inputs = FunctionInputs.parse_raw(function_inputs)
+
+    client = SpeckleClient(project_data.speckle_server_url, use_ssl=False)
+    client.authenticate_with_token(speckle_token)
+    commit = client.commit.get(project_data.project_id, project_data.version_id)
+    branch = client.branch.get(project_data.project_id, project_data.model_id, 1)
+
+    memory_transport = MemoryTransport()
+    server_transport = ServerTransport(project_data.project_id, client)
+    base = receive(commit.referencedObject, server_transport, memory_transport)
+
+    random_beam = random.choice(
+        [b for b in flatten_base(base) if b.speckle_type == "IFCBEAM"]
+    )
+
+    make_comment(
+        client,
+        project_data.project_id,
+        branch.id,
+        project_data.version_id,
+        inputs.comment_text,
+        random_beam.id,
+    )
+
+    print(
+        "Ran function with",
+        f"{speckle_project_data} {function_inputs}",
+    )
+
+
+if __name__ == "__main__":
+    # main(
+    #     '{"projectId":"bbb3aba8d4", "modelId":"automateTest", "versionId": "d37ee808db", "speckleServerUrl": "http://hyperion:3000" }',
+    #     '{"commentText": "automate made me to do this"}',
+    #     "c3e6536e570a94e5d84590c51b29198b26dce89439",
+    # )
+    typer.run(main)