Fixed bug with names displayed in "usage" entries in locust output

In Python parser.

Author: zomglings

locust/cli.py

@@ -2,10 +2,7 @@
 The Locust CLI
 """
 import argparse
-import json
-import os
 import sys
-from typing import Any, Dict, Optional
 
 from . import git
 from . import parse
@@ -18,6 +15,13 @@ def generate_argument_parser() -> argparse.ArgumentParser:
         description="Locust: Analyze Python code across git references",
         epilog=f"Version {version.LOCUST_VERSION}",
     )
+    parser.add_argument(
+        "-v",
+        "--version",
+        action="version",
+        version=f"locust {version.LOCUST_VERSION}",
+        help="Print the locust version",
+    )
     git.populate_argument_parser(parser)
     parse.populate_argument_parser(parser)
     render.populate_argument_parser(parser)

locust/parse.py

@@ -205,12 +205,17 @@ def visit_Attribute(self, node: ast.Attribute) -> None:
         ]
         symbols = self._current_symbols()
         parent = self._current_scope_parent()
+        final_component = cumulative_components[-1]
         for cumulative_component in cumulative_components:
             qualifications = symbols.get(cumulative_component)
             if qualifications:
                 for qualification in qualifications:
+                    # Replace prefix with qualification in final_component
+                    name = final_component.replace(
+                        cumulative_component, qualification, 1
+                    )
                     definition = RawDefinition(
-                        name=qualification,
+                        name=name,
                         change_type=self.context_type.value,
                         line=node.lineno,
                         offset=node.col_offset,

locust/version.py

@@ -1,4 +1,4 @@
-LOCUST_VERSION = "0.3.1"
+LOCUST_VERSION = "0.3.2"
 
 
 def main():

test.sh

@@ -15,4 +15,6 @@ then
 	exit 1
 fi
 
-python -m unittest discover -v
+TEST_COMMAND=${@:-"discover -v"}
+
+python -m unittest $TEST_COMMAND

tests/fixtures/test_parse.json

@@ -127,7 +127,7 @@
       "parent": {}
     },
     {
-      "name": "argparse",
+      "name": "argparse.ArgumentParser",
       "change_type": "usage",
       "filepath": "sample.py",
       "revision": "0838585",

tests/fixtures/test_parse_dependencies.json

@@ -312,7 +312,7 @@
       "parent": {}
     },
     {
-      "name": "json",
+      "name": "json.dumps",
       "change_type": "usage",
       "filepath": "mod/submod/print.py",
       "revision": "04cec01",

tests/fixtures/test_render.json

@@ -20,7 +20,7 @@
           "children": []
         },
         {
-          "name": "argparse",
+          "name": "argparse.ArgumentParser",
           "type": "usage",
           "line": 7,
           "changed_lines": 1,