updates

InvenTree/part/models.py

@@ -1,5 +1,5 @@
 from __future__ import unicode_literals
-
+from django.utils.translation import ugettext as _
 from django.db import models
 from django.db.models import Sum
 from django.core.exceptions import ObjectDoesNotExist, ValidationError
@@ -10,15 +10,15 @@ from InvenTree.models import InvenTreeTree
 class PartCategory(InvenTreeTree):
     """ PartCategory provides hierarchical organization of Part objects.
     """
-    
+
     class Meta:
         verbose_name = "Part Category"
         verbose_name_plural = "Part Categories"
-        
-        
+
+
 class Part(models.Model):
     """ Represents a """
-    
+
     name = models.CharField(max_length=100)
     description = models.CharField(max_length=250, blank=True)
     IPN = models.CharField(max_length=100, blank=True)
@@ -26,7 +26,7 @@ class Part(models.Model):
     minimum_stock = models.IntegerField(default=0)
     units = models.CharField(max_length=20, default="pcs", blank=True)
     trackable = models.BooleanField(default=False)
-    
+
     def __str__(self):
         if self.IPN:
             return "{name} ({ipn})".format(
@@ -34,39 +34,39 @@ class Part(models.Model):
                 name=self.name)
         else:
             return self.name
-        
+
     class Meta:
         verbose_name = "Part"
         verbose_name_plural = "Parts"
-        
+
     @property
     def stock(self):
         """ Return the total stock quantity for this part.
         Part may be stored in multiple locations
         """
-        
+
         stocks = self.locations.all()
         if len(stocks) == 0:
             return 0
-            
+
         result = stocks.aggregate(total=Sum('quantity'))
         return result['total']
-        
+
     @property
     def projects(self):
         """ Return a list of unique projects that this part is associated with
         """
-        
+
         project_ids = set()
         project_parts = self.projectpart_set.all()
-        
+
         projects = []
-        
+
         for pp in project_parts:
             if pp.project.id not in project_ids:
                 project_ids.add(pp.project.id)
                 projects.append(pp.project)
-                
+
         return projects
 
 
@@ -78,28 +78,31 @@ class PartParameterTemplate(models.Model):
     name = models.CharField(max_length=20)
     description = models.CharField(max_length=100, blank=True)
     units = models.CharField(max_length=10, blank=True)
-    
+
     default_value = models.CharField(max_length=50, blank=True)
     default_min = models.CharField(max_length=50, blank=True)
     default_max = models.CharField(max_length=50, blank=True)
-    
+
     # Parameter format
     PARAM_NUMERIC = 10
     PARAM_TEXT = 20
     PARAM_BOOL = 30
-    
+
+    PARAM_TYPE_CODES = {
+        PARAM_NUMERIC: _("Numeric"),
+        PARAM_TEXT: _("Text"),
+        PARAM_BOOL: _("Bool")
+    }
+
     format = models.IntegerField(
         default=PARAM_NUMERIC,
-        choices=[
-            (PARAM_NUMERIC, "Numeric"),
-            (PARAM_TEXT, "Text"),
-            (PARAM_BOOL, "Boolean")])
-    
+        choices=PARAM_TYPE_CODES.items())
+
     def __str__(self):
         return "{name} ({units})".format(
             name=self.name,
             units=self.units)
-    
+
     class Meta:
         verbose_name = "Parameter Template"
         verbose_name_plural = "Parameter Templates"
@@ -110,7 +113,7 @@ class CategoryParameterLink(models.Model):
     """
     category = models.ForeignKey(PartCategory, on_delete=models.CASCADE)
     template = models.ForeignKey(PartParameterTemplate, on_delete=models.CASCADE)
-    
+
     def __str__(self):
         return "{name} - {cat}".format(
             name=self.template.name,
@@ -119,20 +122,20 @@ class CategoryParameterLink(models.Model):
     class Meta:
         verbose_name = "Category Parameter"
         verbose_name_plural = "Category Parameters"
-            
+
 
 class PartParameter(models.Model):
     """ PartParameter is associated with a single part
     """
-    
+
     part = models.ForeignKey(Part, on_delete=models.CASCADE, related_name='parameters')
     template = models.ForeignKey(PartParameterTemplate)
-    
+
     # Value data
     value = models.CharField(max_length=50, blank=True)
     min_value = models.CharField(max_length=50, blank=True)
     max_value = models.CharField(max_length=50, blank=True)
-    
+
     # Prevent multiple parameters of the same template
     # from being added to the same part
     def save(self, *args, **kwargs):
@@ -141,27 +144,27 @@ class PartParameter(models.Model):
             return
         if len(params) == 1 and params[0].id != self.id:
             return
-            
+
         super(PartParameter, self).save(*args, **kwargs)
-    
+
     def __str__(self):
         return "{name} : {val}{units}".format(
             name=self.template.name,
             val=self.value,
             units=self.template.units)
-            
+
     @property
     def units(self):
         return self.template.units
-        
+
     @property
     def name(self):
         return self.template.name
-            
+
     class Meta:
         verbose_name = "Part Parameter"
         verbose_name_plural = "Part Parameters"
-    
+
 
 class PartRevision(models.Model):
     """ A PartRevision represents a change-notification to a Part
@@ -169,12 +172,12 @@ class PartRevision(models.Model):
     which should be tracked.
     UniqueParts can have a single associated PartRevision
     """
-    
+
     part = models.ForeignKey(Part, on_delete=models.CASCADE)
-    
+
     name = models.CharField(max_length=100)
     description = models.CharField(max_length=500)
     revision_date = models.DateField(auto_now_add=True)
-    
+
     def __str__(self):
         return self.name