Merge pull request #4 from SchrodingersGat/master

Added more models
This commit is contained in:
Oliver 2017-03-28 23:14:25 +11:00 committed by GitHub
commit 2d9206478e
10 changed files with 101 additions and 53 deletions

View File

@ -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 {} +

View File

@ -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

View File

@ -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):

View File

@ -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
"""

View File

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

View File

@ -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)

View File

@ -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,

View File

@ -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)

View File

@ -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

View File

@ -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,
@ -24,9 +26,30 @@ class UniquePart(models.Model):
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)