set default currency if not set

allegro · Sep 11, 2020 · 8e099b8 · 8e099b8
1 parent 53b7003
commit 8e099b8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/ralph/data_importer/fields.py b/src/ralph/data_importer/fields.py
@@ -3,6 +3,8 @@
 from djmoney.money import Money
 from import_export import fields
 
+from ralph.settings import DEFAULT_CURRENCY_CODE
+
 logger = logging.getLogger(__name__)
 
 
@@ -87,4 +89,8 @@ def save(self, obj, data):
 
 class PriceField(fields.Field):
     def save(self, obj, data):
-        setattr(obj, 'price', Money(data['price'], data['price_currency']))
+        price = Money(
+            data['price'],
+            data.get('price_currency', DEFAULT_CURRENCY_CODE)
+        )
+        setattr(obj, 'price', price)
diff --git a/src/ralph/lib/mixins/models.py b/src/ralph/lib/mixins/models.py
@@ -7,6 +7,7 @@
 from taggit.managers import TaggableManager as TaggableManagerOriginal
 
 from ralph.lib.mixins.fields import TaggitTagField
+from ralph.settings import DEFAULT_CURRENCY_CODE
 
 
 class NamedMixin(models.Model):
@@ -123,7 +124,7 @@ class Meta:
 class PriceMixin(models.Model):
     price = MoneyField(
         max_digits=15, decimal_places=2, null=True, default=0,
-        default_currency='XXX'
+        default_currency=DEFAULT_CURRENCY_CODE
     )
 
     class Meta: