NFVBENCH-184: Add a feature (+ options) allowing to change ownership of shared log...

[nfvbench.git] / nfvbench / utils.py

diff --git a/nfvbench/utils.py b/nfvbench/utils.py
index 2ce735b..6da14ed 100644 (file)
--- a/nfvbench/utils.py
+++ b/nfvbench/utils.py
@@ -51,7 +51,7 @@ def timeout(seconds=10, error_message=os.strerror(errno.ETIME)):
 
 
 def save_json_result(result, json_file, std_json_path, service_chain, service_chain_count,
-                     flow_count, frame_sizes):
+                     flow_count, frame_sizes, user_id=None, group_id=None):
     """Save results in json format file."""
     filepaths = []
     if json_file:
@@ -71,6 +71,11 @@ def save_json_result(result, json_file, std_json_path, service_chain, service_ch
                           sort_keys=True,
                           separators=(',', ': '),
                           default=lambda obj: obj.to_json())
+                # possibly change file ownership
+                if group_id is None:
+                    group_id = user_id
+                if user_id is not None:
+                    os.chown(file_path, user_id, group_id)
 
 
 def dict_to_json_dict(record):
@@ -155,7 +160,6 @@ def get_intel_pci(nic_slot=None, nic_ports=None):
     return pcis
 
 
-
 def parse_flow_count(flow_count):
     flow_count = str(flow_count)
     input_fc = flow_count