Make validation whitelist dynamic

Unidata · Oct 25, 2021 · 91ec453 · 91ec453
1 parent 0922ea8
commit 91ec453
Showing 1 changed file with 4 additions and 18 deletions.
diff --git a/src/metpy/plots/declarative.py b/src/metpy/plots/declarative.py
@@ -508,7 +508,7 @@ def plot_kwargs(data):
     return kwargs
 
 
-class ValidationMixin(object):
+class ValidationMixin:
     """Provides validation of attribute names when set by user."""
 
     def __setattr__(self, name, value):
@@ -530,25 +530,11 @@ def _levenshtein(s1, s2):
                 distances = new_distances
             return distances[-1]
 
-        whitelist = ('area',
-                     'ax',
-                     'clabels',
-                     'contours',
-                     'data',
-                     'layers',
-                     'layout',
-                     'linecolor',
+        whitelist = ['ax',
                      'notify_change',
-                     'panels',
-                     'parent',
-                     'plot_units',
-                     'plots',
-                     'projection',
-                     'size',
-                     'title',
-                     'title_fontsize',
-                     )
+                     ]
 
+        whitelist.extend(self.trait_names())
         if name in whitelist or name.startswith('_'):
             super().__setattr__(name, value)
         else: