Add a lot of stuff.. Can't remember

shimatta_kenkyusho/parts/views.py

@@ -15,7 +15,7 @@ from .models import Storage, Stock, Component, Distributor, Manufacturer, Packag
 from .qr_parser import QrCodeValidator
 from django.core.paginator import Paginator
 from django.core.exceptions import ValidationError
-from .forms import MyTestForm, AddSubStorageForm, DeleteStockForm, EditWatermarkForm, EditStockAmountForm, AddStockForm
+from .forms import MyTestForm, AddSubStorageForm, DeleteStockForm, EditWatermarkForm, EditStockAmountForm, AddStockForm, EditComponentForm
 from django.db.models import Q
 from django.db.models.functions import Lower
 import uuid
@@ -233,6 +233,10 @@ class StockViewDetail(LoginRequiredMixin, BaseTemplateMixin, DetailView):
         if search is None or search == '':
             return stocks_in_storage
 
+        if search.startswith('[comp_uuid]'):
+            search = search.replace('[comp_uuid]', '')
+
+
         # Check if the searhc equals a UUID
         test_uuid = None
         try:
@@ -391,11 +395,31 @@ class ComponentDetailView(LoginRequiredMixin, BaseTemplateMixin, DetailView):
         self.base_title = 'Component / '+self.object.name
         context = super().get_context_data(**kwargs)
         context['component'] = self.object
-
-
-
+        context['edit_form'] = EditComponentForm(instance=self.object)
         return context
 
+    def handle_submit_edit_post(self, request, **kwargs):
+        form_error = False
+
+        form = EditComponentForm(instance=self.object, data=request.POST, files=request.FILES)
+        if form.is_valid():
+            form.save()
+        else:
+            print("Error")
+            form_error = True
+
+        context = self.get_context_data(**kwargs)
+        if form_error:
+            context['edit_form'] = form
+        return self.render_to_response(context)
+
+    def post(self, request, *args, **kwargs):
+        self.object = self.get_object()
+        if 'submit-edit-comp' in request.POST:
+            return self.handle_submit_edit_post(request, **kwargs)
+        
+        super().post(request, *args, **kwargs)
+
 class PackageDetailView(LoginRequiredMixin, BaseTemplateMixin, DetailView):
     template_name = 'parts/packages-detail.html'
     model = Package