Merge pull request #4 from SchrodingersGat/master

Added more models

.travis.yml

@@ -1,13 +1,10 @@
 language: python
 python:
-    - 2.7
     - 3.3
-    - 3.4
-    - 3.5
 
 before_install:
     - pip install pep8
     
 script:
     # TODO - Only perform PEP8 checks on files that have been changed in this push / PR
-    - find . -name \*.py -exec pep8 --ignore=E402 {} +
+    - find . -name \*.py -exec pep8 --ignore=E402,W293 {} +

InvenTree/InvenTree/models.py

@@ -4,12 +4,37 @@ from django.db import models
 from django.core.exceptions import ObjectDoesNotExist
 from django.contrib.contenttypes.models import ContentType
 
+class Company(models.Model):
+    """ Abstract model representing an external company
+    """
+    
+    class Meta:
+        abstract = True
+        
+    name = models.CharField(max_length=100)
+    URL = models.URLField(blank=True)
+    address = models.CharField(max_length=200,
+                               blank=True)
+    phone = models.CharField(max_length=50,
+                             blank=True)
+    email = models.EmailField(blank=True)
+    contact = models.CharField(max_length=100,
+                               blank=True)
+    notes = models.CharField(max_length=500,
+                             blank=True)
+                             
+    def __str__(self):
+        return self.name
+
 class InvenTreeTree(models.Model):
     """ Provides an abstracted self-referencing tree model for data categories.
     - Each Category has one parent Category, which can be blank (for a top-level Category).
     - Each Category can have zero-or-more child Categor(y/ies)
     """
     
+    class Meta:
+        abstract = True
+    
     name = models.CharField(max_length=100)
     description = models.CharField(max_length=250)
     parent = models.ForeignKey('self',
@@ -124,7 +149,3 @@ class InvenTreeTree(models.Model):
         """
         
         return self.path
-            
-    
-    class Meta:
-        abstract = True

InvenTree/part/admin.py

@@ -4,7 +4,7 @@ from .models import PartCategory, Part
 
 class PartAdmin(admin.ModelAdmin):
     
-    list_display = ('name', 'IPN', 'quantity', 'category')
+    list_display = ('name', 'IPN', 'stock', 'category')
 
 # Custom form for PartCategory
 class PartCategoryAdmin(admin.ModelAdmin):

InvenTree/part/models.py

@@ -45,7 +45,7 @@ class Part(models.Model):
         return self.stockitem_set.all()
         
     @property
-    def quantity(self):
+    def stock(self):
         """ Return the total stock quantity for this part.
         Part may be stored in multiple locations
         """

InvenTree/part/serializers.py

@@ -9,7 +9,7 @@ class PartSerializer(serializers.ModelSerializer):
                   'IPN',
                   'description',
                   'category',
-                  'quantity')
+                  'stock')
         
 class PartCategorySerializer(serializers.ModelSerializer):
     class Meta:

InvenTree/stock/admin.py

@@ -6,7 +6,7 @@ class WarehouseAdmin(admin.ModelAdmin):
     list_display = ('name', 'path', 'description')
     
 class StockItemAdmin(admin.ModelAdmin):
-    list_display = ('part', 'quantity', 'location', 'updated')
+    list_display = ('part', 'quantity', 'location', 'status', 'updated')
 
 admin.site.register(Warehouse, WarehouseAdmin)
 admin.site.register(StockItem, StockItemAdmin)

InvenTree/stock/models.py

@@ -15,6 +15,20 @@ class StockItem(models.Model):
     quantity = models.IntegerField()
     updated = models.DateField(auto_now=True)
     
+    # Stock status types
+    ITEM_IN_PROGRESS = 0
+    ITEM_DAMAGED = 10
+    ITEM_ATTENTION = 20
+    ITEM_COMPLETE = 50
+    
+    status = models.IntegerField(default=ITEM_IN_PROGRESS,
+                                 choices=[
+                                 (ITEM_IN_PROGRESS, "In progress"),
+                                 (ITEM_DAMAGED, "Damaged"),
+                                 (ITEM_ATTENTION, "Requires attention"),
+                                 (ITEM_COMPLETE, "Complete")
+                                 ])
+    
     def __str__(self):
         return "{n} x {part} @ {loc}".format(
             n = self.quantity,

InvenTree/supplier/admin.py

@@ -1,9 +1,10 @@
 from django.contrib import admin
 
-from .models import Supplier, SupplierPart
+from .models import Supplier, SupplierPart, Customer
 
-class SupplierAdmin(admin.ModelAdmin):
+class CompanyAdmin(admin.ModelAdmin):
     list_display=('name','URL','contact')
 
-admin.site.register(Supplier, SupplierAdmin)
+admin.site.register(Customer, CompanyAdmin)
+admin.site.register(Supplier, CompanyAdmin)
 admin.site.register(SupplierPart)

InvenTree/supplier/models.py

@@ -2,26 +2,20 @@ from __future__ import unicode_literals
 
 from django.db import models
 
+from InvenTree.models import Company
 from part.models import Part
 
-class Supplier(models.Model):
+
+class Supplier(Company):
     """ Represents a manufacturer or supplier
     """
     
-    name = models.CharField(max_length=100)
-    URL = models.URLField(blank=True)
-    address = models.CharField(max_length=200,
-                               blank=True)
-    phone = models.CharField(max_length=50,
-                             blank=True)
-    email = models.EmailField(blank=True)
-    contact = models.CharField(max_length=100,
-                               blank=True)
-    notes = models.CharField(max_length=500,
-                             blank=True)
+    pass
+
+
+class Customer(Company):
+    pass
 
-    def __str__(self):
-        return self.name
 
 class SupplierPart(models.Model):
     """ Represents a unique part as provided by a Supplier
@@ -61,7 +55,7 @@ class SupplierPriceBreak(models.Model):
 
     def __str__(self):
         return "{mpn} - {cost}{currency} @ {quan}".format(
-            mpn = part.MPN,
-            cost = self.cost,
-            currency = self.currency if self.currency else '',
-            quan = self.quantity)
+            mpn=part.MPN,
+            cost=self.cost,
+            currency=self.currency if self.currency else '',
+            quan=self.quantity)

InvenTree/track/models.py

@@ -3,8 +3,10 @@ from __future__ import unicode_literals
 from django.db import models
 from django.contrib.auth.models import User
 
+from supplier.models import Customer
 from part.models import Part, PartRevision
 
+
 class UniquePart(models.Model):
     """ A unique instance of a Part object.
     Used for tracking parts based on serial numbers,
@@ -19,14 +21,35 @@ class UniquePart(models.Model):
                                  null=True)
 
     creation_date = models.DateField(auto_now_add=True,
-                               editable=False)
+                                     editable=False)
     serial = models.IntegerField()
 
     createdBy = models.ForeignKey(User)
 
+    customer = models.ForeignKey(Customer, blank=True, null=True)
+
+    # Part status types
+    PART_IN_PROGRESS = 0
+    PART_IN_STOCK = 10
+    PART_SHIPPED = 20
+    PART_RETURNED = 30
+    PART_DAMAGED = 40
+    PART_DESTROYED = 50
+
+    status = models.IntegerField(default=PART_IN_PROGRESS,
+                                choices=[
+                                (PART_IN_PROGRESS, "In progress"),
+                                (PART_IN_STOCK, "In stock"),
+                                (PART_SHIPPED, "Shipped"),
+                                (PART_RETURNED, "Returned"),
+                                (PART_DAMAGED, "Damaged"),
+                                (PART_DESTROYED, "Destroyed"),
+                                ])
+
     def __str__(self):
         return self.part.name
 
+
 class PartTrackingInfo(models.Model):
     """ Single data-point in the life of a UniquePart
     Each time something happens to the UniquePart,
@@ -37,5 +60,3 @@ class PartTrackingInfo(models.Model):
     date = models.DateField(auto_now_add=True,
                             editable=False)
     notes = models.CharField(max_length=500)
-    
-