Altered shipping scales to use attribute code, not name

django-oscar · May 31, 2012 · adfc96c · adfc96c
1 parent 0537ba6
commit adfc96c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/oscar/apps/shipping/__init__.py b/oscar/apps/shipping/__init__.py
@@ -5,13 +5,13 @@ class Scales(object):
     """
     For calculating the weight of a product or basket
     """
-    def __init__(self, attribute='weight', default_weight=None):
-        self.attribute = attribute
+    def __init__(self, attribute_code='weight', default_weight=None):
+        self.attribute = attribute_code
         self.default_weight = default_weight
 
     def weigh_product(self, product):
         try:
-            attr_val = product.attribute_values.get(attribute__name=self.attribute)
+            attr_val = product.attribute_values.get(attribute__code=self.attribute)
         except ObjectDoesNotExist:
             if self.default_weight is None:
                 raise ValueError("No attribute %s found for product %s" % (self.attribute, product))