Addressed comments from PR.

Author: mirelap-amazon

codeguru_profiler_agent/sampling_utils.py

@@ -87,12 +87,9 @@ def _maybe_append_synthetic_frame(result, frame, line_no):
 
 def _extract_frames(end_frame, max_depth):
     stack = list(traceback.walk_stack(end_frame))[::-1][0:max_depth]
-    # When running the sample app with uwsgi for Python 3.8.10-3.9.2, the traceback command
-    # returns a file path that contains "/.".
+    # When running the sample app with uwsgi for Python 3.8.10 - 3.9.2, the traceback command
+    # returns a file path that contains "/./" instead of just a "/" between the app directory and the module path.
     # To not let the path go into the module name, we are removing it later in the ProfileEncoder.
-    # Examples:
-    # - file '/Users/mirelap/Documents/workspace/JSON/aws-codeguru-profiler-python-demo-application/sample-demo-django-app/./polls/views.py', line 104, code get_queryset>, 104
-    # - file '/Users/mirelap/Documents/workspace/JSON/aws-codeguru-profiler-python-demo-application/sample-demo-django-app/polls/views.py', line 104, code get_queryset>, 104
     stack_entries = _extract_stack(stack, max_depth)
 
     if len(stack_entries) == max_depth:

codeguru_profiler_agent/sdk_reporter/profile_encoder.py

@@ -10,7 +10,6 @@
 GZIP_BALANCED_COMPRESSION_LEVEL = 6
 DEFAULT_FRAME_COMPONENT_DELIMITER = ":"
 
-
 def _get_module_path(file_path, sys_paths):
     """
     We tried to remove the python library root path in order to give a reasonable expression of the module path.
@@ -23,6 +22,12 @@ def _get_module_path(file_path, sys_paths):
     will get turned into `polls.views' given that the file path contains the current path.
     This should not happen usually, but we've found a case where the "/." is added when calling traceback.walk_stack(..)
     in a uwsgi application. Check sampling_utils.py file for details.
+
+    sampling_utils.py returns different values when calling traceback.walk_stack(..) for uwsgi vs non-uwsgi
+    for Python 3.8.10-Python 3.9.2.
+    Examples of results:
+    - file '/Users/mirelap/Documents/workspace/JSON/aws-codeguru-profiler-python-demo-application/sample-demo-django-app/./polls/views.py', line 104, code get_queryset>, 104
+    - file '/Users/mirelap/Documents/workspace/JSON/aws-codeguru-profiler-python-demo-application/sample-demo-django-app/polls/views.py', line 104, code get_queryset>, 104
     """
     module_path = file_path
 
@@ -49,7 +54,7 @@ def _get_module_path(file_path, sys_paths):
 def _remove_prefix_path(module_path, sys_paths):
     current_path = str(Path().absolute())
     if current_path in module_path:
-        return module_path.replace(current_path, "").replace("/.", "")
+        return module_path.replace(current_path, "").replace("/./", "/")
     for root in sys_paths:
         if root in module_path:
             return module_path.replace(root, "")