FIX: MouseButton representation in boilerplate generated signatures

tacaswell · tacaswell · commit 97d07b87d23a · 2021-05-03T20:58:55.000-04:00
In Python 3.10 the repr and str representation of Enums changed from str: 'ClassName.NAME' -> 'NAME' repr: '<ClassName.NAME: value>' -> 'ClassName.NAME' which is more consistent with what str/repr should do, however this breaks boilerplate which needs to get the ClassName.NAME version in all versions of Python. Thus, we locally monkey patch our preferred str representation in here. bpo-40066 python/cpython#22392
diff --git a/tools/boilerplate.py b/tools/boilerplate.py
@@ -25,6 +25,28 @@
 from matplotlib import _api, mlab
 from matplotlib.axes import Axes
 from matplotlib.figure import Figure
+from matplotlib.backend_bases import MouseButton
+
+
+# we need to define a custom str because py310 change
+# In Python 3.10 the repr and str representation of Enums changed from
+#
+#  str: 'ClassName.NAME' -> 'NAME'
+#  repr: '<ClassName.NAME: value>' -> 'ClassName.NAME'
+#
+# which is more consistent with what str/repr should do, however this breaks
+# boilerplate which needs to get the ClassName.NAME version in all versions of
+# Python. Thus, we locally monkey patch our preferred str representation in
+# here.
+#
+# bpo-40066
+# https://github.com/python/cpython/pull/22392/
+def enum_str_back_compat_patch(self):
+    return f'{type(self).__name__}.{self.name}'
+
+# only monkey patch if we have to.
+if str(MouseButton.LEFT) != 'MouseButton.Left':
+    MouseButton.__str__ = enum_str_back_compat_patch
 
 
 # This is the magic line that must exist in pyplot, after which the boilerplate
