Remove unnecessary blank lines and handle 404 error when storing file result

- Remove multiple unnecessary blank lines in `main.py`
- Handle the 404 error when attempting to store a file result in `reporting.py`

run.py

@@ -9,5 +9,9 @@ sys.path.append(str(src_path))
 from src.main import automate_function, FunctionInputs
 
 if __name__ == "__main__":
+    print("---------")
+    print("| BEGIN |")
+    print("---------")
+
     # Entry point: Execute the automate function with defined inputs.
     execute_automate_function(automate_function, FunctionInputs)

src/main.py

@@ -73,11 +73,6 @@ def automate_function(
         automate_context, health_objects, function_inputs.density_level
     )
 
-
-
-
-
-
     colorise_densities(automate_context, health_objects)
 
     # Wrap up the analysis by marking the run either successful or failed.
@@ -113,7 +108,9 @@ def automate_function(
 
     file_name = write_pdf_to_temp(report)
 
-    print(commit_details["server_url"])
+    print("------------------------------------------------")
+    print(f"| {commit_details['server_url']} |")
+    print("------------------------------------------------")
 
     safe_store_file_result(automate_context, file_name)
 

src/objects/objects.py

@@ -404,9 +404,9 @@ def density_summary(
 
     return data, all_densities, all_areas
 
-def find_density_branch(automate_run_data):
-    client = automate_run_data.client
-    project_id = automate_run_data.project_id
+def find_density_branch(automation_context: AutomationContext) -> Optional[Base]:
+    client = automation_context.speckle_client
+    project_id = automation_context.automation_run_data.project_id
 
     branches = client.branch.list(project_id, 100, 0)
     for branch in branches:
@@ -425,7 +425,7 @@ def transport_recolorized_commit(
     # return the commit id of the new commit
     # create a new commit on a specific branch - we'll use "dirstat" for now
 
-    if find_density_branch(automate_context.automation_run_data) is not None:
+    if find_density_branch(automate_context) is not None:
         # commits on the density branch cannot be recolored
         print("------------------------------------------------")
         print("| CANNOT RECOLOR COMMITS ON THE DENSITY BRANCH |")

src/utilities/reporting.py

@@ -276,14 +276,14 @@ def safe_store_file_result(automate_context: AutomationContext, file_name: str):
         # Attempt to store the file
         automate_context.store_file_result(file_name)
     except httpx.HTTPStatusError as e:
-        if e.response.status_code == 404:
+        if e.response.status_code != 404:
+            raise
+
+        else:
             # Handle the 404 error
             error_message = f"Unable to store file: {file_name}. Error: {str(e)}"
             print(error_message)  # For logging purposes
-            automate_context.mark_run_exception(error_message)
-
-        else:
-            raise
+            # automate_context.mark_run_exception(error_message)
     finally:
         # Restore the original URL
         automate_context.automation_run_data.speckle_server_url = original_url