Fix @rename to also affect display name

jtackaberry · Oct 24, 2021 · 45a2cea · 45a2cea
1 parent 135175b
commit 45a2cea
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/reference.py b/src/reference.py
@@ -260,10 +260,14 @@ def _set_name(self):
                 self._name = default_scope_name
                 display = display or self._name
         else:
-            if rename:
-                self.symbol = rename
             self._name = self.symbol
 
+        if rename:
+            if '.' in rename:
+                self.symbol = rename
+            else:
+                self.symbol = ''.join(re.split(r'([.:])', self.symbol)[:-1]) + rename
+
         if not display:
             if '.' not in self.symbol and ':' not in self.symbol and scope != '.':
                 display = (scope or default_scope_name) + '.' + self.symbol