Update utils.py

utils.py

@@ -84,3 +84,61 @@ def updateStreamAnalysisFast(client, new_data, stream_id, branch_name, geometryG
     print("commit created")
     if return_original:
         return objects_raw  # as back-up
+
+        
+def getSpeckleStream(stream_id,
+                     branch_name,
+                     client,
+                     commit_id=""
+                     ):
+    """
+    Retrieves data from a specific branch of a speckle stream.
+    Args:
+        stream_id (str): The ID of the speckle stream.
+        branch_name (str): The name of the branch within the speckle stream.
+        client (specklepy.api.client.Client, optional): A speckle client. Defaults to a global `client`.
+        commit_id (str): id of a commit, if nothing is specified, the latest commit will be fetched
+    Returns:
+        dict: The speckle stream data received from the specified branch.
+    This function retrieves the last commit from a specific branch of a speckle stream.
+    It uses the provided speckle client to get the branch and commit information, and then 
+    retrieves the speckle stream data associated with the last commit.
+    It prints out the branch details and the creation dates of the last three commits for debugging purposes.
+    """
+
+    print("updated A")
+
+    # set stream and branch
+    try:
+        branch = client.branch.get(stream_id, branch_name, 3)
+        print(branch)
+    except:
+        branch = client.branch.get(stream_id, branch_name, 1)
+        print(branch)
+
+    print("last three commits:")
+    [print(ite.createdAt) for ite in branch.commits.items]
+
+    if commit_id == "":
+        latest_commit = branch.commits.items[0]
+        choosen_commit_id = latest_commit.id
+        commit = client.commit.get(stream_id, choosen_commit_id)
+        print("latest commit ", branch.commits.items[0].createdAt, " was choosen")
+    elif type(commit_id) == type("s"): # string, commit uuid
+        choosen_commit_id = commit_id
+        commit = client.commit.get(stream_id, choosen_commit_id)
+        print("provided commit ", choosen_commit_id, " was choosen")
+    elif type(commit_id) == type(1): #int 
+        latest_commit = branch.commits.items[commit_id]
+        choosen_commit_id = latest_commit.id
+        commit = client.commit.get(stream_id, choosen_commit_id)
+
+
+    print(commit)
+    print(commit.referencedObject)
+    # get transport
+    transport = ServerTransport(client=client, stream_id=stream_id)
+    #speckle stream
+    res = operations.receive(commit.referencedObject, transport)
+
+    return res