mirror of
https://github.com/inventree/InvenTree
synced 2024-08-30 18:33:04 +00:00
1d85b70313
* Update models: add MetadataMixin * Fix name of model in Metadata API definition * Add API endpoints * Update API version * Fix syntax * Add API endpoint for RO, RO line, RO line extra item * Add Metadata to Contacts * Fix link in API version * Fix name of model * Fix error? * Fix error? * Fix all errors, hopefully.. * Add tests for order, line, extraline metadata Extend for PO, SO * Add tests for metadata for Company-related models * Fix spelling * Consolidate metadata test for all part models into one test * Add test for all Stock metadata * Update stock test_api * Add all metadata tests for orders * Fix various errors in tests * Fix model name * Add migration files * Update tests for metadata * Resolve conflict around API version number * Rename migration file * Rename migration file * Will Contact edit endpoint work better? * Revert changes in URL definitions * Remove test, duplicate * Fix tests with fixed PK, not from fixtures, to use a dynamic PK * Fix migration overlap
574 lines
16 KiB
Python
574 lines
16 KiB
Python
"""Provides a JSON API for the Company app."""
|
|
|
|
from django.db.models import Q
|
|
from django.urls import include, path, re_path
|
|
|
|
from django_filters import rest_framework as rest_filters
|
|
from django_filters.rest_framework import DjangoFilterBackend
|
|
|
|
import part.models
|
|
from InvenTree.api import (AttachmentMixin, ListCreateDestroyAPIView,
|
|
MetadataView)
|
|
from InvenTree.filters import (ORDER_FILTER, SEARCH_ORDER_FILTER,
|
|
SEARCH_ORDER_FILTER_ALIAS)
|
|
from InvenTree.helpers import str2bool
|
|
from InvenTree.mixins import ListCreateAPI, RetrieveUpdateDestroyAPI
|
|
|
|
from .models import (Company, CompanyAttachment, Contact, ManufacturerPart,
|
|
ManufacturerPartAttachment, ManufacturerPartParameter,
|
|
SupplierPart, SupplierPriceBreak)
|
|
from .serializers import (CompanyAttachmentSerializer, CompanySerializer,
|
|
ContactSerializer,
|
|
ManufacturerPartAttachmentSerializer,
|
|
ManufacturerPartParameterSerializer,
|
|
ManufacturerPartSerializer, SupplierPartSerializer,
|
|
SupplierPriceBreakSerializer)
|
|
|
|
|
|
class CompanyList(ListCreateAPI):
|
|
"""API endpoint for accessing a list of Company objects.
|
|
|
|
Provides two methods:
|
|
|
|
- GET: Return list of objects
|
|
- POST: Create a new Company object
|
|
"""
|
|
|
|
serializer_class = CompanySerializer
|
|
queryset = Company.objects.all()
|
|
|
|
def get_queryset(self):
|
|
"""Return annotated queryset for the company list endpoint"""
|
|
queryset = super().get_queryset()
|
|
queryset = CompanySerializer.annotate_queryset(queryset)
|
|
|
|
return queryset
|
|
|
|
filter_backends = SEARCH_ORDER_FILTER
|
|
|
|
filterset_fields = [
|
|
'is_customer',
|
|
'is_manufacturer',
|
|
'is_supplier',
|
|
'name',
|
|
]
|
|
|
|
search_fields = [
|
|
'name',
|
|
'description',
|
|
'website',
|
|
]
|
|
|
|
ordering_fields = [
|
|
'name',
|
|
'parts_supplied',
|
|
'parts_manufactured',
|
|
]
|
|
|
|
ordering = 'name'
|
|
|
|
|
|
class CompanyDetail(RetrieveUpdateDestroyAPI):
|
|
"""API endpoint for detail of a single Company object."""
|
|
|
|
queryset = Company.objects.all()
|
|
serializer_class = CompanySerializer
|
|
|
|
def get_queryset(self):
|
|
"""Return annotated queryset for the company detail endpoint"""
|
|
queryset = super().get_queryset()
|
|
queryset = CompanySerializer.annotate_queryset(queryset)
|
|
|
|
return queryset
|
|
|
|
|
|
class CompanyAttachmentList(AttachmentMixin, ListCreateDestroyAPIView):
|
|
"""API endpoint for the CompanyAttachment model"""
|
|
|
|
queryset = CompanyAttachment.objects.all()
|
|
serializer_class = CompanyAttachmentSerializer
|
|
|
|
filter_backends = [
|
|
DjangoFilterBackend,
|
|
]
|
|
|
|
filterset_fields = [
|
|
'company',
|
|
]
|
|
|
|
|
|
class CompanyAttachmentDetail(AttachmentMixin, RetrieveUpdateDestroyAPI):
|
|
"""Detail endpoint for CompanyAttachment model."""
|
|
|
|
queryset = CompanyAttachment.objects.all()
|
|
serializer_class = CompanyAttachmentSerializer
|
|
|
|
|
|
class ContactList(ListCreateDestroyAPIView):
|
|
"""API endpoint for list view of Company model"""
|
|
|
|
queryset = Contact.objects.all()
|
|
serializer_class = ContactSerializer
|
|
|
|
filter_backends = SEARCH_ORDER_FILTER
|
|
|
|
filterset_fields = [
|
|
'company',
|
|
]
|
|
|
|
search_fields = [
|
|
'company__name',
|
|
'name',
|
|
]
|
|
|
|
ordering_fields = [
|
|
'name',
|
|
]
|
|
|
|
ordering = 'name'
|
|
|
|
|
|
class ContactDetail(RetrieveUpdateDestroyAPI):
|
|
"""Detail endpoint for Company model"""
|
|
|
|
queryset = Contact.objects.all()
|
|
serializer_class = ContactSerializer
|
|
|
|
|
|
class ManufacturerPartFilter(rest_filters.FilterSet):
|
|
"""Custom API filters for the ManufacturerPart list endpoint."""
|
|
|
|
class Meta:
|
|
"""Metaclass options."""
|
|
|
|
model = ManufacturerPart
|
|
fields = [
|
|
'manufacturer',
|
|
'MPN',
|
|
'part',
|
|
'tags__name',
|
|
'tags__slug',
|
|
]
|
|
|
|
# Filter by 'active' status of linked part
|
|
active = rest_filters.BooleanFilter(field_name='part__active')
|
|
|
|
|
|
class ManufacturerPartList(ListCreateDestroyAPIView):
|
|
"""API endpoint for list view of ManufacturerPart object.
|
|
|
|
- GET: Return list of ManufacturerPart objects
|
|
- POST: Create a new ManufacturerPart object
|
|
"""
|
|
|
|
queryset = ManufacturerPart.objects.all().prefetch_related(
|
|
'part',
|
|
'manufacturer',
|
|
'supplier_parts',
|
|
'tags',
|
|
)
|
|
|
|
serializer_class = ManufacturerPartSerializer
|
|
filterset_class = ManufacturerPartFilter
|
|
|
|
def get_serializer(self, *args, **kwargs):
|
|
"""Return serializer instance for this endpoint"""
|
|
# Do we wish to include extra detail?
|
|
try:
|
|
params = self.request.query_params
|
|
|
|
kwargs['part_detail'] = str2bool(params.get('part_detail', None))
|
|
kwargs['manufacturer_detail'] = str2bool(params.get('manufacturer_detail', None))
|
|
kwargs['pretty'] = str2bool(params.get('pretty', None))
|
|
except AttributeError:
|
|
pass
|
|
|
|
kwargs['context'] = self.get_serializer_context()
|
|
|
|
return self.serializer_class(*args, **kwargs)
|
|
|
|
filter_backends = SEARCH_ORDER_FILTER
|
|
|
|
search_fields = [
|
|
'manufacturer__name',
|
|
'description',
|
|
'MPN',
|
|
'part__IPN',
|
|
'part__name',
|
|
'part__description',
|
|
'tags__name',
|
|
'tags__slug',
|
|
]
|
|
|
|
|
|
class ManufacturerPartDetail(RetrieveUpdateDestroyAPI):
|
|
"""API endpoint for detail view of ManufacturerPart object.
|
|
|
|
- GET: Retrieve detail view
|
|
- PATCH: Update object
|
|
- DELETE: Delete object
|
|
"""
|
|
|
|
queryset = ManufacturerPart.objects.all()
|
|
serializer_class = ManufacturerPartSerializer
|
|
|
|
|
|
class ManufacturerPartAttachmentList(AttachmentMixin, ListCreateDestroyAPIView):
|
|
"""API endpoint for listing (and creating) a ManufacturerPartAttachment (file upload)."""
|
|
|
|
queryset = ManufacturerPartAttachment.objects.all()
|
|
serializer_class = ManufacturerPartAttachmentSerializer
|
|
|
|
filter_backends = [
|
|
DjangoFilterBackend,
|
|
]
|
|
|
|
filterset_fields = [
|
|
'manufacturer_part',
|
|
]
|
|
|
|
|
|
class ManufacturerPartAttachmentDetail(AttachmentMixin, RetrieveUpdateDestroyAPI):
|
|
"""Detail endpooint for ManufacturerPartAttachment model."""
|
|
|
|
queryset = ManufacturerPartAttachment.objects.all()
|
|
serializer_class = ManufacturerPartAttachmentSerializer
|
|
|
|
|
|
class ManufacturerPartParameterFilter(rest_filters.FilterSet):
|
|
"""Custom filterset for the ManufacturerPartParameterList API endpoint"""
|
|
|
|
class Meta:
|
|
"""Metaclass options"""
|
|
model = ManufacturerPartParameter
|
|
fields = [
|
|
'name',
|
|
'value',
|
|
'units',
|
|
'manufacturer_part',
|
|
]
|
|
|
|
manufacturer = rest_filters.ModelChoiceFilter(queryset=Company.objects.all(), field_name='manufacturer_part__manufacturer')
|
|
|
|
part = rest_filters.ModelChoiceFilter(queryset=part.models.Part.objects.all(), field_name='manufacturer_part__part')
|
|
|
|
|
|
class ManufacturerPartParameterList(ListCreateDestroyAPIView):
|
|
"""API endpoint for list view of ManufacturerPartParamater model."""
|
|
|
|
queryset = ManufacturerPartParameter.objects.all()
|
|
serializer_class = ManufacturerPartParameterSerializer
|
|
filterset_class = ManufacturerPartParameterFilter
|
|
|
|
def get_serializer(self, *args, **kwargs):
|
|
"""Return serializer instance for this endpoint"""
|
|
# Do we wish to include any extra detail?
|
|
try:
|
|
params = self.request.query_params
|
|
|
|
optional_fields = [
|
|
'manufacturer_part_detail',
|
|
]
|
|
|
|
for key in optional_fields:
|
|
kwargs[key] = str2bool(params.get(key, None))
|
|
|
|
except AttributeError:
|
|
pass
|
|
|
|
kwargs['context'] = self.get_serializer_context()
|
|
|
|
return self.serializer_class(*args, **kwargs)
|
|
|
|
filter_backends = SEARCH_ORDER_FILTER
|
|
|
|
search_fields = [
|
|
'name',
|
|
'value',
|
|
'units',
|
|
]
|
|
|
|
|
|
class ManufacturerPartParameterDetail(RetrieveUpdateDestroyAPI):
|
|
"""API endpoint for detail view of ManufacturerPartParameter model."""
|
|
|
|
queryset = ManufacturerPartParameter.objects.all()
|
|
serializer_class = ManufacturerPartParameterSerializer
|
|
|
|
|
|
class SupplierPartFilter(rest_filters.FilterSet):
|
|
"""API filters for the SupplierPartList endpoint"""
|
|
|
|
class Meta:
|
|
"""Metaclass option"""
|
|
|
|
model = SupplierPart
|
|
fields = [
|
|
'supplier',
|
|
'part',
|
|
'manufacturer_part',
|
|
'SKU',
|
|
'tags__name',
|
|
'tags__slug',
|
|
]
|
|
|
|
# Filter by 'active' status of linked part
|
|
active = rest_filters.BooleanFilter(field_name='part__active')
|
|
|
|
# Filter by the 'MPN' of linked manufacturer part
|
|
MPN = rest_filters.CharFilter(
|
|
label='Manufacturer Part Number',
|
|
field_name='manufacturer_part__MPN',
|
|
lookup_expr='iexact'
|
|
)
|
|
|
|
|
|
class SupplierPartList(ListCreateDestroyAPIView):
|
|
"""API endpoint for list view of SupplierPart object.
|
|
|
|
- GET: Return list of SupplierPart objects
|
|
- POST: Create a new SupplierPart object
|
|
"""
|
|
|
|
queryset = SupplierPart.objects.all().prefetch_related(
|
|
'tags',
|
|
)
|
|
filterset_class = SupplierPartFilter
|
|
|
|
def get_queryset(self, *args, **kwargs):
|
|
"""Return annotated queryest object for the SupplierPart list"""
|
|
queryset = super().get_queryset(*args, **kwargs)
|
|
queryset = SupplierPartSerializer.annotate_queryset(queryset)
|
|
|
|
return queryset
|
|
|
|
def filter_queryset(self, queryset):
|
|
"""Custom filtering for the queryset."""
|
|
queryset = super().filter_queryset(queryset)
|
|
|
|
params = self.request.query_params
|
|
|
|
# Filter by manufacturer
|
|
manufacturer = params.get('manufacturer', None)
|
|
|
|
if manufacturer is not None:
|
|
queryset = queryset.filter(manufacturer_part__manufacturer=manufacturer)
|
|
|
|
# Filter by EITHER manufacturer or supplier
|
|
company = params.get('company', None)
|
|
|
|
if company is not None:
|
|
queryset = queryset.filter(Q(manufacturer_part__manufacturer=company) | Q(supplier=company))
|
|
|
|
return queryset
|
|
|
|
def get_serializer(self, *args, **kwargs):
|
|
"""Return serializer instance for this endpoint"""
|
|
|
|
# Do we wish to include extra detail?
|
|
try:
|
|
params = self.request.query_params
|
|
kwargs['part_detail'] = str2bool(params.get('part_detail', None))
|
|
kwargs['supplier_detail'] = str2bool(params.get('supplier_detail', True))
|
|
kwargs['manufacturer_detail'] = str2bool(params.get('manufacturer_detail', None))
|
|
kwargs['pretty'] = str2bool(params.get('pretty', None))
|
|
except AttributeError:
|
|
pass
|
|
|
|
kwargs['context'] = self.get_serializer_context()
|
|
|
|
return self.serializer_class(*args, **kwargs)
|
|
|
|
serializer_class = SupplierPartSerializer
|
|
|
|
filter_backends = SEARCH_ORDER_FILTER_ALIAS
|
|
|
|
ordering_fields = [
|
|
'SKU',
|
|
'part',
|
|
'supplier',
|
|
'manufacturer',
|
|
'MPN',
|
|
'packaging',
|
|
'pack_quantity',
|
|
'in_stock',
|
|
'updated',
|
|
]
|
|
|
|
ordering_field_aliases = {
|
|
'part': 'part__name',
|
|
'supplier': 'supplier__name',
|
|
'manufacturer': 'manufacturer_part__manufacturer__name',
|
|
'MPN': 'manufacturer_part__MPN',
|
|
'pack_quantity': ['pack_quantity_native', 'pack_quantity'],
|
|
}
|
|
|
|
search_fields = [
|
|
'SKU',
|
|
'supplier__name',
|
|
'manufacturer_part__manufacturer__name',
|
|
'description',
|
|
'manufacturer_part__MPN',
|
|
'part__IPN',
|
|
'part__name',
|
|
'part__description',
|
|
'part__keywords',
|
|
'tags__name',
|
|
'tags__slug',
|
|
]
|
|
|
|
|
|
class SupplierPartDetail(RetrieveUpdateDestroyAPI):
|
|
"""API endpoint for detail view of SupplierPart object.
|
|
|
|
- GET: Retrieve detail view
|
|
- PATCH: Update object
|
|
- DELETE: Delete object
|
|
"""
|
|
|
|
queryset = SupplierPart.objects.all()
|
|
serializer_class = SupplierPartSerializer
|
|
|
|
read_only_fields = [
|
|
]
|
|
|
|
|
|
class SupplierPriceBreakFilter(rest_filters.FilterSet):
|
|
"""Custom API filters for the SupplierPriceBreak list endpoint"""
|
|
|
|
class Meta:
|
|
"""Metaclass options"""
|
|
|
|
model = SupplierPriceBreak
|
|
fields = [
|
|
'part',
|
|
'quantity',
|
|
]
|
|
|
|
base_part = rest_filters.ModelChoiceFilter(
|
|
label='Base Part',
|
|
queryset=part.models.Part.objects.all(),
|
|
field_name='part__part',
|
|
)
|
|
|
|
supplier = rest_filters.ModelChoiceFilter(
|
|
label='Supplier',
|
|
queryset=Company.objects.all(),
|
|
field_name='part__supplier',
|
|
)
|
|
|
|
|
|
class SupplierPriceBreakList(ListCreateAPI):
|
|
"""API endpoint for list view of SupplierPriceBreak object.
|
|
|
|
- GET: Retrieve list of SupplierPriceBreak objects
|
|
- POST: Create a new SupplierPriceBreak object
|
|
"""
|
|
|
|
queryset = SupplierPriceBreak.objects.all()
|
|
serializer_class = SupplierPriceBreakSerializer
|
|
filterset_class = SupplierPriceBreakFilter
|
|
|
|
def get_serializer(self, *args, **kwargs):
|
|
"""Return serializer instance for this endpoint"""
|
|
|
|
try:
|
|
params = self.request.query_params
|
|
|
|
kwargs['part_detail'] = str2bool(params.get('part_detail', False))
|
|
kwargs['supplier_detail'] = str2bool(params.get('supplier_detail', False))
|
|
|
|
except AttributeError:
|
|
pass
|
|
|
|
kwargs['context'] = self.get_serializer_context()
|
|
|
|
return self.serializer_class(*args, **kwargs)
|
|
|
|
filter_backends = ORDER_FILTER
|
|
|
|
ordering_fields = [
|
|
'quantity',
|
|
]
|
|
|
|
ordering = 'quantity'
|
|
|
|
|
|
class SupplierPriceBreakDetail(RetrieveUpdateDestroyAPI):
|
|
"""Detail endpoint for SupplierPriceBreak object."""
|
|
|
|
queryset = SupplierPriceBreak.objects.all()
|
|
serializer_class = SupplierPriceBreakSerializer
|
|
|
|
|
|
manufacturer_part_api_urls = [
|
|
|
|
# Base URL for ManufacturerPartAttachment API endpoints
|
|
re_path(r'^attachment/', include([
|
|
path(r'<int:pk>/', ManufacturerPartAttachmentDetail.as_view(), name='api-manufacturer-part-attachment-detail'),
|
|
re_path(r'^$', ManufacturerPartAttachmentList.as_view(), name='api-manufacturer-part-attachment-list'),
|
|
])),
|
|
|
|
re_path(r'^parameter/', include([
|
|
path(r'<int:pk>/', ManufacturerPartParameterDetail.as_view(), name='api-manufacturer-part-parameter-detail'),
|
|
|
|
# Catch anything else
|
|
re_path(r'^.*$', ManufacturerPartParameterList.as_view(), name='api-manufacturer-part-parameter-list'),
|
|
])),
|
|
|
|
re_path(r'^(?P<pk>\d+)/?', include([
|
|
re_path('^metadata/', MetadataView.as_view(), {'model': ManufacturerPart}, name='api-manufacturer-part-metadata'),
|
|
re_path('^.*$', ManufacturerPartDetail.as_view(), name='api-manufacturer-part-detail'),
|
|
])),
|
|
|
|
# Catch anything else
|
|
re_path(r'^.*$', ManufacturerPartList.as_view(), name='api-manufacturer-part-list'),
|
|
]
|
|
|
|
|
|
supplier_part_api_urls = [
|
|
|
|
re_path(r'^(?P<pk>\d+)/?', include([
|
|
re_path('^metadata/', MetadataView.as_view(), {'model': SupplierPart}, name='api-supplier-part-metadata'),
|
|
re_path('^.*$', SupplierPartDetail.as_view(), name='api-supplier-part-detail'),
|
|
])),
|
|
|
|
# Catch anything else
|
|
re_path(r'^.*$', SupplierPartList.as_view(), name='api-supplier-part-list'),
|
|
]
|
|
|
|
|
|
company_api_urls = [
|
|
re_path(r'^part/manufacturer/', include(manufacturer_part_api_urls)),
|
|
|
|
re_path(r'^part/', include(supplier_part_api_urls)),
|
|
|
|
# Supplier price breaks
|
|
re_path(r'^price-break/', include([
|
|
|
|
re_path(r'^(?P<pk>\d+)/?', SupplierPriceBreakDetail.as_view(), name='api-part-supplier-price-detail'),
|
|
re_path(r'^.*$', SupplierPriceBreakList.as_view(), name='api-part-supplier-price-list'),
|
|
])),
|
|
|
|
re_path(r'^(?P<pk>\d+)/?', include([
|
|
re_path(r'^metadata/', MetadataView.as_view(), {'model': Company}, name='api-company-metadata'),
|
|
re_path(r'^.*$', CompanyDetail.as_view(), name='api-company-detail'),
|
|
])),
|
|
|
|
re_path(r'^attachment/', include([
|
|
path(r'<int:pk>/', CompanyAttachmentDetail.as_view(), name='api-company-attachment-detail'),
|
|
re_path(r'^$', CompanyAttachmentList.as_view(), name='api-company-attachment-list'),
|
|
])),
|
|
|
|
re_path(r'^contact/', include([
|
|
re_path(r'^(?P<pk>\d+)/?', include([
|
|
re_path('^metadata/', MetadataView.as_view(), {'model': Contact}, name='api-contact-metadata'),
|
|
re_path('^.*$', ContactDetail.as_view(), name='api-contact-detail'),
|
|
])),
|
|
re_path(r'^.*$', ContactList.as_view(), name='api-contact-list'),
|
|
])),
|
|
|
|
re_path(r'^.*$', CompanyList.as_view(), name='api-company-list'),
|
|
|
|
]
|