Cleanup for part.views

This commit is contained in:
Oliver Walters 2019-05-11 09:51:45 +10:00
parent 63114a29f1
commit 343850c4f0

View File

@ -17,14 +17,7 @@ from .models import PartCategory, Part, PartAttachment
from .models import BomItem from .models import BomItem
from .models import SupplierPart from .models import SupplierPart
from .forms import PartImageForm from . import forms as part_forms
from .forms import EditPartForm
from .forms import EditPartAttachmentForm
from .forms import EditCategoryForm
from .forms import EditBomItemForm
from .forms import BomExportForm
from .forms import EditSupplierPartForm
from InvenTree.views import AjaxView, AjaxCreateView, AjaxUpdateView, AjaxDeleteView from InvenTree.views import AjaxView, AjaxCreateView, AjaxUpdateView, AjaxDeleteView
from InvenTree.views import QRCodeView from InvenTree.views import QRCodeView
@ -60,7 +53,7 @@ class PartAttachmentCreate(AjaxCreateView):
- The view only makes sense if a Part object is passed to it - The view only makes sense if a Part object is passed to it
""" """
model = PartAttachment model = PartAttachment
form_class = EditPartAttachmentForm form_class = part_forms.EditPartAttachmentForm
ajax_form_title = "Add part attachment" ajax_form_title = "Add part attachment"
ajax_template_name = "modal_form.html" ajax_template_name = "modal_form.html"
@ -99,7 +92,7 @@ class PartAttachmentCreate(AjaxCreateView):
class PartAttachmentEdit(AjaxUpdateView): class PartAttachmentEdit(AjaxUpdateView):
""" View for editing a PartAttachment object """ """ View for editing a PartAttachment object """
model = PartAttachment model = PartAttachment
form_class = EditPartAttachmentForm form_class = part_forms.EditPartAttachmentForm
ajax_template_name = 'modal_form.html' ajax_template_name = 'modal_form.html'
ajax_form_title = 'Edit attachment' ajax_form_title = 'Edit attachment'
@ -139,7 +132,7 @@ class PartCreate(AjaxCreateView):
- Copy an existing Part - Copy an existing Part
""" """
model = Part model = Part
form_class = EditPartForm form_class = part_forms.EditPartForm
ajax_form_title = 'Create new part' ajax_form_title = 'Create new part'
ajax_template_name = 'modal_form.html' ajax_template_name = 'modal_form.html'
@ -260,7 +253,7 @@ class PartImage(AjaxUpdateView):
model = Part model = Part
ajax_template_name = 'modal_form.html' ajax_template_name = 'modal_form.html'
ajax_form_title = 'Upload Part Image' ajax_form_title = 'Upload Part Image'
form_class = PartImageForm form_class = part_forms.PartImageForm
def get_data(self): def get_data(self):
return { return {
@ -272,7 +265,7 @@ class PartEdit(AjaxUpdateView):
""" View for editing Part object """ """ View for editing Part object """
model = Part model = Part
form_class = EditPartForm form_class = part_forms.EditPartForm
ajax_template_name = 'modal_form.html' ajax_template_name = 'modal_form.html'
ajax_form_title = 'Edit Part Properties' ajax_form_title = 'Edit Part Properties'
context_object_name = 'part' context_object_name = 'part'
@ -298,7 +291,7 @@ class BomExport(AjaxView):
ajax_form_title = 'Export BOM' ajax_form_title = 'Export BOM'
ajax_template_name = 'part/bom_export.html' ajax_template_name = 'part/bom_export.html'
context_object_name = 'part' context_object_name = 'part'
form_class = BomExportForm form_class = part_forms.BomExportForm
def get_object(self): def get_object(self):
return get_object_or_404(Part, pk=self.kwargs['pk']) return get_object_or_404(Part, pk=self.kwargs['pk'])
@ -396,7 +389,7 @@ class CategoryDetail(DetailView):
class CategoryEdit(AjaxUpdateView): class CategoryEdit(AjaxUpdateView):
""" Update view to edit a PartCategory """ """ Update view to edit a PartCategory """
model = PartCategory model = PartCategory
form_class = EditCategoryForm form_class = part_forms.EditCategoryForm
ajax_template_name = 'modal_form.html' ajax_template_name = 'modal_form.html'
ajax_form_title = 'Edit Part Category' ajax_form_title = 'Edit Part Category'
@ -449,7 +442,7 @@ class CategoryCreate(AjaxCreateView):
ajax_form_action = reverse_lazy('category-create') ajax_form_action = reverse_lazy('category-create')
ajax_form_title = 'Create new part category' ajax_form_title = 'Create new part category'
ajax_template_name = 'modal_form.html' ajax_template_name = 'modal_form.html'
form_class = EditCategoryForm form_class = part_forms.EditCategoryForm
def get_context_data(self, **kwargs): def get_context_data(self, **kwargs):
""" Add extra context data to template. """ Add extra context data to template.
@ -496,7 +489,7 @@ class BomItemDetail(DetailView):
class BomItemCreate(AjaxCreateView): class BomItemCreate(AjaxCreateView):
""" Create view for making a new BomItem object """ """ Create view for making a new BomItem object """
model = BomItem model = BomItem
form_class = EditBomItemForm form_class = part_forms.EditBomItemForm
ajax_template_name = 'modal_form.html' ajax_template_name = 'modal_form.html'
ajax_form_title = 'Create BOM item' ajax_form_title = 'Create BOM item'
@ -554,7 +547,7 @@ class BomItemEdit(AjaxUpdateView):
""" Update view for editing BomItem """ """ Update view for editing BomItem """
model = BomItem model = BomItem
form_class = EditBomItemForm form_class = part_forms.EditBomItemForm
ajax_template_name = 'modal_form.html' ajax_template_name = 'modal_form.html'
ajax_form_title = 'Edit BOM item' ajax_form_title = 'Edit BOM item'
@ -580,7 +573,7 @@ class SupplierPartEdit(AjaxUpdateView):
model = SupplierPart model = SupplierPart
context_object_name = 'part' context_object_name = 'part'
form_class = EditSupplierPartForm form_class = part_forms.EditSupplierPartForm
ajax_template_name = 'modal_form.html' ajax_template_name = 'modal_form.html'
ajax_form_title = 'Edit Supplier Part' ajax_form_title = 'Edit Supplier Part'
@ -589,7 +582,7 @@ class SupplierPartCreate(AjaxCreateView):
""" Create view for making new SupplierPart """ """ Create view for making new SupplierPart """
model = SupplierPart model = SupplierPart
form_class = EditSupplierPartForm form_class = part_forms.EditSupplierPartForm
ajax_template_name = 'modal_form.html' ajax_template_name = 'modal_form.html'
ajax_form_title = 'Create new Supplier Part' ajax_form_title = 'Create new Supplier Part'
context_object_name = 'part' context_object_name = 'part'