Carry over units to vector based VirtualLayers

Former-commit-id: 3ee75f6
NOAA-ORR-ERD · Jul 17, 2015 · 15d2afd · 15d2afd
1 parent 6eb5653
commit 15d2afd
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/wms/models/layer.py b/wms/models/layer.py
@@ -88,9 +88,16 @@ def make_vector_layer(cls, us, vs, std_name, style, dataset_id):
                 v_match = '_'.join([ x for x in v.standard_name.split('_') if x not in ['y', 'northward']])
 
                 if u_match == v_match:
+                    if hasattr(u, 'units'):
+                        units = u.units
+                    elif hasattr(v, 'units'):
+                        units = v.units
+                    else:
+                        units = None
                     try:
                         vl = VirtualLayer.objects.create(var_name='{},{}'.format(u._name, v._name),
                                                          std_name=std_name,
+                                                         units=units,
                                                          description="U ({}) and V ({}) vectors".format(u._name, v._name),
                                                          dataset_id=dataset_id,
                                                          active=True)