Merge pull request #75 from auscompgeek/enum-underscores

Remove trailing underscore from enum variant names
robotpy · Dec 4, 2018 · acb40bd · acb40bd
2 parents 575d5f5 + 61d6ae8
commit acb40bd
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/gen/hooks.py b/gen/hooks.py
@@ -81,13 +81,11 @@ def header_hook(header, data):
             name = v['name']
             if name.startswith(ename):
                 name = name[len(ename):]
+            name = name.rstrip('_')
             if name == 'None':
                 name = 'None_'
-            try:
-                int(name[0])
+            elif name[0].isdigit():
                 name = v['name'][0] + name
-            except ValueError:
-                pass
             v['x_name'] = name
 
 def function_hook(fn, data):