MAINT: Add a helper

HaoZeke · HaoZeke · commit 3d58d8cbce4a · 2023-04-15T19:00:09.000Z
diff --git a/numpy/f2py/auxfuncs.py b/numpy/f2py/auxfuncs.py
@@ -16,6 +16,7 @@
 """
 import pprint
 import sys
+import re
 import types
 from functools import reduce
 
@@ -26,7 +27,7 @@
     'applyrules', 'debugcapi', 'dictappend', 'errmess', 'gentitle',
     'getargs2', 'getcallprotoargument', 'getcallstatement',
     'getfortranname', 'getpymethoddef', 'getrestdoc', 'getusercode',
-    'getusercode1', 'hasbody', 'hascallstatement', 'hascommon',
+    'getusercode1', 'getdimension', 'hasbody', 'hascallstatement', 'hascommon',
     'hasexternals', 'hasinitvalue', 'hasnote', 'hasresultnote',
     'isallocatable', 'isarray', 'isarrayofstrings',
     'ischaracter', 'ischaracterarray', 'ischaracter_or_characterarray',
@@ -417,6 +418,13 @@ def isexternal(var):
     return 'attrspec' in var and 'external' in var['attrspec']
 
 
+def getdimension(var):
+    dimpattern = r"\((.*?)\)"
+    if 'attrspec' in var.keys():
+        if any('dimension' in s for s in var['attrspec']):
+            return [re.findall(dimpattern, v) for v in var['attrspec']][0]
+
+
 def isrequired(var):
     return not isoptional(var) and isintent_nothide(var)
 
