Merge pull request 'issue/16-docker-uid-gid' (#28) from issue/16-docker-uid-gid into develop

Reviewed-on: #28

.dockerignore

@@ -0,0 +1,2 @@
+start_server.sh
+run/*

.env.example

@@ -0,0 +1,38 @@
+####################################################################################################
+# Example configuration. Must be edited and copied to ".env" next to the compose.yaml
+####################################################################################################
+
+# User id to use for the web application. This determines the user id, the media and static files are written to the volumes.
+# Make sure the user has rw access to these directories.
+DJANGO_USER_ID=1000
+
+# Group id to use for the web application
+DJANGO_USER_GID=1000
+
+# Path to to mount as the directory for static data. Must be served by a webserver on the /static path
+DJANGO_STATIC_VOL=/path/to/static/root
+
+# Path to the media root. Must be served by a webserver on the media URL
+DJANGO_MEDIA_VOL=/path/to/media/root
+
+# folder for DB
+PGDATA_VOL=/path/to/pgdata 
+
+# Port to serve the App
+PORT=8000
+
+# Secret Key. Must be edited before deployment
+DJANGO_SECRET_KEY=534hj5jgh4365ghj35jgh245jgh24
+
+# Allowed host to be accessed. Currently it is only possible to specify a single URL
+DJANGO_ALLOWED_HOST=lab.example.com
+
+# Media URL for images and other content
+DJANGO_MEDIA_URL=media.lab.example.com/
+
+# DO NOT SET DEBUG MODE IN PRODUCTION
+# DJANGO_FORCE_DEV_MODE=True
+
+# Set this password if you want to use a custom postgres password. The db should be confined inside the docker network.
+# Using the standard PW is therefore not a problem
+# DJANGO_POSTGRESQL_PW=myfancynewpassword123donotsharemewithanyone

.gitignore

@@ -125,3 +125,6 @@ dmypy.json
 
 
 myenv/*
+
+run/*
+.env

.vscode/launch.json

@@ -6,10 +6,9 @@
     "configurations": [
         {
             "name": "Python: Django",
-            "type": "python",
+            "type": "debugpy",
             "request": "launch",
-            "pythonPath": "${workspaceFolder}/../myenv/bin/python",
+            "program": "${workspaceFolder}/shimatta_kenkyusho/manage.py",
-            "program": "${workspaceFolder}/manage.py",
             "args": [
                 "runserver",
                 "0.0.0.0:8000"

Dockerfile

@@ -0,0 +1,6 @@
+FROM alpine:latest
+RUN apk add --no-cache python3 py3-pip python3-dev py3-setuptools gcc python3-dev jpeg-dev zlib-dev musl-dev py3-gunicorn curl
+COPY . /home/shimatta/kenkyusho
+WORKDIR /home/shimatta/kenkyusho
+RUN python3 -m venv /home/shimatta/kenkyusho/.venv && . /home/shimatta/kenkyusho/.venv/bin/activate && pip install -r requirements.txt
+ENTRYPOINT ["/home/shimatta/kenkyusho/entrypoint.sh"]

compose.yaml

@@ -0,0 +1,54 @@
+x-op-restart-policy: &restart_policy
+  restart: unless-stopped
+
+services:
+  shimatta-kenkyusho-web:
+    <<: *restart_policy
+    build: .
+    user: "${DJANGO_USER_ID}:${DJANGO_USER_GID}"
+    volumes:
+      - "${DJANGO_STATIC_VOL:-./run/static}:/var/static"
+      - "${DJANGO_MEDIA_VOL:-./run/media}:/var/media"
+    environment:
+      DJANGO_POSTGRESQL_PW: "${DJANGO_POSTGRESQL_PW:-p4ssw0rd}"
+      DJANGO_POSTGRESQL_USER: "postgres"
+      DJANGO_SECRET_KEY: "${DJANGO_SECRET_KEY}"
+      DJANGO_ALLOWED_HOST: "${DJANGO_ALLOWED_HOST}"
+      DJANGO_STATIC_ROOT: "/var/static"
+      DJANGO_MEDIA_URL: "${DJANGO_MEDIA_URL}"
+      DJANGO_MEDIA_ROOT: "/var/media"
+      DJANGO_POSTGRESQL_SOCKET: "shimatta-kenkyusho-db"
+      DJANGO_POSTGRESQL_PORT: "5432"
+      DJANGO_FORCE_DEV_MODE: ${DJANGO_FORCE_DEV_MODE:-False}
+    ports:
+      - "${PORT}:8000"
+    networks:
+      - backendnet
+    depends_on:
+      shimatta-kenkyusho-db:
+        condition: service_healthy
+    healthcheck:
+      test: ["CMD-SHELL", "curl -f localhost:8000/healthcheck"]
+      interval: 5s
+      timeout: 5s
+      retries: 5
+      start_period: 30s
+
+  shimatta-kenkyusho-db:
+    <<: *restart_policy
+    image: postgres:16.5-alpine
+    environment:
+      POSTGRES_PASSWORD: "${DJANGO_POSTGRESQL_PW:-p4ssw0rd}"
+      POSTGRES_DB: "shimatta_kenkyusho"
+    volumes:
+      - "${PGDATA_VOL:-./run/pgdata}:/var/lib/postgresql/data"
+    networks:
+      - backendnet
+    healthcheck:
+      test: ["CMD-SHELL", "pg_isready -U postgres"]
+      interval: 5s
+      timeout: 5s
+      retries: 5
+
+networks:
+  backendnet:

entrypoint.sh

@@ -0,0 +1,8 @@
+#!/bin/sh
+source /home/shimatta/kenkyusho/.venv/bin/activate
+cd /home/shimatta/kenkyusho/shimatta_kenkyusho
+python manage.py migrate --settings shimatta_kenkyusho.settings_production
+python manage.py collectstatic --settings shimatta_kenkyusho.settings_production --noinput
+python manage.py create_kenkyusho_admin_user --settings shimatta_kenkyusho.settings_production
+
+gunicorn -w 4 --bind 0.0.0.0:8000 shimatta_kenkyusho.wsgi:application

entrypoint_self_hosted.sh

@@ -0,0 +1,7 @@
+#!/bin/sh
+source /home/shimatta/kenkyusho/.venv/bin/activate
+cd /home/shimatta/kenkyusho/shimatta_kenkyusho
+python manage.py migrate --settings shimatta_kenkyusho.settings_production
+python manage.py create_kenkyusho_admin_user --settings shimatta_kenkyusho.settings_production
+
+python manage.py runserver 0.0.0.0:8000 --settings shimatta_kenkyusho.settings_production

requirements.txt

@@ -1,25 +1,36 @@
-asgiref==3.4.1
+annotated-types==0.7.0
+asgiref==3.8.1
 astroid==2.6.5
-crispy-bootstrap5==0.6
+certifi==2024.8.30
-Django==3.2.5
+charset-normalizer==3.4.0
-django-crispy-forms==1.13.0
+crispy-bootstrap5==2024.10
+Django==5.1.3
+django-crispy-forms==2.3
 django-filter==2.4.0
-django-qr-code==2.2.0
+django-qr-code==4.1.0
 django-rest-framework==0.1.0
-django-tex==1.1.9.post1
+django-tex==1.1.10
-djangorestframework==3.12.4
+djangorestframework==3.15.2
+gunicorn==21.2.0
+idna==3.10
 isort==5.9.3
 Jinja2==3.0.1
 lazy-object-proxy==1.6.0
 MarkupSafe==2.0.1
 mccabe==0.6.1
+packaging==24.2
 Pillow==8.3.1
-psycopg2==2.9.1
+pipdeptree==2.23.4
+psycopg2-binary==2.9.9
+pydantic==2.9.2
+pydantic_core==2.23.4
 pylint==2.9.6
-pytz==2021.1
+requests==2.32.3
-qrcode==7.2
+segno==1.6.1
-segno==1.3.3
+setuptools==75.3.0
-six==1.16.0
 sqlparse==0.4.1
 toml==0.10.2
+typing_extensions==4.12.2
+tzdata==2024.2
+urllib3==2.2.3
 wrapt==1.12.1

shimatta_kenkyusho/api/serializers.py

@@ -1,11 +1,12 @@
-from django.contrib.auth.models import User, Group
+from django.contrib.auth.models import Group
+from django.contrib.auth import get_user_model
 from rest_framework import serializers
 from parts import models as parts_models
 
 class UserSerializer(serializers.HyperlinkedModelSerializer):
     class Meta:
-        model = User
+        model = get_user_model()
-        fields = ['username', 'email', 'first_name', 'last_name', 'groups']
+        fields = ['id', 'username', 'email', 'first_name', 'last_name', 'groups']
 
 class GroupSerializer(serializers.HyperlinkedModelSerializer):
     class Meta:
@@ -23,22 +24,43 @@ class PackageSerializer(serializers.HyperlinkedModelSerializer):
         model = parts_models.Package
         fields = '__all__'
 
-class StorageSerializer(serializers.HyperlinkedModelSerializer):
+class ComponentParameterSerializer(serializers.HyperlinkedModelSerializer):
-    full_path = serializers.ReadOnlyField(source='get_full_path')
+    id = serializers.ReadOnlyField()
-
+    ro_parameter_type = serializers.ReadOnlyField(source='parameter_type.parameter_name')
     class Meta:
-        model = parts_models.Storage
+        model = parts_models.ComponentParameter
-        fields = ['url', 'id', 'name', 'parent_storage', 'responsible', 'full_path']
+        fields = '__all__'
+
+class ComponentDistributorNumSerializer(serializers.HyperlinkedModelSerializer):
+    id = serializers.ReadOnlyField()
+    class Meta:
+        model = parts_models.DistributorNum
+        fields = '__all__'
 
 class ComponentSerializer(serializers.HyperlinkedModelSerializer):
     package_data = PackageSerializerNoLink(source='package', read_only=True)
     ro_manufacturer_name = serializers.ReadOnlyField(source='manufacturer.name')
     ro_image = serializers.ReadOnlyField(source='get_resolved_image')
+    ro_component_type = serializers.ReadOnlyField(source='component_type.class_name')
+    ro_parameters = ComponentParameterSerializer(many=True, source='componentparameter_set', read_only=True)
+    ro_distributor_numbers = ComponentDistributorNumSerializer(many=True, source='distributornum_set', read_only=True)
 
     class Meta:
         model = parts_models.Component
-        fields = ['url', 'id', 'name', 'package_data', 'package', 'pref_distri', 'image', 'manufacturer', 'ro_manufacturer_name', 'ro_image']
+        fields = ['url',
-        
+                  'id',
+                  'name',
+                  'package_data',
+                  'package',
+                  'pref_distri',
+                  'image',
+                  'manufacturer',
+                  'component_type',
+                  'ro_manufacturer_name',
+                  'ro_image',
+                  'ro_component_type',
+                  'ro_parameters',
+                  'ro_distributor_numbers']
 
 class StockSerializer(serializers.HyperlinkedModelSerializer):
     id = serializers.ReadOnlyField()
@@ -50,6 +72,22 @@ class StockSerializer(serializers.HyperlinkedModelSerializer):
         model = parts_models.Stock
         fields = '__all__'
 
+class StockSerializerExpandComponent(StockSerializer):
+    ro_component = ComponentSerializer(read_only=True, source='component')
+
+class StorageSerializer(serializers.HyperlinkedModelSerializer):
+    full_path = serializers.ReadOnlyField(source='get_full_path')
+
+    class Meta:
+        model = parts_models.Storage
+        fields = ['url', 'id', 'name', 'verbose_name', 'parent_storage', 'responsible', 'template', 'full_path']
+
+class StorageSerializerStocksExpanded(StorageSerializer):
+    ro_stocks = StockSerializerExpandComponent(many=True, read_only=True, source='stock_set')
+
+    class Meta(StorageSerializer.Meta):
+        fields = StorageSerializer.Meta.fields + ['ro_stocks']
+
 class DistributorSerializer(serializers.HyperlinkedModelSerializer):
     id = serializers.ReadOnlyField()
     class Meta:

shimatta_kenkyusho/api/urls.py

@@ -1,13 +1,15 @@
-from django.urls import include, path
+from django.urls import include, path, re_path
 from rest_framework import routers
 from .views import *
-from django.conf.urls import url
 
 router = routers.DefaultRouter()
 router.register(r'users', UserViewSet)
 router.register(r'groups', GroupViewSet)
 router.register(r'parts/storages', PartsStorageViewSet)
+router.register(r'parts/storage_templates', PartsStorageTemplatesViewSet, basename='storage-template')
 router.register(r'parts/components', PartsComponentViewSet)
+router.register(r'parts/component-parameters', PartsComponentParameterViewSet)
+router.register(r'parts/component-distributor-numbers', PartsComponentDistributorNumViewSet)
 router.register(r'parts/stocks', PartsStockViewSet)
 router.register(r'parts/packages', PartsPackageViewSet)
 router.register(r'parts/distributors', PartsDistributorviewSet)
@@ -17,6 +19,6 @@ router.register(r'parts/component-param-types', PartsComponentParameterTypeViewS
 
 urlpatterns = [
     path('', include(router.urls)),
-    url(r'^token-auth/', ObtainExpiringAuthToken.as_view()),
+    re_path(r'^token-auth/', ObtainExpiringAuthToken.as_view()),
-	url(r'^token-logout/', TokenLogout.as_view()),
+	re_path(r'^token-logout/', TokenLogout.as_view()),
 ]

shimatta_kenkyusho/api/views.py

@@ -1,5 +1,6 @@
 from django.shortcuts import render
-from django.contrib.auth.models import User, Group
+from django.contrib.auth.models import Group
+from django.contrib.auth import get_user_model
 from django.core.exceptions import ObjectDoesNotExist
 from rest_framework import viewsets, status
 from rest_framework import permissions
@@ -25,7 +26,7 @@ class UserViewSet(viewsets.ReadOnlyModelViewSet):
     """
     API endpoint that allows users to be viewed or edited.
     """
-    queryset = User.objects.all()
+    queryset = get_user_model().objects.all()
     serializer_class = UserSerializer
     permission_classes = [permissions.IsAuthenticated]
     filter_backends = [filters.SearchFilter]
@@ -41,11 +42,24 @@ class GroupViewSet(viewsets.ReadOnlyModelViewSet):
 
 class PartsStorageViewSet(viewsets.ModelViewSet):
     queryset = parts_models.Storage.objects.all()
-    serializer_class = StorageSerializer
     permission_classes = [permissions.DjangoModelPermissions]
     filter_backends = [django_filters.rest_framework.DjangoFilterBackend]
+    search_fields = ['id', 'name', 'parent_storage']
     filterset_fields = ['id', 'name', 'parent_storage']
 
+    def get_serializer_class(self):
+        if self.request.GET.get('expand_stocks'):
+            return StorageSerializerStocksExpanded
+        return StorageSerializer
+
+class PartsStorageTemplatesViewSet(viewsets.ReadOnlyModelViewSet):
+    queryset = parts_models.Storage.objects.filter(is_template=True)
+    serializer_class = StorageSerializer
+    permission_classes = [permissions.IsAuthenticated]
+    filter_backends = [filters.SearchFilter]
+    search_fields = ['id', 'name']
+    filterset_fields = ['id', 'name']
+
 class PartsComponentViewSet(viewsets.ModelViewSet):
     queryset = parts_models.Component.objects.all()
     serializer_class = ComponentSerializer
@@ -54,6 +68,22 @@ class PartsComponentViewSet(viewsets.ModelViewSet):
     search_fields = ['id', 'name', 'package__name', 'manufacturer__name']
     filterset_fields = ['id', 'name']
 
+class PartsComponentParameterViewSet(viewsets.ModelViewSet):
+    queryset = parts_models.ComponentParameter.objects.all()
+    serializer_class = ComponentParameterSerializer
+    permission_classes = [permissions.DjangoModelPermissions]
+    filter_backends = [filters.SearchFilter, django_filters.rest_framework.DjangoFilterBackend]
+    search_fields = ['id', 'parameter_type__parameter_name']
+    filterset_fields = ['id']
+
+class PartsComponentDistributorNumViewSet(viewsets.ModelViewSet):
+    queryset = parts_models.DistributorNum.objects.all()
+    serializer_class = ComponentDistributorNumSerializer
+    permission_classes = [permissions.DjangoModelPermissions]
+    filter_backends = [filters.SearchFilter, django_filters.rest_framework.DjangoFilterBackend]
+    search_fields = ['id', 'distributor', 'distributor_part_number']
+    filterset_fields = ['id']
+
 class PartsComponentTypeViewSet(viewsets.ModelViewSet):
     queryset = parts_models.ComponentType.objects.all()
     serializer_class = ComponentTypeSerializer

shimatta_kenkyusho/parts/forms.py

@@ -1,23 +1,27 @@
 from django import forms
-from django.forms import widgets
+from django.contrib.auth import get_user_model
 from django.core.exceptions import ValidationError
+from django.forms import widgets
 from parts import models as parts_models
 from shimatta_modules.EngineeringNumberConverter import EngineeringNumberConverter
 import uuid
 from django.urls import reverse
+from .qr_parser import QrCodeValidator
+
 
 from crispy_forms.helper import FormHelper
-from crispy_forms.layout import Layout, Fieldset, Row, Column
+from crispy_forms.layout import Layout, Row, Column
 
 class AutoCompleteWidget(widgets.Input):
     template_name = 'widgets/autocomplete-foreign-key.html'
 
-    def __init__(self, api_search_url, image_field_name, foreign_model, name_field_name, *args, **kwargs):
+    def __init__(self, api_search_url, image_field_name, foreign_model, name_field_name, prepend=None, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.image_field_name = image_field_name
         self.foreign_model = foreign_model
         self.api_search_url = api_search_url
         self.name_field_name = name_field_name
+        self.prepend = prepend
 
     def get_context(self, name, value, attrs):
         context = super().get_context(name, value, attrs)
@@ -47,14 +51,25 @@ class AutoCompleteWidget(widgets.Input):
             'current_instance': instance,
             'image': image,
             'name_field_name': self.name_field_name,
+            'prepend': self.prepend,
             'name': display_name,
         } 
         return context
 
 class AutocompleteForeingKeyField(forms.UUIDField):
-    def __init__(self, foreign_model=None, api_search_url=None, image_field_name='image', name_field_name='name', **kwargs):
+    def __init__(self,
+                 foreign_model=None,
+                 api_search_url=None,
+                 image_field_name='image',
+                 name_field_name='name',
+                 prepend=None,
+                 **kwargs):
         super().__init__(**kwargs)
-        self.widget = AutoCompleteWidget(api_search_url, image_field_name, foreign_model, name_field_name)
+        self.widget = AutoCompleteWidget(api_search_url,
+                                         image_field_name,
+                                         foreign_model,
+                                         name_field_name,
+                                         prepend)
         self.foreign_model = foreign_model
 
     
@@ -74,13 +89,25 @@ class AutocompleteForeingKeyField(forms.UUIDField):
             raise ValidationError('Given element does not exist')
         return obj
 
-
 class MyTestForm(forms.Form):
     pass
 
-class AddSubStorageForm(forms.Form):
+class ChangeStorageForm(forms.Form):
-    storage_name = forms.CharField(label="storage_name", initial='')
+    storage_name = forms.CharField(label="Name", initial='')
-    responsible = forms.CharField(label='responsible_user')
+    verbose_name = forms.CharField(label="Verbose Name", initial='', required=False)
+    responsible = AutocompleteForeingKeyField(api_search_url='user-list',
+                                              image_field_name=None,
+                                              name_field_name='username',
+                                              foreign_model=get_user_model(),
+                                              prepend='@')
+
+    is_template = forms.BooleanField(label='is_template', required=False)
+
+class AddSubStorageForm(ChangeStorageForm):
+    template = AutocompleteForeingKeyField(api_search_url='storage-template-list',
+                                           image_field_name=None,
+                                           foreign_model=parts_models.Storage,
+                                           required=False)
 
 class DeleteStockForm(forms.Form):
     stock_uuid = forms.UUIDField()
@@ -188,6 +215,9 @@ class ComponentForm(forms.ModelForm):
         model = parts_models.Component
         fields = '__all__'
 
+class ImportComponentForm(forms.Form):
+    csv_file = forms.FileField(label="CSV File")
+
 class PackageForm(forms.ModelForm):
     class Meta:
         model = parts_models.Package
@@ -309,3 +339,12 @@ class ComponentParameterCreateForm(forms.Form):
             text_value = ''
             value = self.cleaned_data['number_value']
         parts_models.ComponentParameter.objects.create(parameter_type=param_type, component=component, value=value, text_value=text_value)
+
+class QrSearchForm(forms.Form):
+    my_qr_validator = QrCodeValidator()
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        
+
+    qr_search = forms.CharField(label='qr_search', validators=[my_qr_validator])

shimatta_kenkyusho/parts/management/commands/create_kenkyusho_admin_user.py

@@ -0,0 +1,23 @@
+from django.core.management.base import BaseCommand, CommandParser
+from django.contrib.auth import get_user_model
+
+class Command(BaseCommand):
+	help = "Create a default superuser if no superuser is already present. This aids automatic deployment inside a container."
+
+	def add_arguments(self, parser: CommandParser):
+		parser.add_argument('--user',
+			help='Username to create if no admin account is present',
+			default='admin')
+		parser.add_argument('--password',
+			help='Password to set for newly created user. Ignored, if any admin user is already present',
+			default='admin')
+
+	def handle(self, *args, **options):
+		User = get_user_model()
+
+		# Query if there is any admin user
+		if not User.objects.filter(is_superuser=True).exists():
+			self.stdout.write(f'No superuser present. Creating {options['user']} with supplied password')
+			User.objects.create_superuser(username=options['user'], password=options['password'])
+		else:
+			self.stdout.write('At least one superuser already exists. Skipping superuser creation')

shimatta_kenkyusho/parts/migrations/0011_auto_20241110_1242.py

@@ -0,0 +1,24 @@
+# Generated by Django 3.2.5 on 2024-11-10 12:42
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('parts', '0010_auto_20220103_1606'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='storage',
+            name='is_template',
+            field=models.BooleanField(default=False),
+        ),
+        migrations.AddField(
+            model_name='storage',
+            name='template',
+            field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, related_name='template_of', to='parts.storage'),
+        ),
+    ]

shimatta_kenkyusho/parts/migrations/0012_storage_verbose_name.py

@@ -0,0 +1,18 @@
+# Generated by Django 3.2.5 on 2024-11-16 11:37
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('parts', '0011_auto_20241110_1242'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='storage',
+            name='verbose_name',
+            field=models.CharField(blank=True, max_length=100, null=True),
+        ),
+    ]

shimatta_kenkyusho/parts/models.py

@@ -1,7 +1,7 @@
 from django.db import models
 from shimatta_modules import RandomFileName
 from django.db.models import F, Sum
-from django.contrib.auth.models import User as AuthUser
+from django.contrib.auth import get_user_model
 from django.core.exceptions import ValidationError
 from django.core.validators import RegexValidator
 from django.dispatch import receiver
@@ -62,8 +62,17 @@ class Storage(models.Model):
 
 	id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False, unique=True)
 	name = models.CharField(max_length=100, validators=[storage_name_validator])
+	verbose_name = models.CharField(max_length=100, null=True, blank=True)
 	parent_storage = models.ForeignKey('self', on_delete=models.CASCADE, blank=True, null=True)
-	responsible = models.ForeignKey(AuthUser, on_delete=models.SET_NULL, blank=True, null=True)
+	responsible = models.ForeignKey(get_user_model(), on_delete=models.SET_NULL, blank=True, null=True)
+
+	# allow storages to be templates which can be selected when adding new storages
+	is_template = models.BooleanField(default=False)
+	template = models.ForeignKey('self',
+								on_delete=models.SET_NULL,
+								blank=True,
+								null=True,
+								related_name='template_of')
 
 	def get_path_components(self):
 		chain = []
@@ -113,6 +122,29 @@ class Storage(models.Model):
 			sum = 0
 		return sum
 
+	@classmethod
+	def from_path(cls, path, root_storage=None):
+		'''
+		Get the storage object described by its complete path or the sub-path
+		from the passed root_storage uuid
+		'''
+		parts = path.split('/')
+
+		# assemble filter query
+		filter_dict = {}
+
+		layer = 0
+		for part in parts[::-1]:
+			filter_dict[f'{"parent_storage__" * layer}name'] = part
+			layer += 1
+
+		if root_storage:
+			filter_dict[f'{"parent_storage__" * layer}id'] = root_storage
+		else:
+			filter_dict[f'{"parent_storage__" * layer}isnull'] = True
+		obj = cls.objects.get(**filter_dict)
+		return obj
+
 	def save(self, *args, **kwargs):
 		self.validate_unique()
 		super(Storage, self).save(*args, **kwargs)
@@ -288,7 +320,7 @@ class DistributorNum(models.Model):
 
 class QrPrintJob(models.Model):
 	id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False, unique=True)
-	user = models.ForeignKey(AuthUser, on_delete=models.CASCADE, null=False, blank=False)
+	user = models.ForeignKey(get_user_model(), on_delete=models.CASCADE, null=False, blank=False)
 	qrdata = models.CharField(max_length=256, blank=True, null=False)
 	text = models.TextField(max_length=512, blank=True, null=False)
 	print_count = models.PositiveIntegerField(default=1, validators=[MinValueValidator(0), MaxValueValidator(32)])
@@ -336,3 +368,19 @@ def auto_delete_file_on_change(sender, instance, **kwargs):
             return True
         if os.path.isfile(old_file.path):
             os.remove(old_file.path)
+
+@receiver(models.signals.post_save, sender=Storage)
+def auto_apply_template_structure(sender, instance, created, **kwargs):
+    """
+    Add the sub-storages from the template.
+    If there are nested sub-storages these will be added when the sub-storages
+    are created automatically.
+    """
+    if created:
+        if instance.template:
+            for sub_storage in instance.template.storage_set.all():
+                Storage.objects.create(name=sub_storage.name,
+                                       parent_storage=instance,
+                                       responsible=instance.responsible,
+                                       is_template=False,
+                                       template=sub_storage)

shimatta_kenkyusho/parts/urls.py

@@ -14,6 +14,7 @@ urlpatterns = [
     path('components/<slug:uuid>/', parts_views.ComponentDetailView.as_view(), name='parts-components-detail'),
     path('packages/<slug:uuid>/', parts_views.PackageDetailView.as_view(), name='parts-packages-detail'),
     path('distributors/<slug:uuid>/', parts_views.DistributorDetailView.as_view(), name='parts-distributors-detail'),
-    path('manufacturers/', parts_views.ManufacturersViewSet.as_view(), name='parts-manufacturers'),
+    path('manufacturers/', parts_views.ManufacturersView.as_view(), name='parts-manufacturers'),
-    path("manufacturers/<slug:uuid>/", parts_views.ManufacturerDetailViewSet.as_view(), name='parts-manufacturers-detail'),
+    path("manufacturers/<slug:uuid>/", parts_views.ManufacturerDetailView.as_view(), name='parts-manufacturers-detail'),
+    path("healthcheck/", parts_views.health_check_view, name='parts-health-check'),
 ]

shimatta_kenkyusho/parts/views.py

@@ -1,946 +0,0 @@
-from django.shortcuts import render, redirect
-from django.urls import resolve, reverse
-from django.contrib.auth import logout, login
-from django.contrib.auth.models import User
-from django.http import HttpResponse
-from .navbar import NavBar
-from django.contrib.auth.forms import AuthenticationForm as AuthForm
-from django.contrib.auth.forms import PasswordChangeForm
-from django.contrib.auth import update_session_auth_hash
-from django.views import View
-import django.forms as forms
-from django.views.generic import TemplateView, DetailView
-from django.contrib.auth.mixins import LoginRequiredMixin, PermissionRequiredMixin
-from .models import Storage, Stock, Component, Distributor, Manufacturer, Package, ComponentParameter, ComponentParameterType, DistributorNum
-from .qr_parser import QrCodeValidator
-from django.core.paginator import Paginator
-from django.core.exceptions import ValidationError
-from django.db import IntegrityError
-from django.db.models import ProtectedError
-from .forms import *
-from django.db.models import Q
-from django.db.models.functions import Lower
-from django.forms import formset_factory
-import uuid
-
-ParameterSearchFormSet = formset_factory(ComponentParameterSearchForm, extra=1)
-
-class QrSearchForm(forms.Form):
-    my_qr_validator = QrCodeValidator()
-
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
-        
-
-    qr_search = forms.CharField(label='qr_search', validators=[my_qr_validator])
-
-class BaseTemplateMixin(object):
-    navbar_selected = ''
-    base_title = ''
-
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-        base_context = {
-            'navbar': NavBar.get_navbar(self.navbar_selected, self.request.user),
-            'title': NavBar.get_brand()+' / '+ self.base_title,
-            'login_active': False,
-        }
-        context['base'] = base_context 
-        return context
-
-    def post(self, request, *args, **kwargs):
-        data = request.POST
-        if 'qr_search' not in data:
-            super().post(request, *args, **kwargs)
-
-        print('QR',data['qr_search'])
-        f = QrSearchForm(data)
-        if f.is_valid():
-            return redirect(f.my_qr_validator.get_redirect_url(f.cleaned_data['qr_search']))
-
-        return self.get(request)
-
-class ChangePasswordView(LoginRequiredMixin, BaseTemplateMixin, TemplateView):
-    template_name = 'parts/change-pw.html'
-    navbar_selected = 'Main'
-    base_title = 'Change Password'
-
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-        context['form'] = PasswordChangeForm(self.request.user)
-        return context
-
-    def post(self, request, *args, **kwargs):
-        if 'submit-change-pw' not in request.POST:
-            return super().post(request, *args, **kwargs)
-
-        form = PasswordChangeForm(request.user, data=request.POST)
-
-        if form.is_valid():
-            user = form.save()
-            update_session_auth_hash(request, user)
-            return redirect('parts-main')
-        else:
-            pass
-
-        context = self.get_context_data(**kwargs)
-        if form.errors:
-            context['form'] = form
-
-        return self.render_to_response(context)
-    
-
-class MainView(BaseTemplateMixin, TemplateView):
-    template_name = 'parts/main.html'
-    navbar_selected = 'Main'
-    base_title = 'Main'
-
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-        context['user'] = self.request.user
-        return context
-    
-    
-def logout_view(request):
-    logout(request)
-    return redirect('parts-main')
-
-def login_view(request):
-    base_context = {
-        'navbar': NavBar.get_navbar('Login', request.user),
-        'title': NavBar.get_brand()+' / '+'Login',
-        'login_active': True,
-    }
-
-    if request.user.is_authenticated:
-        next_param = request.GET.get('next')
-        if next_param is not None:
-            return redirect(next_param)
-        else:
-            return redirect('parts-main')
-
-    if request.method == 'POST':
-        form = AuthForm(data=request.POST)
-        if form.is_valid():
-            valid_user = form.get_user()
-            login(request, valid_user)
-            next_param = request.GET.get('next')
-            if next_param is not None:
-                return redirect(next_param)
-            else:
-                return redirect('parts-main')
-    else:
-        form = AuthForm()
-
-
-    context = {
-        'base': base_context,
-        'form': form,
-    }
-
-    return render(request, 'parts/login.html', context)
-
-# Create your views here.
-
-class ComponentView(LoginRequiredMixin, BaseTemplateMixin, TemplateView):
-    template_name = 'parts/components.html'
-    base_title = 'Components'
-    navbar_selected = 'Components'
-    default_page_size = 25
-
-    def get_component_query_set(self, search_string):
-        queryset = Component.objects.all()
-        
-        if not search_string:
-            return queryset
-
-        search_fragments = search_string.strip().split()
-        for search in search_fragments:
-            queryset = queryset.filter(Q(name__icontains = search) | Q(manufacturer__name__icontains = search) | Q(package__name__icontains = search))
-
-        return queryset
-
-    def get_component_queryset_from_advanced_search(self, cleaned_data):
-        queryset = Component.objects.all()
-
-        if cleaned_data['name']:
-            queryset = queryset.filter(Q(name__icontains=cleaned_data['name']))
-        if cleaned_data['package']:
-            queryset = queryset.filter(package=cleaned_data['package'])
-        if cleaned_data['package_pin_count']:
-            queryset = queryset.filter(package__pin_count=cleaned_data['package_pin_count'])
-        if cleaned_data['component_type']:
-            queryset = queryset.filter(component_type=cleaned_data['component_type'])
-        if cleaned_data['distributor_num']:
-            if cleaned_data['distributor']:
-                distri = cleaned_data['distributor']
-                queryset = queryset.filter(Q(distributornum__distributor_part_number__icontains=cleaned_data['distributor_num']) & Q(distributornum__distributor=distri))
-            else:
-                 queryset = queryset.filter(Q(distributornum__distributor_part_number__icontains=cleaned_data['distributor_num']))
-        if cleaned_data['manufacturer']:
-            queryset = queryset.filter(manufacturer=cleaned_data['manufacturer'])
-        return queryset
-
-    def get_context_data_int(self, advanced_search, parameter_formset : ParameterSearchFormSet, **kwargs):
-        context = super().get_context_data(**kwargs)
-
-        comp_page_num = self.request.GET.get('comp_page', default=1)
-
-        if advanced_search and parameter_formset:
-            search = None
-            context['advanced_search_shown'] = True
-            context['advanced_search_form'] = advanced_search
-            context['advanced_search_param_formset'] = parameter_formset
-            if advanced_search.is_valid():
-                paginator_queryset = self.get_component_queryset_from_advanced_search(advanced_search.cleaned_data)
-            else:
-                paginator_queryset = Component.objects.all()
-
-            if parameter_formset.is_valid():
-                # Process parameters
-                pass
-            
-        else:
-            search = self.request.GET.get('search', default=None)
-            paginator_queryset = self.get_component_query_set(search)
-
-        comp_paginator = Paginator(paginator_queryset, self.default_page_size)
-
-        context['components'] = comp_paginator.get_page(comp_page_num)
-        context['comp_form'] = ComponentForm()
-        context['search_string'] = search
-
-        if not parameter_formset:
-            context['advanced_search_param_formset'] = ParameterSearchFormSet()
-
-
-        if not advanced_search:
-            context['advanced_search_form'] = AdvancedComponentSearchForm(auto_id='adv_search_%s')
-
-        return context
-    
-    def get_context_data(self, **kwargs):
-        return self.get_context_data_int(advanced_search = None, parameter_formset=None, **kwargs)    
-    
-
-    def handle_new_component_post(self, request, open=False, **kwargs):
-        cform = ComponentForm(data=request.POST, files=request.FILES)
-        new_component = None
-        if cform.is_valid():
-            new_component = cform.save()
-
-        context = self.get_context_data(**kwargs)
-        if not cform.is_valid():
-            context['comp_form'] = cform
-
-        if open and new_component:
-            return redirect(reverse('parts-components-detail', kwargs={'uuid':new_component.id}))
-        return self.render_to_response(context)
-    
-    def handle_advanced_search_post(self, request, **kwargs):
-        
-        form = AdvancedComponentSearchForm(auto_id='adv_search_%s', data=request.POST)
-        param_formset = ParameterSearchFormSet(data=request.POST)
-
-        if form.is_valid():
-            print('Valid')
-
-        if param_formset.is_valid():
-            print('Formset is valid!')
-
-
-        context = self.get_context_data_int(form, param_formset, **kwargs)
-        return self.render_to_response(context)
-
-    def post(self, request, *args, **kwargs):
-        if 'submit-edit-component' in request.POST:
-            return self.handle_new_component_post(request, open=False, **kwargs)
-        elif 'submit-edit-component-open' in request.POST:
-            return self.handle_new_component_post(request, open=True, **kwargs)
-        elif 'submit-advanced-search' in request.POST:
-            return self.handle_advanced_search_post(request, **kwargs)
-        else:
-            return super().post(request, *args, **kwargs)
-    
-class PackageView(LoginRequiredMixin, BaseTemplateMixin, TemplateView):
-    template_name = 'parts/packages.html'
-    base_title = 'Packages'
-    navbar_selected = 'Packages'
-    default_page_size = 25
-
-    def search_packages(self, search):
-        qs = Package.objects.all()
-
-        if not search:
-            return qs
-        
-        search_fragments = search.strip().split()
-        for search in search_fragments:
-            if search.lower() == 'smd':
-                s_filter = Q(name__icontains = search) | Q(smd = True)
-            else:
-                try:
-                    pin_count = int(search)
-                    s_filter = Q(name__icontains = search) | Q(pin_count=pin_count)
-                except:
-                    s_filter = Q(name__icontains = search)
-
-            qs = qs.filter(s_filter)
-        return qs
-
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-        page_num = self.request.GET.get('page', default=1)
-        search_string = self.request.GET.get('search', default=None)
-        
-        package_queryset = self.search_packages(search_string)
-
-        paginator = Paginator(package_queryset, self.default_page_size)       
-
-
-        context['search_string'] = search_string
-        context['packages'] = paginator.get_page(page_num)
-        context['new_pkg_form'] = PackageForm()
-
-        return context
-    
-    def handle_add_new_package(self, request):
-
-        form = PackageForm(data=request.POST, files=request.FILES)
-        if form.is_valid():
-            form.save()
-
-        context = self.get_context_data()
-
-        if not form.is_valid():
-            context['new_pkg_form'] = form
-
-        return self.render_to_response(context)
-
-    def post(self, request, *args, **kwargs):
-        if 'submit-pkg-add-new' in request.POST:
-            return self.handle_add_new_package(request)
-
-        return super().post(request, *args, **kwargs)
-    
-
-class DistributorView(LoginRequiredMixin, BaseTemplateMixin, TemplateView):
-    template_name = 'parts/distributors.html'
-    base_title = 'Distributors'
-    navbar_selected = 'Distributors'
-    default_page_size = 25
-
-    def search_distributors(self, search):
-        qs = Distributor.objects.all()
-
-        if not search:
-            return qs
-        
-        search_fragments = search.strip().split()
-
-        for search in search_fragments:
-            qs = qs.filter(Q(name__icontains = search) | Q(website__icontains = search))
-        return qs
-
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-        page_num = self.request.GET.get('page', default=1)
-        search_string = self.request.GET.get('search', default=None)
-        
-        queryset = self.search_distributors(search_string)
-
-        paginator = Paginator(queryset, self.default_page_size)       
-
-
-        context['search_string'] = search_string
-        context['distributors'] = paginator.get_page(page_num)
-        context['new_distri_form'] = DistributorForm()
-
-        return context
-    
-    def handle_add_new_distributor(self, request):
-
-        form = DistributorForm(data=request.POST, files=request.FILES)
-        if form.is_valid():
-            form.save()
-
-        context = self.get_context_data()
-
-        if not form.is_valid():
-            context['new_distri_form'] = form
-
-        return self.render_to_response(context)
-
-    def post(self, request, *args, **kwargs):
-        if 'submit-distri-add-new' in request.POST:
-            return self.handle_add_new_distributor(request)
-
-        return super().post(request, *args, **kwargs)
-
-class StockView(LoginRequiredMixin, BaseTemplateMixin, TemplateView):
-    template_name = 'parts/stocks.html'
-    base_title = 'Stocks'
-    navbar_selected = 'Stocks'
-    default_pagination_size = 25
-
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-
-        storage_page = self.request.GET.get('storage_page')
-        if storage_page is None:
-            storage_page = 1
-        low_stock_page = self.request.GET.get('low_stock_page')
-        if low_stock_page is None:
-            low_stock_page = 1
-
-        storage_paginator = Paginator(Storage.objects.filter(parent_storage=None), self.default_pagination_size)
-        low_stock_paginator = Paginator(Stock.get_under_watermark(),
-            self.default_pagination_size)
-
-        context['low_stocks'] = low_stock_paginator.get_page(low_stock_page)
-        context['storages'] = storage_paginator.get_page(storage_page)
-        add_stor_form = AddSubStorageForm()
-        add_stor_form.fields['responsible'].initial = self.request.user.username
-        context['add_storage_form'] = add_stor_form
-        return context
-
-    def handle_add_storage(self, request, **kwargs):
-        return_invalid_form = False
-
-        f = AddSubStorageForm(data=request.POST)
-        if f.is_valid():
-            new_storage_name = f.cleaned_data['storage_name']
-            try: 
-                resp_user = User.objects.get(username=f.cleaned_data['responsible'])
-            except Exception as _:
-                resp_user = None
-                f.add_error('responsible', 'Invalid Responsible User')
-                return_invalid_form = True
-
-            if resp_user is not None:
-                try:
-                    Storage.objects.create(name=new_storage_name, responsible=resp_user, parent_storage=None)
-                except ValidationError as verr:
-                    return_invalid_form = True
-                    f.add_error('storage_name', ' .'.join(verr.messages))
-        else:
-            return_invalid_form = True
-
-        context = self.get_context_data(**kwargs)
-        if return_invalid_form:
-            context['add_storage_form'] = f
-
-        return self.render_to_response(context)
-    
-    def post(self, request, **kwargs):
-        if 'submit-add-storage' in request.POST:
-            return self.handle_add_storage(request, **kwargs)
-
-        return super().post(request, **kwargs)
-    
-class StockViewDetail(LoginRequiredMixin, BaseTemplateMixin, DetailView):
-    template_name = 'parts/stocks-detail.html'
-    model = Storage
-    pk_url_kwarg = 'uuid'
-    base_title = ''
-    navbar_selected = 'Stocks'
-    default_pagination_size = 8
-
-    def get_breadcrumbs(self):
-        crumbs = self.object.get_path_components()
-        # Reverse list and drop the last element of the reversed list
-        crumbs = crumbs[::-1][:-1]
-        return crumbs
-
-    def search_stock_queryset(self, search):
-        stocks_in_storage = Stock.objects.filter(storage=self.object).order_by(Lower('component__name'))
-        
-        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:
-            test_uuid = uuid.UUID(search)
-        except:
-            pass
-
-        if test_uuid is not None:
-            stocks_in_storage = stocks_in_storage.filter(Q(component__id = test_uuid) | Q(id= test_uuid))
-        else:
-            stocks_in_storage = stocks_in_storage.filter(Q(component__name__icontains = search) |
-                Q(component__package__name__icontains = search) |
-                Q(component__manufacturer__name__icontains = search))
-        
-        return stocks_in_storage
-
-    def get_context_data(self, **kwargs):
-        self.base_title = 'Stocks / ' + self.object.name
-        context = super().get_context_data(**kwargs)
-        context['breadcrumbs'] = self.get_breadcrumbs()
-
-        storage_page = self.request.GET.get('storage_page', default=1)
-        storage_paginator = Paginator(Storage.objects.filter(parent_storage=self.object), self.default_pagination_size)
-        stock_search_input = self.request.GET.get('search')
-
-        componente_stock_page = self.request.GET.get('stock_page', default=1)
-        
-        stock_paginator = Paginator(self.search_stock_queryset(stock_search_input), self.default_pagination_size)
-
-        context['storages'] = storage_paginator.get_page(storage_page)
-        stocks = stock_paginator.get_page(componente_stock_page)
-        context['stocks'] = stocks
-        context['stock_search'] = stock_search_input
-        add_storage_form = AddSubStorageForm()
-        add_storage_form.fields['responsible'].initial = self.request.user.username
-        context['add_storage_form'] = add_storage_form
-        context['delete_storage_error'] = None
-        context['add_stock_form'] = AddStockForm()
-
-        return context
-
-    def handle_add_storage_post(self, request, **kwargs):
-        f = AddSubStorageForm(data=request.POST)
-        if f.is_valid():
-            sub_name = f.cleaned_data['storage_name']
-            try:
-                user = User.objects.get(username=f.cleaned_data['responsible'])
-                try:
-                    Storage.objects.create(name=sub_name, parent_storage=self.object, responsible=user)
-                except ValidationError as v_err:
-                    f.add_error('storage_name', '. '.join(v_err.messages))
-            except:
-                f.add_error('responsible', 'Invalid user')
-        context = self.get_context_data(**kwargs)
-        context['add_storage_form'] = f
-        return self.render_to_response(context)
-
-    def handle_del_storage_post(self, request, **kwargs):
-        parent = self.object.parent_storage
-        try:   
-            self.object.delete()
-        except:
-            context = self.get_context_data(**kwargs)
-            context['delete_storage_errors'] = ['Error deleting Storage '+str(self.object)]
-            return self.render_to_response(context)
-        
-        if parent is None:
-            return redirect('parts-stocks')
-        else:
-            return redirect(reverse('parts-stocks-detail', kwargs={'uuid':parent.id}))
-    def handle_del_stock_post(self, request, **kwargs):
-        del_error = None
-        if 'stock_uuid' in request.POST:
-            f = DeleteStockForm(data=request.POST)
-            if f.is_valid():
-                try:
-                    s = Stock.objects.get(id=f.cleaned_data['stock_uuid'])
-                    print(s.storage)
-                    print(self.object)
-                    if s.storage == self.object:
-                       s.delete()
-                    else:
-                        del_error = 'Cannot delete stock from another storage.' 
-                except:
-                    del_error = 'Could not find requested stock in this storage.'
-        
-        context = self.get_context_data(**kwargs)
-        return self.render_to_response(context)
-        
-    def handle_update_watermark(self, request, **kwargs):
-        edit_form = EditWatermarkForm(data=request.POST)
-        update_watermark_error = None
-        if edit_form.is_valid():
-            edit_form.save()
-        else:
-            pass # Todo: Handle error
-
-        context = self.get_context_data(**kwargs)
-        return self.render_to_response(context)
-    
-    def handle_amount_change_post(self, request, increase, **kwargs):
-        edit_form = EditStockAmountForm(data=request.POST)
-        if edit_form.is_valid():
-            edit_form.save(increase)
-
-        context = self.get_context_data(**kwargs)
-        return self.render_to_response(context)
-
-    def handle_add_stock_post(self, request, **kwargs):
-        f = AddStockForm(data=request.POST)
-        error_occured = False
-
-        if f.is_valid():
-            try:
-                f.save(self.object)
-            except Exception as ex:
-                f.add_error('', str(ex))
-                error_occured = True
-        else:
-            error_occured = True
-
-        context = self.get_context_data(**kwargs)
-        if error_occured:
-            context['add_stock_form'] = f
-        return self.render_to_response(context)
-
-    def post(self, request, *args, **kwargs):
-        self.object = self.get_object()
-
-        if 'submit-add-storage' in request.POST:
-            return self.handle_add_storage_post(request, **kwargs)
-        elif 'submit-delete-storage' in request.POST:
-            return self.handle_del_storage_post(request, **kwargs)
-        elif 'submit-delete-stock' in request.POST:
-            return self.handle_del_stock_post(request, **kwargs)
-        elif 'submit-edit-watermark' in request.POST:
-            return self.handle_update_watermark(request, **kwargs)
-        elif 'submit-amount-reduce' in request.POST:
-            return self.handle_amount_change_post(request, False, **kwargs)
-        elif 'submit-amount-increase' in request.POST:
-            return self.handle_amount_change_post(request, True, **kwargs)     
-        elif 'submit-add-stock' in request.POST:
-            return self.handle_add_stock_post(request, **kwargs)
-
-        return super().post(request, *args, **kwargs)
-
-class ComponentDetailView(LoginRequiredMixin, BaseTemplateMixin, DetailView):
-    template_name = 'parts/components-detail.html'
-    model = Component
-    pk_url_kwarg = 'uuid'
-    base_title = ''
-    navbar_selected = 'Components'
-   
-    def get_context_data(self, **kwargs):
-        self.base_title = 'Component / '+self.object.name
-        context = super().get_context_data(**kwargs)
-        context['component'] = self.object
-        context['stocks'] = Stock.objects.filter(component=self.object)
-        context['comp_form'] = ComponentForm(instance=self.object)
-        context['new_distri_num_form'] = DistributorNumberCreateForm()
-        context['new_param_form'] = ComponentParameterCreateForm()
-        context['distri_nums'] = DistributorNum.objects.filter(component=self.object).order_by('distributor__name')
-        context['parameters'] = ComponentParameter.objects.filter(component=self.object).order_by('parameter_type__parameter_name')
-
-        return context
-
-    def handle_submit_edit_component_post(self, request, **kwargs):
-        cform = ComponentForm(instance=self.object, data=request.POST, files=request.FILES)
-        if cform.is_valid():
-            cform.save()
-        
-        context = self.get_context_data(**kwargs)
-        if not cform.is_valid():
-            context['comp_form'] = cform
-
-        return self.render_to_response(context)
-
-    def handle_submit_delete_post(self, request, **kwargs):
-        delete_error = None
-        protected_stuff = None
-
-        try:
-            self.object.delete()
-        except ProtectedError as pe:
-            delete_error = 'Component is protected'
-            protected_stuff = pe.protected_objects
-        except:
-            delete_error = 'Cannot delete component. Unknown error'
-
-        if delete_error is None:
-            return redirect('parts-components')
-        else:
-            context = self.get_context_data(**kwargs)
-            context['delete_error'] = delete_error
-            context['protected_stuff'] = protected_stuff
-            return self.render_to_response(context)
-
-    def handle_submit_new_distri_num_post(self, request, **kwargs):
-        form = DistributorNumberCreateForm(data=request.POST)
-        if form.is_valid():
-            new_number = form.save(commit=False)
-            new_number.component = self.object
-            try:
-                new_number.save()
-            except IntegrityError as ie:
-                form.add_error('__all__', 'Number for given distributor already exists')
-
-        context = self.get_context_data(**kwargs)
-        if not form.is_valid():
-            context['new_distri_num_form'] = form
-        return self.render_to_response(context)
-    
-    def handle_submit_delete_distri_num_post(self, request, **kwargs):
-        form = DistributorNumberDeleteForm(data=request.POST)
-        if form.is_valid():
-            form.cleaned_data['distributor_num'].delete()
-
-        context = self.get_context_data(**kwargs)
-        return self.render_to_response(context)
-    
-    def handle_submit_delete_param_post(self, request, **kwargs):
-        form = ComponentParameterDeleteForm(data=request.POST)
-        if form.is_valid():
-            form.cleaned_data['param_num'].delete()
-        
-        context = self.get_context_data(**kwargs)
-        return self.render_to_response(context)
-    
-    def handle_submit_new_param_post(self, request, **kwargs):
-        form = ComponentParameterCreateForm(data=request.POST)
-        
-        if form.is_valid():
-            try:
-                form.save(self.object)
-            except IntegrityError:
-                form.add_error('__all__', 'This parameter is already set')
-        context = self.get_context_data(**kwargs)
-        if not form.is_valid():
-            context['new_param_form'] = form
-        return self.render_to_response(context)
-
-    def post(self, request, *args, **kwargs):
-        self.object = self.get_object()
-        if 'submit-edit-component' in request.POST:
-            return self.handle_submit_edit_component_post(request, **kwargs)
-        elif 'submit-component-delete' in request.POST:
-            return self.handle_submit_delete_post(request, **kwargs)
-        elif 'submit-create-new-distri-num' in request.POST:
-            return self.handle_submit_new_distri_num_post(request, **kwargs)
-        elif 'submit-delete-distributor-num' in request.POST:
-            return self.handle_submit_delete_distri_num_post(request, **kwargs)
-        elif 'submit-delete-param' in request.POST:
-            return self.handle_submit_delete_param_post(request, **kwargs)
-        elif 'submit-create-new-param' in request.POST:
-            return self.handle_submit_new_param_post(request, **kwargs)
-        else:
-            return super().post(request, *args, **kwargs)
-
-class PackageDetailView(LoginRequiredMixin, BaseTemplateMixin, DetailView):
-    template_name = 'parts/packages-detail.html'
-    model = Package
-    pk_url_kwarg = 'uuid'
-    base_title = ''
-    navbar_selected = 'Packages'
-    
-    def get_context_data(self, **kwargs):
-        self.base_title = 'Package / '+self.object.name
-        context = super().get_context_data(**kwargs)
-        context['package'] = self.object
-        context['edit_form'] = PackageForm(instance=self.object)
- 
-        return context
-
-    def handle_delete_package(self, request):
-        delete_error = None
-        protected_objects = None
-        
-        # Try to delete this instance
-        try:
-            self.object.delete()
-        except ProtectedError as pe:
-            delete_error = 'Cannot delete this package. It is referenced by a component.'
-            protected_objects = pe.protected_objects
-        except:
-            delete_error = 'Cannot delete this package. Unknown error'
-        
-        if delete_error:
-            context = self.get_context_data()
-            context['delete_error'] = delete_error
-            context['protected_components'] = protected_objects
-            return self.render_to_response(context)
-        else:
-            return redirect('parts-packages')
-
-    def edit_package(self, request):
-        edit_form = PackageForm(data=request.POST, files=request.FILES, instance=self.object)
-
-        if edit_form.is_valid():
-            edit_form.save()
-
-        context = self.get_context_data()
-        if not edit_form.is_valid():
-            context['edit_form'] = edit_form
-        return self.render_to_response(context)
-
-    def post(self, request, *args, **kwargs):
-        self.object = self.get_object()
-
-        if 'submit-pkg-delete' in request.POST:
-            return self.handle_delete_package(request)
-        elif 'submit-pkg-edit' in request.POST:
-            return self.edit_package(request)
-        
-        return super().post(request, *args, **kwargs)
-
-class DistributorDetailView(LoginRequiredMixin, BaseTemplateMixin, DetailView):
-    template_name = 'parts/distributors-detail.html'
-    model = Distributor
-    pk_url_kwarg = 'uuid'
-    base_title = ''
-    navbar_selected = 'Distributors'
-    
-    def get_context_data(self, **kwargs):
-        self.base_title = 'Distributor / '+self.object.name
-        context = super().get_context_data(**kwargs)
-        context['distributor'] = self.object
-        context['edit_form'] = DistributorForm(instance=self.object)
- 
-        return context
-
-    def handle_delete_distributor(self, request):
-        delete_error = None
-        protected_objects = None
-        
-        # Try to delete this instance
-        try:
-            self.object.delete()
-        except ProtectedError as pe:
-            delete_error = 'Cannot delete this distributor. It is referenced by a component.'
-            protected_objects = pe.protected_objects
-        except:
-            delete_error = 'Cannot delete this distributor. Unknown error'
-        
-        if delete_error:
-            context = self.get_context_data()
-            context['delete_error'] = delete_error
-            context['protected_components'] = protected_objects
-            return self.render_to_response(context)
-        else:
-            return redirect('parts-distributors')
-
-    def edit_distributor(self, request):
-        edit_form = DistributorForm(data=request.POST, files=request.FILES, instance=self.object)
-
-        if edit_form.is_valid():
-            edit_form.save()
-
-        context = self.get_context_data()
-        if not edit_form.is_valid():
-            context['edit_form'] = edit_form
-        return self.render_to_response(context)
-
-    def post(self, request, *args, **kwargs):
-        self.object = self.get_object()
-
-        if 'submit-distri-delete' in request.POST:
-            return self.handle_delete_distributor(request)
-        elif 'submit-distri-edit' in request.POST:
-            return self.edit_distributor(request)
-        
-        return super().post(request, *args, **kwargs)
-
-class ManufacturersViewSet(LoginRequiredMixin, BaseTemplateMixin, TemplateView):
-    template_name = 'parts/manufacturers.html'
-    base_title = 'Manufacturers'
-    navbar_selected = 'Manufacturers'
-    default_page_size = 25
-
-    def search_manufacturers(self, search):
-        qs = Manufacturer.objects.all()
-
-        if not search:
-            return qs
-        
-        search_fragements = search.strip().split()
-
-        for search in search_fragements:
-            qs = qs.filter(Q(name__icontains = search) | Q(website__icontains = search))
-
-        return qs
-
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-        page_num = self.request.GET.get('page', default=1)
-        search_string = self.request.GET.get('search', default=None)
-        
-        queryset = self.search_manufacturers(search_string)
-
-        paginator = Paginator(queryset, self.default_page_size)       
-
-
-        context['search_string'] = search_string
-        context['manufacturers'] = paginator.get_page(page_num)
-        context['new_manufacturer_form'] = ManufacturerForm()
-
-        return context
-    
-    def handle_add_new_manufacturer(self, request):
-
-        form = ManufacturerForm(data=request.POST, files=request.FILES)
-        if form.is_valid():
-            form.save()
-
-        context = self.get_context_data()
-
-        if not form.is_valid():
-            context['new_manufacturer_form'] = form
-
-        return self.render_to_response(context)
-
-    def post(self, request, *args, **kwargs):
-        if 'submit-manufacturer-add-new' in request.POST:
-            return self.handle_add_new_manufacturer(request)
-
-        return super().post(request, *args, **kwargs)
-
-class ManufacturerDetailViewSet(LoginRequiredMixin, BaseTemplateMixin, DetailView):
-    template_name = 'parts/manufacturers-detail.html'
-    model = Manufacturer
-    pk_url_kwarg = 'uuid'
-    base_title = ''
-    navbar_selected = 'Manufacturers'
-    
-    def get_context_data(self, **kwargs):
-        self.base_title = 'Manufacturer / '+self.object.name
-        context = super().get_context_data(**kwargs)
-        context['manufacturer'] = self.object
-        context['edit_form'] = ManufacturerForm(instance=self.object)
- 
-        return context
-
-    def handle_delete_manufacturer(self, request):
-        delete_error = None
-        protected_objects = None
-        
-        # Try to delete this instance
-        try:
-            self.object.delete()
-        except ProtectedError as pe:
-            delete_error = 'Cannot delete this distributor. It is referenced by a component.'
-            protected_objects = pe.protected_objects
-        except:
-            delete_error = 'Cannot delete this distributor. Unknown error'
-        
-        if delete_error:
-            context = self.get_context_data()
-            context['delete_error'] = delete_error
-            context['protected_components'] = protected_objects
-            return self.render_to_response(context)
-        else:
-            return redirect('parts-manufacturers')
-
-    def edit_manufacturer(self, request):
-        edit_form = ManufacturerForm(data=request.POST, files=request.FILES, instance=self.object)
-
-        if edit_form.is_valid():
-            edit_form.save()
-
-        context = self.get_context_data()
-        if not edit_form.is_valid():
-            context['edit_form'] = edit_form
-        return self.render_to_response(context)
-
-    def post(self, request, *args, **kwargs):
-        self.object = self.get_object()
-
-        if 'submit-manufacturer-delete' in request.POST:
-            return self.handle_delete_manufacturer(request)
-        elif 'submit-manufacturer-edit' in request.POST:
-            return self.edit_manufacturer(request)
-        
-        return super().post(request, *args, **kwargs)

shimatta_kenkyusho/parts/views/__init__.py

@@ -0,0 +1,6 @@
+from .component_views import *
+from .distributor_views import *
+from .generic_views import *
+from .manufacturer_views import *
+from .package_views import *
+from .storage_views import *

shimatta_kenkyusho/parts/views/component_import.py

@@ -0,0 +1,110 @@
+import io
+import csv
+import requests
+from django.db import transaction
+from django.core.files.images import ImageFile
+from ..models import ComponentParameter, ComponentType, Manufacturer, Component, \
+                    DistributorNum, Stock, Storage, Distributor, Package, ComponentParameterType
+
+def _stock_component(component, storage_uuid, substorage_path, amount, lot, watermark):
+
+    if not amount or not any([storage_uuid, substorage_path]):
+        return None
+
+    storage = Storage.from_path(substorage_path, storage_uuid)
+
+    stock = Stock.objects.create(component=component,
+                                 storage=storage,
+                                 amount=amount,
+                                 lot=lot,
+                                 watermark=watermark)
+    return stock
+
+def _set_additional_parameters(component, type, value):
+
+    if type.startswith('param:'):
+        type = type[6:]
+        param_type = ComponentParameterType.objects.get(parameter_name=type)
+
+        param = ComponentParameter.objects.create(component=component,
+                                                  parameter_type=param_type)
+        if param_type.parameter_type == 'F':
+            param.text_value = value
+        else:
+            param.value = float(value)
+        param.save()
+
+        return param
+
+    elif type.startswith('distri:'):
+        distri_name = type[7:]
+        distri = Distributor.objects.get(name=distri_name)
+
+        distri_num = DistributorNum.objects.create(component=component,
+                                                   distributor=distri,
+                                                   distributor_part_number=value)
+        return distri_num
+
+def import_components_from_csv(csv_file):
+    """
+    Imports components from a csv file containing the component model fields as
+    well as storage information in the heading.
+
+    Parameters can be set by param:<parameter name>, distri numbers by
+    distri:<distri name>.
+    """
+
+    with transaction.atomic():
+
+        # simulate a text-file for the csv lib
+        with io.TextIOWrapper(csv_file, encoding='utf8') as csv_text_file:
+            rows = csv.DictReader(csv_text_file, delimiter=";")
+
+            for data in rows:
+
+                image = None
+                image_url = data.pop('image_url')
+                if image_url:
+                    response = requests.get(image_url)
+                    image_content = response.content
+                    image_file = io.BytesIO(image_content)
+                    image = ImageFile(image_file, 'downloaded_file')
+
+                manufacturer = None
+                manufacturer_name = data.pop('manufacturer')
+                if manufacturer_name:
+                    manufacturer = Manufacturer.objects.get(name=manufacturer_name)
+
+                component_type = None
+                component_type_name = data.pop('component_type')
+                if component_type_name:
+                    component_type = ComponentType.objects.get(class_name=component_type_name)
+
+                distributor = None
+                pref_distri = data.pop('pref_distri')
+                if pref_distri:
+                    distributor = Distributor.objects.get(name=pref_distri)
+
+                package = None
+                package_name = data.pop('package')
+                if package_name:
+                    package = Package.objects.get(name=package_name)
+
+                comp = Component.objects.create(name=data.pop('name'),
+                                                manufacturer=manufacturer,
+                                                component_type=component_type,
+                                                pref_distri=distributor,
+                                                description=data.pop('description'),
+                                                datasheet_link=data.pop('datasheet_link'),
+                                                package=package,
+                                                image=image)
+
+                _stock_component(comp,
+                                 data.pop('storage_uuid'),
+                                 data.pop('substorage_path'),
+                                 data.pop('amount'),
+                                 data.pop('lot'),
+                                 data.pop('watermark'))
+
+                for key, value in data.items():
+                    _set_additional_parameters(comp, key, value)

shimatta_kenkyusho/parts/views/component_views.py

@@ -0,0 +1,272 @@
+import uuid
+from django.shortcuts import redirect
+from django.urls import reverse
+from django.views.generic import TemplateView, DetailView
+from django.contrib.auth.mixins import LoginRequiredMixin
+from django.core.paginator import Paginator
+from django.db.models import Q
+from django.forms import formset_factory
+from django.db import IntegrityError
+from django.db.models import ProtectedError
+from ..models import Stock, Component, ComponentParameter, DistributorNum
+from ..forms import *
+from .component_import import import_components_from_csv
+from .generic_views import BaseTemplateMixin
+
+
+ParameterSearchFormSet = formset_factory(ComponentParameterSearchForm, extra=1)
+
+# Create your views here.
+
+class ComponentView(LoginRequiredMixin, BaseTemplateMixin, TemplateView):
+    template_name = 'parts/components.html'
+    base_title = 'Components'
+    navbar_selected = 'Components'
+    default_page_size = 25
+
+    def get_component_query_set(self, search_string):
+        queryset = Component.objects.all()
+
+        if not search_string:
+            return queryset
+
+        search_fragments = search_string.strip().split()
+        for search in search_fragments:
+            queryset = queryset.filter(Q(name__icontains = search) | Q(manufacturer__name__icontains = search) | Q(package__name__icontains = search))
+
+        return queryset
+
+    def get_component_queryset_from_advanced_search(self, cleaned_data):
+        queryset = Component.objects.all()
+
+        if cleaned_data['name']:
+            queryset = queryset.filter(Q(name__icontains=cleaned_data['name']))
+        if cleaned_data['package']:
+            queryset = queryset.filter(package=cleaned_data['package'])
+        if cleaned_data['package_pin_count']:
+            queryset = queryset.filter(package__pin_count=cleaned_data['package_pin_count'])
+        if cleaned_data['component_type']:
+            queryset = queryset.filter(component_type=cleaned_data['component_type'])
+        if cleaned_data['distributor_num']:
+            if cleaned_data['distributor']:
+                distri = cleaned_data['distributor']
+                queryset = queryset.filter(Q(distributornum__distributor_part_number__icontains=cleaned_data['distributor_num']) & Q(distributornum__distributor=distri))
+            else:
+                 queryset = queryset.filter(Q(distributornum__distributor_part_number__icontains=cleaned_data['distributor_num']))
+        if cleaned_data['manufacturer']:
+            queryset = queryset.filter(manufacturer=cleaned_data['manufacturer'])
+        return queryset
+
+    def get_context_data_int(self, advanced_search, parameter_formset : ParameterSearchFormSet, **kwargs):
+        context = super().get_context_data(**kwargs)
+
+        comp_page_num = self.request.GET.get('comp_page', default=1)
+
+        if advanced_search and parameter_formset:
+            search = None
+            context['advanced_search_shown'] = True
+            context['advanced_search_form'] = advanced_search
+            context['advanced_search_param_formset'] = parameter_formset
+            if advanced_search.is_valid():
+                paginator_queryset = self.get_component_queryset_from_advanced_search(
+			advanced_search.cleaned_data)
+            else:
+                paginator_queryset = Component.objects.all()
+
+            if parameter_formset.is_valid():
+                # Process parameters
+                pass
+
+        else:
+            search = self.request.GET.get('search', default=None)
+            paginator_queryset = self.get_component_query_set(search)
+
+        comp_paginator = Paginator(paginator_queryset, self.default_page_size)
+
+        context['components'] = comp_paginator.get_page(comp_page_num)
+        context['comp_form'] = ComponentForm()
+        context['import_comp_form'] = ImportComponentForm()
+        context['search_string'] = search
+
+        if not parameter_formset:
+            context['advanced_search_param_formset'] = ParameterSearchFormSet()
+
+
+        if not advanced_search:
+            context['advanced_search_form'] = AdvancedComponentSearchForm(auto_id='adv_search_%s')
+
+        return context
+
+    def get_context_data(self, **kwargs):
+        return self.get_context_data_int(advanced_search = None, parameter_formset=None, **kwargs)
+
+    def handle_new_component_post(self, request, open=False, **kwargs):
+        cform = ComponentForm(data=request.POST, files=request.FILES)
+        new_component = None
+        if cform.is_valid():
+            new_component = cform.save()
+
+        context = self.get_context_data(**kwargs)
+        if not cform.is_valid():
+            context['comp_form'] = cform
+
+        if open and new_component:
+            return redirect(reverse('parts-components-detail', kwargs={'uuid':new_component.id}))
+        return self.render_to_response(context)
+
+    def handle_import_components_post(self, request, open=False, **kwargs):
+        cform = ImportComponentForm(data=request.POST, files=request.FILES)
+        context = self.get_context_data(**kwargs)
+        if cform.is_valid():
+            try:
+                import_components_from_csv(cform.files['csv_file'].file)
+            except Exception as ex:
+                cform.add_error('csv_file', str(ex))
+                context['import_comp_form'] = cform
+        else:
+            context['import_comp_form'] = cform
+
+        return self.render_to_response(context)
+
+    def handle_advanced_search_post(self, request, **kwargs):
+
+        form = AdvancedComponentSearchForm(auto_id='adv_search_%s', data=request.POST)
+        param_formset = ParameterSearchFormSet(data=request.POST)
+
+        if form.is_valid():
+            print('Valid')
+
+        if param_formset.is_valid():
+            print('Formset is valid!')
+
+
+        context = self.get_context_data_int(form, param_formset, **kwargs)
+        return self.render_to_response(context)
+
+    def post(self, request, *args, **kwargs):
+        if 'submit-edit-component' in request.POST:
+            return self.handle_new_component_post(request, open=False, **kwargs)
+        elif 'submit-edit-component-open' in request.POST:
+            return self.handle_new_component_post(request, open=True, **kwargs)
+        elif 'submit-import-components' in request.POST:
+            return self.handle_import_components_post(request, open=True, **kwargs)
+        elif 'submit-advanced-search' in request.POST:
+            return self.handle_advanced_search_post(request, **kwargs)
+        else:
+            return super().post(request, *args, **kwargs)
+
+class ComponentDetailView(LoginRequiredMixin, BaseTemplateMixin, DetailView):
+    template_name = 'parts/components-detail.html'
+    model = Component
+    pk_url_kwarg = 'uuid'
+    base_title = ''
+    navbar_selected = 'Components'
+
+    def get_context_data(self, **kwargs):
+        self.base_title = 'Component / '+self.object.name
+        context = super().get_context_data(**kwargs)
+        context['component'] = self.object
+        context['stocks'] = Stock.objects.filter(component=self.object)
+        context['comp_form'] = ComponentForm(instance=self.object)
+        context['new_distri_num_form'] = DistributorNumberCreateForm()
+        context['new_param_form'] = ComponentParameterCreateForm()
+        context['distri_nums'] = DistributorNum.objects.filter(component=self.object).order_by(
+            'distributor__name')
+        context['parameters'] = ComponentParameter.objects.filter(component=self.object).order_by(
+            'parameter_type__parameter_name')
+
+        return context
+
+    def handle_submit_edit_component_post(self, request, **kwargs):
+        cform = ComponentForm(instance=self.object, data=request.POST, files=request.FILES)
+        if cform.is_valid():
+            cform.save()
+
+        context = self.get_context_data(**kwargs)
+        if not cform.is_valid():
+            context['comp_form'] = cform
+
+        return self.render_to_response(context)
+
+    def handle_submit_delete_post(self, request, **kwargs):
+        delete_error = None
+        protected_stuff = None
+
+        try:
+            self.object.delete()
+        except ProtectedError as pe:
+            delete_error = 'Component is protected'
+            protected_stuff = pe.protected_objects
+        except:
+            delete_error = 'Cannot delete component. Unknown error'
+
+        if delete_error is None:
+            return redirect('parts-components')
+        else:
+            context = self.get_context_data(**kwargs)
+            context['delete_error'] = delete_error
+            context['protected_stuff'] = protected_stuff
+            return self.render_to_response(context)
+
+    def handle_submit_new_distri_num_post(self, request, **kwargs):
+        form = DistributorNumberCreateForm(data=request.POST)
+        if form.is_valid():
+            new_number = form.save(commit=False)
+            new_number.component = self.object
+            try:
+                new_number.save()
+            except IntegrityError as ie:
+                form.add_error('__all__', 'Number for given distributor already exists')
+
+        context = self.get_context_data(**kwargs)
+        if not form.is_valid():
+            context['new_distri_num_form'] = form
+        return self.render_to_response(context)
+
+    def handle_submit_delete_distri_num_post(self, request, **kwargs):
+        form = DistributorNumberDeleteForm(data=request.POST)
+        if form.is_valid():
+            form.cleaned_data['distributor_num'].delete()
+
+        context = self.get_context_data(**kwargs)
+        return self.render_to_response(context)
+
+    def handle_submit_delete_param_post(self, request, **kwargs):
+        form = ComponentParameterDeleteForm(data=request.POST)
+        if form.is_valid():
+            form.cleaned_data['param_num'].delete()
+
+        context = self.get_context_data(**kwargs)
+        return self.render_to_response(context)
+
+    def handle_submit_new_param_post(self, request, **kwargs):
+        form = ComponentParameterCreateForm(data=request.POST)
+
+        if form.is_valid():
+            try:
+                form.save(self.object)
+            except IntegrityError:
+                form.add_error('__all__', 'This parameter is already set')
+        context = self.get_context_data(**kwargs)
+        if not form.is_valid():
+            context['new_param_form'] = form
+        return self.render_to_response(context)
+
+    def post(self, request, *args, **kwargs):
+        self.object = self.get_object()
+        if 'submit-edit-component' in request.POST:
+            return self.handle_submit_edit_component_post(request, **kwargs)
+        elif 'submit-component-delete' in request.POST:
+            return self.handle_submit_delete_post(request, **kwargs)
+        elif 'submit-create-new-distri-num' in request.POST:
+            return self.handle_submit_new_distri_num_post(request, **kwargs)
+        elif 'submit-delete-distributor-num' in request.POST:
+            return self.handle_submit_delete_distri_num_post(request, **kwargs)
+        elif 'submit-delete-param' in request.POST:
+            return self.handle_submit_delete_param_post(request, **kwargs)
+        elif 'submit-create-new-param' in request.POST:
+            return self.handle_submit_new_param_post(request, **kwargs)
+        else:
+            return super().post(request, *args, **kwargs)
+
+

shimatta_kenkyusho/parts/views/distributor_views.py

@@ -0,0 +1,120 @@
+from django.shortcuts import redirect
+from django.views.generic import TemplateView, DetailView
+from django.contrib.auth.mixins import LoginRequiredMixin
+from ..models import Distributor
+from django.core.paginator import Paginator
+from django.db.models import ProtectedError
+from ..forms import *
+from django.db.models import Q
+from .generic_views import BaseTemplateMixin
+
+class DistributorView(LoginRequiredMixin, BaseTemplateMixin, TemplateView):
+    template_name = 'parts/distributors.html'
+    base_title = 'Distributors'
+    navbar_selected = 'Distributors'
+    default_page_size = 25
+
+    def search_distributors(self, search):
+        qs = Distributor.objects.all()
+
+        if not search:
+            return qs
+
+        search_fragments = search.strip().split()
+
+        for search in search_fragments:
+            qs = qs.filter(Q(name__icontains = search) | Q(website__icontains = search))
+        return qs
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        page_num = self.request.GET.get('page', default=1)
+        search_string = self.request.GET.get('search', default=None)
+
+        queryset = self.search_distributors(search_string)
+
+        paginator = Paginator(queryset, self.default_page_size)
+
+
+        context['search_string'] = search_string
+        context['distributors'] = paginator.get_page(page_num)
+        context['new_distri_form'] = DistributorForm()
+
+        return context
+
+    def handle_add_new_distributor(self, request):
+
+        form = DistributorForm(data=request.POST, files=request.FILES)
+        if form.is_valid():
+            form.save()
+
+        context = self.get_context_data()
+
+        if not form.is_valid():
+            context['new_distri_form'] = form
+
+        return self.render_to_response(context)
+
+    def post(self, request, *args, **kwargs):
+        if 'submit-distri-add-new' in request.POST:
+            return self.handle_add_new_distributor(request)
+
+        return super().post(request, *args, **kwargs)
+
+
+class DistributorDetailView(LoginRequiredMixin, BaseTemplateMixin, DetailView):
+    template_name = 'parts/distributors-detail.html'
+    model = Distributor
+    pk_url_kwarg = 'uuid'
+    base_title = ''
+    navbar_selected = 'Distributors'
+
+    def get_context_data(self, **kwargs):
+        self.base_title = 'Distributor / '+self.object.name
+        context = super().get_context_data(**kwargs)
+        context['distributor'] = self.object
+        context['edit_form'] = DistributorForm(instance=self.object)
+
+        return context
+
+    def handle_delete_distributor(self, request):
+        delete_error = None
+        protected_objects = None
+
+        # Try to delete this instance
+        try:
+            self.object.delete()
+        except ProtectedError as pe:
+            delete_error = 'Cannot delete this distributor. It is referenced by a component.'
+            protected_objects = pe.protected_objects
+        except:
+            delete_error = 'Cannot delete this distributor. Unknown error'
+
+        if delete_error:
+            context = self.get_context_data()
+            context['delete_error'] = delete_error
+            context['protected_components'] = protected_objects
+            return self.render_to_response(context)
+        else:
+            return redirect('parts-distributors')
+
+    def edit_distributor(self, request):
+        edit_form = DistributorForm(data=request.POST, files=request.FILES, instance=self.object)
+
+        if edit_form.is_valid():
+            edit_form.save()
+
+        context = self.get_context_data()
+        if not edit_form.is_valid():
+            context['edit_form'] = edit_form
+        return self.render_to_response(context)
+
+    def post(self, request, *args, **kwargs):
+        self.object = self.get_object()
+
+        if 'submit-distri-delete' in request.POST:
+            return self.handle_delete_distributor(request)
+        elif 'submit-distri-edit' in request.POST:
+            return self.edit_distributor(request)
+
+        return super().post(request, *args, **kwargs)

shimatta_kenkyusho/parts/views/generic_views.py

@@ -0,0 +1,121 @@
+from django.shortcuts import render, redirect
+from django.contrib.auth import logout, login
+from django.contrib.auth.forms import AuthenticationForm as AuthForm
+from django.contrib.auth.forms import PasswordChangeForm
+from django.contrib.auth import update_session_auth_hash
+from django.views.generic import TemplateView
+from django.contrib.auth.mixins import LoginRequiredMixin
+from django.http import HttpResponse
+from ..navbar import NavBar
+from ..forms import QrSearchForm
+
+class BaseTemplateMixin():
+    navbar_selected = ''
+    base_title = ''
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        base_context = {
+            'navbar': NavBar.get_navbar(self.navbar_selected, self.request.user),
+            'title': NavBar.get_brand()+' / '+ self.base_title,
+            'login_active': False,
+        }
+        context['base'] = base_context
+        return context
+
+    def post(self, request, *args, **kwargs):
+        data = request.POST
+        if 'qr_search' not in data:
+            super().post(request, *args, **kwargs)
+
+        print('QR',data['qr_search'])
+        f = QrSearchForm(data)
+        if f.is_valid():
+            return redirect(f.my_qr_validator.get_redirect_url(f.cleaned_data['qr_search']))
+
+        return self.get(request)
+
+class ChangePasswordView(LoginRequiredMixin, BaseTemplateMixin, TemplateView):
+    template_name = 'parts/change-pw.html'
+    navbar_selected = 'Main'
+    base_title = 'Change Password'
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        context['form'] = PasswordChangeForm(self.request.user)
+        return context
+
+    def post(self, request, *args, **kwargs):
+        if 'submit-change-pw' not in request.POST:
+            return super().post(request, *args, **kwargs)
+
+        form = PasswordChangeForm(request.user, data=request.POST)
+
+        if form.is_valid():
+            user = form.save()
+            update_session_auth_hash(request, user)
+            return redirect('parts-main')
+
+        context = self.get_context_data(**kwargs)
+        if form.errors:
+            context['form'] = form
+
+        return self.render_to_response(context)
+
+
+class MainView(BaseTemplateMixin, TemplateView):
+    template_name = 'parts/main.html'
+    navbar_selected = 'Main'
+    base_title = 'Main'
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        context['user'] = self.request.user
+        return context
+
+
+def logout_view(request):
+    logout(request)
+    return redirect('parts-main')
+
+def login_view(request):
+    base_context = {
+        'navbar': NavBar.get_navbar('Login', request.user),
+        'title': NavBar.get_brand()+' / '+'Login',
+        'login_active': True,
+    }
+
+    if request.user.is_authenticated:
+        next_param = request.GET.get('next')
+        if next_param is not None:
+            return redirect(next_param)
+        return redirect('parts-main')
+
+    if request.method == 'POST':
+        form = AuthForm(data=request.POST)
+        if form.is_valid():
+            valid_user = form.get_user()
+            login(request, valid_user)
+            next_param = request.GET.get('next')
+            if next_param is not None:
+                return redirect(next_param)
+            return redirect('parts-main')
+    else:
+        form = AuthForm()
+
+
+    context = {
+        'base': base_context,
+        'form': form,
+    }
+
+    return render(request, 'parts/login.html', context)
+
+
+
+def health_check_view(_request) -> HttpResponse:
+    """
+    Health checking view. Returns empty http response with HTTP status OK.
+    This will be used to check if the system is actually running correctly
+    """
+    return HttpResponse(status=200)

shimatta_kenkyusho/parts/views/manufacturer_views.py

@@ -0,0 +1,121 @@
+from django.shortcuts import redirect
+from django.views.generic import TemplateView, DetailView
+from django.contrib.auth.mixins import LoginRequiredMixin
+from ..models import Manufacturer
+from django.core.paginator import Paginator
+from django.db.models import ProtectedError
+from ..forms import *
+from django.db.models import Q
+from .generic_views import BaseTemplateMixin
+
+class ManufacturersView(LoginRequiredMixin, BaseTemplateMixin, TemplateView):
+    template_name = 'parts/manufacturers.html'
+    base_title = 'Manufacturers'
+    navbar_selected = 'Manufacturers'
+    default_page_size = 25
+
+    def search_manufacturers(self, search):
+        qs = Manufacturer.objects.all()
+
+        if not search:
+            return qs
+
+        search_fragements = search.strip().split()
+
+        for search in search_fragements:
+            qs = qs.filter(Q(name__icontains = search) | Q(website__icontains = search))
+
+        return qs
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        page_num = self.request.GET.get('page', default=1)
+        search_string = self.request.GET.get('search', default=None)
+
+        queryset = self.search_manufacturers(search_string)
+
+        paginator = Paginator(queryset, self.default_page_size)
+
+
+        context['search_string'] = search_string
+        context['manufacturers'] = paginator.get_page(page_num)
+        context['new_manufacturer_form'] = ManufacturerForm()
+
+        return context
+
+    def handle_add_new_manufacturer(self, request):
+
+        form = ManufacturerForm(data=request.POST, files=request.FILES)
+        if form.is_valid():
+            form.save()
+
+        context = self.get_context_data()
+
+        if not form.is_valid():
+            context['new_manufacturer_form'] = form
+
+        return self.render_to_response(context)
+
+    def post(self, request, *args, **kwargs):
+        if 'submit-manufacturer-add-new' in request.POST:
+            return self.handle_add_new_manufacturer(request)
+
+        return super().post(request, *args, **kwargs)
+
+
+class ManufacturerDetailView(LoginRequiredMixin, BaseTemplateMixin, DetailView):
+    template_name = 'parts/manufacturers-detail.html'
+    model = Manufacturer
+    pk_url_kwarg = 'uuid'
+    base_title = ''
+    navbar_selected = 'Manufacturers'
+
+    def get_context_data(self, **kwargs):
+        self.base_title = 'Manufacturer / '+self.object.name
+        context = super().get_context_data(**kwargs)
+        context['manufacturer'] = self.object
+        context['edit_form'] = ManufacturerForm(instance=self.object)
+
+        return context
+
+    def handle_delete_manufacturer(self, request):
+        delete_error = None
+        protected_objects = None
+
+        # Try to delete this instance
+        try:
+            self.object.delete()
+        except ProtectedError as pe:
+            delete_error = 'Cannot delete this distributor. It is referenced by a component.'
+            protected_objects = pe.protected_objects
+        except:
+            delete_error = 'Cannot delete this distributor. Unknown error'
+
+        if delete_error:
+            context = self.get_context_data()
+            context['delete_error'] = delete_error
+            context['protected_components'] = protected_objects
+            return self.render_to_response(context)
+        else:
+            return redirect('parts-manufacturers')
+
+    def edit_manufacturer(self, request):
+        edit_form = ManufacturerForm(data=request.POST, files=request.FILES, instance=self.object)
+
+        if edit_form.is_valid():
+            edit_form.save()
+
+        context = self.get_context_data()
+        if not edit_form.is_valid():
+            context['edit_form'] = edit_form
+        return self.render_to_response(context)
+
+    def post(self, request, *args, **kwargs):
+        self.object = self.get_object()
+
+        if 'submit-manufacturer-delete' in request.POST:
+            return self.handle_delete_manufacturer(request)
+        elif 'submit-manufacturer-edit' in request.POST:
+            return self.edit_manufacturer(request)
+
+        return super().post(request, *args, **kwargs)

shimatta_kenkyusho/parts/views/package_views.py

@@ -0,0 +1,128 @@
+from django.shortcuts import  redirect
+from django.views.generic import TemplateView, DetailView
+from django.contrib.auth.mixins import LoginRequiredMixin
+from django.core.paginator import Paginator
+from django.db.models import ProtectedError
+from django.db.models import Q
+from ..forms import *
+from ..models import Package
+from .generic_views import BaseTemplateMixin
+
+class PackageView(LoginRequiredMixin, BaseTemplateMixin, TemplateView):
+    template_name = 'parts/packages.html'
+    base_title = 'Packages'
+    navbar_selected = 'Packages'
+    default_page_size = 25
+
+    def search_packages(self, search):
+        qs = Package.objects.all()
+
+        if not search:
+            return qs
+
+        search_fragments = search.strip().split()
+        for search in search_fragments:
+            if search.lower() == 'smd':
+                s_filter = Q(name__icontains = search) | Q(smd = True)
+            else:
+                try:
+                    pin_count = int(search)
+                    s_filter = Q(name__icontains = search) | Q(pin_count=pin_count)
+                except:
+                    s_filter = Q(name__icontains = search)
+
+            qs = qs.filter(s_filter)
+        return qs
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        page_num = self.request.GET.get('page', default=1)
+        search_string = self.request.GET.get('search', default=None)
+
+        package_queryset = self.search_packages(search_string)
+
+        paginator = Paginator(package_queryset, self.default_page_size)
+
+
+        context['search_string'] = search_string
+        context['packages'] = paginator.get_page(page_num)
+        context['new_pkg_form'] = PackageForm()
+
+        return context
+
+    def handle_add_new_package(self, request):
+
+        form = PackageForm(data=request.POST, files=request.FILES)
+        if form.is_valid():
+            form.save()
+
+        context = self.get_context_data()
+
+        if not form.is_valid():
+            context['new_pkg_form'] = form
+
+        return self.render_to_response(context)
+
+    def post(self, request, *args, **kwargs):
+        if 'submit-pkg-add-new' in request.POST:
+            return self.handle_add_new_package(request)
+
+        return super().post(request, *args, **kwargs)
+
+
+class PackageDetailView(LoginRequiredMixin, BaseTemplateMixin, DetailView):
+    template_name = 'parts/packages-detail.html'
+    model = Package
+    pk_url_kwarg = 'uuid'
+    base_title = ''
+    navbar_selected = 'Packages'
+
+    def get_context_data(self, **kwargs):
+        self.base_title = 'Package / '+self.object.name
+        context = super().get_context_data(**kwargs)
+        context['package'] = self.object
+        context['edit_form'] = PackageForm(instance=self.object)
+
+        return context
+
+    def handle_delete_package(self, request):
+        delete_error = None
+        protected_objects = None
+
+        # Try to delete this instance
+        try:
+            self.object.delete()
+        except ProtectedError as pe:
+            delete_error = 'Cannot delete this package. It is referenced by a component.'
+            protected_objects = pe.protected_objects
+        except:
+            delete_error = 'Cannot delete this package. Unknown error'
+
+        if delete_error:
+            context = self.get_context_data()
+            context['delete_error'] = delete_error
+            context['protected_components'] = protected_objects
+            return self.render_to_response(context)
+        else:
+            return redirect('parts-packages')
+
+    def edit_package(self, request):
+        edit_form = PackageForm(data=request.POST, files=request.FILES, instance=self.object)
+
+        if edit_form.is_valid():
+            edit_form.save()
+
+        context = self.get_context_data()
+        if not edit_form.is_valid():
+            context['edit_form'] = edit_form
+        return self.render_to_response(context)
+
+    def post(self, request, *args, **kwargs):
+        self.object = self.get_object()
+
+        if 'submit-pkg-delete' in request.POST:
+            return self.handle_delete_package(request)
+        elif 'submit-pkg-edit' in request.POST:
+            return self.edit_package(request)
+
+        return super().post(request, *args, **kwargs)

shimatta_kenkyusho/parts/views/storage_views.py

@@ -0,0 +1,254 @@
+import uuid
+from django.shortcuts import redirect
+from django.urls import reverse
+from django.views.generic import TemplateView, DetailView
+from django.contrib.auth.mixins import LoginRequiredMixin
+from django.core.paginator import Paginator
+from django.core.exceptions import ValidationError
+from django.db.models import Q
+from django.db.models.functions import Lower
+from ..models import Storage, Stock
+from ..forms import *
+from .generic_views import BaseTemplateMixin
+
+class StockView(LoginRequiredMixin, BaseTemplateMixin, TemplateView):
+    template_name = 'parts/stocks.html'
+    base_title = 'Stocks'
+    navbar_selected = 'Stocks'
+    default_pagination_size = 25
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+
+        storage_page = self.request.GET.get('storage_page')
+        if storage_page is None:
+            storage_page = 1
+        low_stock_page = self.request.GET.get('low_stock_page')
+        if low_stock_page is None:
+            low_stock_page = 1
+
+        storage_paginator = Paginator(Storage.objects.filter(parent_storage=None), self.default_pagination_size)
+        low_stock_paginator = Paginator(Stock.get_under_watermark(),
+            self.default_pagination_size)
+
+        context['low_stocks'] = low_stock_paginator.get_page(low_stock_page)
+        context['storages'] = storage_paginator.get_page(storage_page)
+        add_stor_form = AddSubStorageForm()
+        add_stor_form.fields['responsible'].initial = self.request.user.id
+        context['add_storage_form'] = add_stor_form
+        return context
+
+    def handle_add_storage(self, request, **kwargs):
+        f = AddSubStorageForm(data=request.POST)
+        if f.is_valid():
+            sub_name = f.cleaned_data['storage_name']
+            try:
+                Storage.objects.create(name=sub_name,
+                                        responsible=f.cleaned_data['responsible'],
+                                        is_template=f.cleaned_data['is_template'],
+                                        template=f.cleaned_data.get('template'))
+            except ValidationError as v_err:
+                f.add_error('storage_name', '. '.join(v_err.messages))
+        context = self.get_context_data(**kwargs)
+        context['add_storage_form'] = f
+        return self.render_to_response(context)
+
+    def post(self, request, **kwargs):
+        if 'submit-add-storage' in request.POST:
+            return self.handle_add_storage(request, **kwargs)
+
+        return super().post(request, **kwargs)
+
+class StockViewDetail(LoginRequiredMixin, BaseTemplateMixin, DetailView):
+    template_name = 'parts/stocks-detail.html'
+    model = Storage
+    pk_url_kwarg = 'uuid'
+    base_title = ''
+    navbar_selected = 'Stocks'
+    default_pagination_size = 8
+
+    def get_breadcrumbs(self):
+        crumbs = self.object.get_path_components()
+        # Reverse list and drop the last element of the reversed list
+        crumbs = crumbs[::-1][:-1]
+        return crumbs
+
+    def search_stock_queryset(self, search):
+        stocks_in_storage = Stock.objects.filter(storage=self.object).order_by(Lower('component__name'))
+
+        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:
+            test_uuid = uuid.UUID(search)
+        except:
+            pass
+
+        if test_uuid is not None:
+            stocks_in_storage = stocks_in_storage.filter(Q(component__id = test_uuid) | Q(id= test_uuid))
+        else:
+            stocks_in_storage = stocks_in_storage.filter(Q(component__name__icontains = search) |
+                Q(component__package__name__icontains = search) |
+                Q(component__manufacturer__name__icontains = search))
+
+        return stocks_in_storage
+
+    def get_context_data(self, **kwargs):
+        self.base_title = 'Stocks / ' + self.object.name
+        context = super().get_context_data(**kwargs)
+        context['breadcrumbs'] = self.get_breadcrumbs()
+
+        storage_page = self.request.GET.get('storage_page', default=1)
+        storage_paginator = Paginator(Storage.objects.filter(parent_storage=self.object), self.default_pagination_size)
+        stock_search_input = self.request.GET.get('search')
+
+        componente_stock_page = self.request.GET.get('stock_page', default=1)
+
+        stock_paginator = Paginator(self.search_stock_queryset(stock_search_input), self.default_pagination_size)
+
+        context['storages'] = storage_paginator.get_page(storage_page)
+        stocks = stock_paginator.get_page(componente_stock_page)
+        context['stocks'] = stocks
+        context['stock_search'] = stock_search_input
+        add_storage_form = AddSubStorageForm()
+        add_storage_form.fields['responsible'].initial = self.request.user.id
+        context['add_storage_form'] = add_storage_form
+        change_storage_form = ChangeStorageForm()
+        change_storage_form.fields['storage_name'].initial = self.object.name
+        change_storage_form.fields['verbose_name'].initial = self.object.verbose_name
+        change_storage_form.fields['responsible'].initial = self.object.responsible.id
+        change_storage_form.fields['is_template'].initial = self.object.is_template
+        context['change_storage_form'] = change_storage_form
+        context['delete_storage_error'] = None
+        context['add_stock_form'] = AddStockForm()
+
+        return context
+
+    def handle_add_storage_post(self, request, **kwargs):
+        f = AddSubStorageForm(data=request.POST)
+        if f.is_valid():
+            sub_name = f.cleaned_data['storage_name']
+            try:
+                Storage.objects.create(name=sub_name,
+                                        verbose_name=f.cleaned_data.get('verbose_name'),
+                                        parent_storage=self.object,
+                                        responsible=f.cleaned_data['responsible'],
+                                        is_template=f.cleaned_data['is_template'],
+                                        template=f.cleaned_data.get('template'))
+            except ValidationError as v_err:
+                f.add_error('storage_name', '. '.join(v_err.messages))
+        context = self.get_context_data(**kwargs)
+        context['add_storage_form'] = f
+        return self.render_to_response(context)
+
+    def handle_change_storage_post(self, request, **kwargs):
+        f = ChangeStorageForm(data=request.POST)
+        if f.is_valid():
+            try:
+                self.object.name = f.cleaned_data['storage_name']
+                self.object.verbose_name = f.cleaned_data.get('verbose_name')
+                self.object.responsible = f.cleaned_data['responsible']
+                self.object.is_template = f.cleaned_data['is_template']
+                self.object.save()
+            except ValidationError as v_err:
+                f.add_error('storage_name', '. '.join(v_err.messages))
+        context = self.get_context_data(**kwargs)
+        context['change_storage_form'] = f
+        return self.render_to_response(context)
+
+    def handle_del_storage_post(self, request, **kwargs):
+        parent = self.object.parent_storage
+        try:
+            self.object.delete()
+        except:
+            context = self.get_context_data(**kwargs)
+            context['delete_storage_errors'] = ['Error deleting Storage '+str(self.object)]
+            return self.render_to_response(context)
+
+        if parent is None:
+            return redirect('parts-stocks')
+        else:
+            return redirect(reverse('parts-stocks-detail', kwargs={'uuid':parent.id}))
+
+    def handle_del_stock_post(self, request, **kwargs):
+        del_error = None # TODO: Check error handling. This is clearly not working as intended :P
+        if 'stock_uuid' in request.POST:
+            f = DeleteStockForm(data=request.POST)
+            if f.is_valid():
+                try:
+                    s = Stock.objects.get(id=f.cleaned_data['stock_uuid'])
+                    print(s.storage)
+                    print(self.object)
+                    if s.storage == self.object:
+                       s.delete()
+                    else:
+                        del_error = 'Cannot delete stock from another storage.'
+                except:
+                    del_error = 'Could not find requested stock in this storage.'
+
+        context = self.get_context_data(**kwargs)
+        return self.render_to_response(context)
+
+    def handle_update_watermark(self, request, **kwargs):
+        edit_form = EditWatermarkForm(data=request.POST)
+        if edit_form.is_valid():
+            edit_form.save()
+        else:
+            pass # Todo: Handle error
+
+        context = self.get_context_data(**kwargs)
+        return self.render_to_response(context)
+
+    def handle_amount_change_post(self, request, increase, **kwargs):
+        edit_form = EditStockAmountForm(data=request.POST)
+        if edit_form.is_valid():
+            edit_form.save(increase)
+
+        context = self.get_context_data(**kwargs)
+        return self.render_to_response(context)
+
+    def handle_add_stock_post(self, request, **kwargs):
+        f = AddStockForm(data=request.POST)
+        error_occured = False
+
+        if f.is_valid():
+            try:
+                f.save(self.object)
+            except Exception as ex:
+                f.add_error('', str(ex))
+                error_occured = True
+        else:
+            error_occured = True
+
+        context = self.get_context_data(**kwargs)
+        if error_occured:
+            context['add_stock_form'] = f
+        return self.render_to_response(context)
+
+    def post(self, request, *args, **kwargs):
+        self.object = self.get_object()
+
+        if 'submit-add-storage' in request.POST:
+            return self.handle_add_storage_post(request, **kwargs)
+        elif 'submit-change-storage' in request.POST:
+            return self.handle_change_storage_post(request, **kwargs)
+        elif 'submit-delete-storage' in request.POST:
+            return self.handle_del_storage_post(request, **kwargs)
+        elif 'submit-delete-stock' in request.POST:
+            return self.handle_del_stock_post(request, **kwargs)
+        elif 'submit-edit-watermark' in request.POST:
+            return self.handle_update_watermark(request, **kwargs)
+        elif 'submit-amount-reduce' in request.POST:
+            return self.handle_amount_change_post(request, False, **kwargs)
+        elif 'submit-amount-increase' in request.POST:
+            return self.handle_amount_change_post(request, True, **kwargs)
+        elif 'submit-add-stock' in request.POST:
+            return self.handle_add_stock_post(request, **kwargs)
+
+        return super().post(request, *args, **kwargs)

shimatta_kenkyusho/shimatta_kenkyusho/settings_production.py

@@ -16,9 +16,13 @@ The following environment variables have to be set:
 - DJANGO_MEDIA_URL
 - DJANGO_MEDIA_ROOT
 - DJANGO_POSTGRESQL_SOCKET
+- DJANGO_POSTGRESQL_PORT
 
 The following can be set
+- DJANGO_POSTGRESQL_PW (assumed empty if missing)
+- DJANGO_POSTGRESQL_USER (assmumed empty if mssing)
 - DJANGO_SECURE_HSTS_SECONDS (defaults to 120)
+- DJANGO_FORCE_DEV_MODE
 
 """
 
@@ -46,9 +50,11 @@ SECRET_KEY = get_env_value('DJANGO_SECRET_KEY')
 
 # SECURITY WARNING: don't run with debug turned on in production!
 DEBUG = False
+if get_env_value('DJANGO_FORCE_DEV_MODE', default=False) == 'True':
+    DEBUG = True
 
-ALLOWED_HOSTS = ['localhost', get_env_value('DJANGO_ALLOWED_HOST')]
 
+ALLOWED_HOSTS = ['127.0.0.1', 'localhost', get_env_value('DJANGO_ALLOWED_HOST')]
 
 # Application definition
 
@@ -138,6 +144,7 @@ DATABASES = {
         'USER': db_user,
         'PASSWORD': db_pw,
         'HOST': get_env_value('DJANGO_POSTGRESQL_SOCKET'),
+        'PORT': get_env_value('DJANGO_POSTGRESQL_PORT'),
     }
 }
 
@@ -229,6 +236,9 @@ CRISPY_TEMPLATE_PACK = "bootstrap5"
 SESSION_COOKIE_SECURE = True
 CSRF_COOKIE_SECURE = True
 
-SECURE_SSL_REDIRECT = True
+SECURE_SSL_REDIRECT = False
+
+# allow detection of https behind "old" nginx
+SECURE_PROXY_SSL_HEADER = ("HTTP_X_FORWARDED_PROTO", "https")
 
 SECURE_HSTS_SECONDS = get_env_value('DJANGO_SECURE_HSTS_SECONDS', default=120)

shimatta_kenkyusho/shimatta_kenkyusho/urls.py

@@ -14,14 +14,13 @@ Including another URLconf
     2. Add a URL to urlpatterns:  path('blog/', include('blog.urls'))
 """
 from django.contrib import admin
-from django.urls import path, include
+from django.urls import path, include, re_path
-from django.conf.urls import url
 from django.conf.urls.static import static
 from django.conf import settings
 from parts import views as parts_views
 
 urlpatterns = [
-    url(r'^admin/login/', parts_views.login_view),
+    re_path(r'^admin/login/', parts_views.login_view),
     path('admin/', admin.site.urls),
     path('api/v1/', include('api.urls'), name='api-root'),
     path('', include('parts.urls')),

shimatta_kenkyusho/static/example/import_csv/import_csv.csv

@@ -0,0 +1,2 @@
+name;manufacturer;component_type;pref_distri;description;datasheet_link;package;image_url;storage_uuid;substorage_path;amount;lot;watermark;param:Resistance;param:Temperature coefficient;param:Tolerance;distri:Mouser
+1K41;Phycomp;Resistor;;KVG - expired;;R0603;https://t3.ftcdn.net/jpg/02/95/44/22/360_F_295442295_OXsXOmLmqBUfZreTnGo9PREuAPSLQhff.jpg;a4909bb4-4d69-4e58-8fbd-fba4fa62d9fb;bew;1111;4000;0;1330;100;1;gggg

shimatta_kenkyusho/static/example/import_csv/readme.md

@@ -0,0 +1,14 @@
+# The shimatta kenkyusho can import components from a CSV file uploaded to the website
+
+All parameters are passed as names (e.g. package, manufacturer etc.).
+
+Component parameters can be set using a prefix ``param:`` followed by the
+parameter name. The value is then parsed bysed on the parameter type.
+
+Distributor part numbers can be set similarly by prepending ``distri:``.
+
+It is also possible to create initial stocks to one storage by passing the
+storage uuid and/or the path to the storage (as printed in the breadcrumbs) or
+a combination of a storage uuid and the path from this storage.
+
+See example for details.

shimatta_kenkyusho/static/js/autocomplete-foreign-key-field.js

@@ -12,9 +12,9 @@ function initialize_autocompletion_foreign_key_field(search_element) {
     var name_field_name = search_element.getAttribute('data-ac-name-field');
     var search_url = search_element.getAttribute('data-ac-url');
     var base_id = search_element.getAttribute('id');
-    var uuid_field = search_element.parentElement.querySelector('#'+base_id+'-uuid-field');
+    var uuid_field = search_element.parentElement.parentElement.querySelector('#'+base_id+'-uuid-field');
-    var dflex_container = search_element.parentElement.querySelector('#'+base_id+'-dflex-container');
+    var dflex_container = search_element.parentElement.parentElement.querySelector('#'+base_id+'-dflex-container');
-    var initial_delete_button = search_element.parentElement.querySelector('[data-ac-delete]');
+    var initial_delete_button = search_element.parentElement.parentElement.querySelector('[data-ac-delete]');
 
     console.log(initial_delete_button);
     console.log(image_field_name);
@@ -27,6 +27,17 @@ function initialize_autocompletion_foreign_key_field(search_element) {
         });
     }
 
+    // select first match if any on enter
+    search_element.addEventListener('keydown', (e) => {
+        if (e.key === "Enter") {
+            e.preventDefault();
+            first_search_result = search_element.parentElement.querySelector('li')
+            if (first_search_result) {
+                first_search_result.click()
+            }
+        }
+    });
+
     new AutocompleteCustomUi(
         base_id, base_id+'-ac-ul', function(search_query, autocomplete_obj) {
             api_ajax_request_without_send('GET', search_url+`?search=${encodeURIComponent(search_query)}`, function(method, url, json) {

shimatta_kenkyusho/static/js/autocomplete.js

@@ -38,7 +38,9 @@ class AutocompleteCustomUi {
         this.query_callback = query_function.bind(this);
     
         document.getElementById(text_id).addEventListener("keyup", this.ac_delay(function(event) {
+            if (event.key != 'Enter') {
                 this.query_callback(document.getElementById(this.text_id).value, this);
+            }
         }, autocomplete_query_delay_ms).bind(this));
 
         this.dropdown_data = {};

shimatta_kenkyusho/static/js/kenkyusho-api-v1.js

@@ -29,6 +29,10 @@ function api_search_user(search, onSuccess, onFail) {
     return api_ajax_request_without_send('GET', api_urls_v1['user-list']+`?search=${encodeURIComponent(search)}`, function(method, url, json) {onSuccess(json);}, onFail);
 }
 
+function api_search_storage_template(search, onSuccess, onFail) {
+    return api_ajax_request_without_send('GET', api_urls_v1['storage-template-list']+`?search=${encodeURIComponent(search)}`, function(method, url, json) {onSuccess(json);}, onFail);
+}
+
 function api_search_component(search, onSuccess, onFail) {
     return api_ajax_request_without_send('GET', api_urls_v1['component-list']+`?search=${encodeURIComponent(search)}`, function(method, url, json) {onSuccess(json);}, onFail);
 }

shimatta_kenkyusho/templates/base.html

@@ -65,6 +65,7 @@
             'user-list': '{% url 'user-list' %}',
             'groups-list': '{% url 'user-list' %}',
             'storage-list': '{% url 'storage-list' %}',
+            'storage-template-list': '{% url 'storage-template-list' %}',
             'component-list': '{% url 'component-list' %}',
             'package-list': '{% url 'package-list' %}',
             'stock-list': '{% url 'stock-list' %}',
@@ -74,9 +75,22 @@
             'component-parameter-type-list': '{% url 'componentparametertype-list' %}',
         };
         </script>
-        <script type="text/javascript" src="{% static 'js/kenyusho-api-v1.js' %}"></script>
+        <script type="text/javascript" src="{% static 'js/kenkyusho-api-v1.js' %}"></script>
         <script type="text/javascript" src="{% static 'js/autocomplete.js' %}"></script>
         <script type="text/javascript" src="{% static 'js/autocomplete-foreign-key-field.js' %}"></script>
+        <!-- Initialize bootstrap popovers -->
+        <script type="text/javascript">
+            const popoverTriggerList = document.querySelectorAll('[data-bs-toggle="popover"]')
+            const popoverList = [...popoverTriggerList].map(popoverTriggerEl => new bootstrap.Popover(popoverTriggerEl))
+        </script>
+        <!-- Select search field on start of QR scan if no input is currently selevted([) -->
+        <script type="text/javascript">
+            window.addEventListener("keydown", (event)=>{
+                if (document.activeElement.nodeName != 'INPUT' && event.key == '[') {
+                    document.getElementById("qr_search_field").focus()
+                }
+            })
+        </script>
         {% block custom_scripts %}
         {% endblock custom_scripts %}
 

shimatta_kenkyusho/templates/parts/components.html

@@ -14,6 +14,7 @@
                     </button>
                     <button class="btn btn-secondary" type="button" data-bs-toggle="collapse" href="#advanced-search-collapse">Advanced <i class="bi bi-search"></i></button>
                     <button class="btn btn-success" type="button" data-bs-toggle="modal" data-bs-target="#comp-edit-modal"><i class="bi bi-plus-circle"></i> Add Component</button>
+                    <button class="btn btn-success" type="button" data-bs-toggle="modal" data-bs-target="#comp-import-modal"><i class="bi bi-plus-circle"></i> Import CSV</button>
                 </div>
             </form>
             <div class="collapse mb-3{% if advanced_search_shown %} show{% endif %}" id="advanced-search-collapse" aria-expanded="{% if advanced_search_shown %}true{% else %}false{% endif %}">
@@ -64,6 +65,7 @@
 </div>
 
 {% include 'parts/modals/edit-component-modal.html' with form=comp_form heading='New Component' open_component_button=True %}
+{% include 'parts/modals/import-component-modal.html' with form=import_comp_form %}
 
 {% endblock content %}
 {% block custom_scripts %}
@@ -72,5 +74,8 @@
 {% if comp_form.errors %}
 bootstrap.Modal.getOrCreateInstance(document.getElementById('comp-edit-modal')).show()
 {% endif %}
+{% if import_comp_form.errors %}
+bootstrap.Modal.getOrCreateInstance(document.getElementById('comp-import-modal')).show()
+{% endif %}
 </script>
 {% endblock custom_scripts %}

shimatta_kenkyusho/templates/parts/modals/add-substorage-modal.html

@@ -0,0 +1,22 @@
+{% load static %}
+{% load crispy_forms_tags %}
+<div class="modal fade" id="add-sub-modal">
+    <div class="modal-dialog">
+        <div class="modal-content">
+            <div class="modal-header">
+                <h5 class="modal-title">Add Storage</h5>
+                <button type="button" class="btn-close" data-bs-dismiss="modal" aria-label="Close"></button>
+            </div>
+            <form method="post" enctype="multipart/form-data">
+                {% csrf_token %}
+                <div class="modal-body">
+                    {{form|crispy}}
+                </div>
+                <div class="modal-footer">
+                    <button type="button" class="btn btn-secondary" data-bs-dismiss="modal">Close</button>
+                    <input type="submit" class="btn btn-primary" value="Add Storage" name="submit-add-storage">
+                </div>
+            </form>
+        </div>
+    </div>
+</div>

shimatta_kenkyusho/templates/parts/modals/change-storage-modal.html

@@ -0,0 +1,22 @@
+{% load static %}
+{% load crispy_forms_tags %}
+<div class="modal fade" id="change-modal">
+    <div class="modal-dialog">
+        <div class="modal-content">
+            <div class="modal-header">
+                <h5 class="modal-title">Change Storage</h5>
+                <button type="button" class="btn-close" data-bs-dismiss="modal" aria-label="Close"></button>
+            </div>
+            <form method="post" enctype="multipart/form-data">
+                {% csrf_token %}
+                <div class="modal-body">
+                    {{form|crispy}}
+                </div>
+                <div class="modal-footer">
+                    <button type="button" class="btn btn-secondary" data-bs-dismiss="modal">Close</button>
+                    <input type="submit" class="btn btn-primary" value="Change Storage" name="submit-change-storage">
+                </div>
+            </form>
+        </div>
+    </div>
+</div>

shimatta_kenkyusho/templates/parts/modals/import-component-modal.html

@@ -0,0 +1,38 @@
+{% load static %}
+{% load crispy_forms_tags %}
+
+<div class="modal fade" id="comp-import-modal" tabindex="-1">
+    <div class="modal-dialog">
+        <div class="modal-content">
+            <div class="modal-header">
+                <h2>Import Component CSV</h2>
+                <button type="button"
+                        class="btn"
+                        data-bs-toggle="popover"
+                        data-bs-title="Import Component CSV"
+                        data-bs-html="true"
+                        data-bs-content="The shimatta kenkyusho can import components from a CSV file uploaded to the website.<br>
+                                         All parameters are passed as names (e.g. package, manufacturer etc.).<br><br>
+                                         Component parameters can be set using a prefix <code>param:</code> followed by the
+                                         parameter name. The value is then parsed bysed on the parameter type.<br><br>
+                                         Distributor part numbers can be set similarly by prepending <code>distri:</code>.<br><br>
+                                         It is also possible to create initial stocks to one storage by passing the
+                                         storage uuid and/or the path to the storage (as printed in the breadcrumbs) or
+                                         a combination of a storage uuid and the path from this storage.<br><br>
+                                         See <a href='{% static 'example/import_csv/import_csv.csv' %}'>example</a> for details.">
+                    <i class="bi bi-info-circle"></i>
+                </button>
+                <button type="button" class="btn-close" data-bs-dismiss="modal" aria-label="Close"></button>
+            </div>
+            <form method="post" enctype="multipart/form-data">
+                {% csrf_token %}
+                <div class="modal-body">
+                    {{form|crispy}}
+                </div>
+                <div class="modal-footer">
+                    <input type="submit" name="submit-import-components" class="btn btn-primary" value="Save">
+                </div>
+            </form>
+        </div>
+    </div>
+</div>

shimatta_kenkyusho/templates/parts/modals/new-substorage-modal.html

@@ -1,46 +0,0 @@
-{% load static %}
-<div class="modal fade" id="add-sub-modal">
-    <div class="modal-dialog">
-        <div class="modal-content">
-            <div class="modal-header">
-                <h5 class="modal-title">Add Storage</h5>
-                <button type="button" class="btn-close" data-bs-dismiss="modal" aria-label="Close"></button>
-            </div>
-            <form action="" method="post">
-                {% csrf_token %}
-                <div class="modal-body">
-                    <label for="add-storage-name" class="form-label">Storage Name</label>
-                    <div class="input-group has-validation">
-                        <input value="{{form.storage_name.value}}" class="form-control{% if form.storage_name.errors or form.non_field_errors %} is-invalid{% endif %}" id="add-storage-name" name="{{form.storage_name.name}}" type="text" aria-describedby="validationStorageName" required>
-                        <div id="validationStorageName" class="invalid-feedback">
-                            {% for msg in form.storage_name.errors %}
-                                {{msg}}
-                            {% endfor %}
-                        </div>
-                    </div>
-                    <label for="{{form.responsible.id_for_label}}">Responsible</label>
-                    <div class="input-group has-validation dropdown">
-                        <span class="input-group-text" id="add_storage_username_prepend">@</span><input autocomplete="off" data-bs-toggle="dropdown" type="text" value="{{form.responsible.value}}" class="form-control{% if form.responsible.errors or form.non_field_errors %} is-invalid{% endif %}" id="{{form.responsible.id_for_label}}" name="{{form.responsible.name}}" aria-describedby="add_storage_username_prepend validationServerUsernameFeedback" required>
-                        <ul class="dropdown-menu" aria-labelledby="{{form.responsible.id_for_label}}" id="{{form.responsible.id_for_label}}-ac-dropdown">
-                        </ul>
-                        <div id="validationServerUsernameFeedback" class="invalid-feedback">
-                            {% for msg in form.responsible.errors %}
-                                {{msg}}
-                            {% endfor %}
-                        </div>
-                    </div>
-                </div>
-                <div class="modal-footer">
-                    <button type="button" class="btn btn-secondary" data-bs-dismiss="modal">Close</button>
-                    <input type="submit" class="btn btn-primary" value="Add Storage" name="submit-add-storage">
-                    
-                    {% if form.non_field_errors %}
-                        {% for error in form.non_field_errors %}
-                        <p class="text-danger text-center">{{ error }}</p>
-                        {% endfor %}
-                    {% endif %}
-                </div>
-            </form>
-        </div>
-    </div>
-</div>

shimatta_kenkyusho/templates/parts/stocks-detail.html

@@ -20,6 +20,9 @@
             <div class="row">
                 {% qr_from_text object.get_qr_code size="m" image_format="svg" %}
             </div>
+            <div class="row">
+                <h4>{{storage.name}}{% if storage.verbose_name %}<small> ({{storage.verbose_name}})</small>{% endif %}</h4>
+            </div>
             <div class="row">
                 {% if object.parent_storage %}
                 <h1>Sub-Storages <a class="btn btn-secondary" href="{% url 'parts-stocks-detail' uuid=object.parent_storage.id %}">Parent Storage</a>            {% else %}
@@ -27,13 +30,14 @@
                 {% endif %}
                 <button type="button" class="btn btn-danger" data-bs-toggle="modal" data-bs-target="#delete-storage-modal">Delete</button>
                 <button type="button" class="btn btn-primary" data-bs-toggle="modal" data-bs-target="#add-sub-modal"><i class="bi bi-plus-circle"></i></button>
+                <button type="button" class="btn btn-primary" data-bs-toggle="modal" data-bs-target="#change-modal"><i class="bi bi-pencil-square"></i></button>
                 </h1>
                 <div class="list-group">
                 {% for storage in storages %}
                     <a href="{% url 'parts-stocks-detail' uuid=storage.id %}" class="text-decoration-none">
                         <li class="list-group-item list-group-item-action justify-content-between align-items-center d-flex">
                             <div> 
-                            <h5>{{storage.name}}</h5>
+                            <h5>{{storage.name}}{% if storage.verbose_name %}<small> ({{storage.verbose_name}})</small>{% endif %}</h5>
                             Responsible: {{ storage.responsible }}
                             </div>
                             <span class="badge bg-primary rounded-pill">{{storage.get_total_stock_amount}}</span>
@@ -109,7 +113,11 @@
     {% endfor %}
     <!-- Modal for adding a substorage-->
     {% with add_storage_form as form %}
-    {% include 'parts/modals/new-substorage-modal.html' %}
+    {% include 'parts/modals/add-substorage-modal.html' %}
+    {% endwith %}
+    <!-- Modal to change current storag-->
+    {% with change_storage_form as form %}
+    {% include 'parts/modals/change-storage-modal.html' %}
     {% endwith %}
     <!-- Modal for deleting this storage -->
     {% with delete_storage_errors as err_msgs %}
@@ -158,18 +166,6 @@ api_get_component_from_id(uuid, function(component){
 }
 {% endif %}
 
-new AutocompleteText('{{add_storage_form.responsible.id_for_label}}', '{{add_storage_form.responsible.id_for_label}}-ac-dropdown', 
-function(search, autocomplete_obj) {
-    api_search_user(search, function(results) {
-        var usernames = new Array();
-        console.log(results);
-        for (var i = 0; i < results.results.length; i++) {
-            usernames.push(results.results[i].username);
-        }
-        console.log(usernames);
-        autocomplete_obj.show_results(usernames);
-    }, function(){});
-});
 </script>
 {% endblock custom_scripts %}
 

shimatta_kenkyusho/templates/parts/stocks.html

@@ -40,7 +40,7 @@
                     <a href="{% url 'parts-stocks-detail' uuid=storage.id %}" class="text-decoration-none">
                         <li class="list-group-item list-group-item-action justify-content-between align-items-center d-flex">
                             <div> 
-                            <h5>{{storage.name}}</h5>
+                            <h5>{{storage.name}}{% if storage.verbose_name %}<small> ({{storage.verbose_name}})</small>{% endif %}</h5>
                             Responsible: {{ storage.responsible }}
                             </div>
                             <span class="badge bg-primary rounded-pill">{{storage.get_total_stock_amount}}</span>
@@ -54,7 +54,7 @@
 
     <!-- Add storage modal form -->
     {% with add_storage_form as form %}
-    {% include 'parts/modals/new-substorage-modal.html' %}
+    {% include 'parts/modals/add-substorage-modal.html' %}
     {% endwith %}
 
 </div>
@@ -67,18 +67,6 @@
     var modal = bootstrap.Modal.getOrCreateInstance(addSubStorageModal);
     modal.show();
     {% endif %}
-    new AutocompleteText('{{add_storage_form.responsible.id_for_label}}', '{{add_storage_form.responsible.id_for_label}}-ac-dropdown', 
-    function(search, autocomplete_obj) {
-        api_search_user(search, function(results) {
-            var usernames = new Array();
-            console.log(results);
-            for (var i = 0; i < results.results.length; i++) {
-                usernames.push(results.results[i].username);
-            }
-            console.log(usernames);
-            autocomplete_obj.show_results(usernames);
-        }, function(){});
-    });
 
 </script>
 {% endblock custom_scripts %}

shimatta_kenkyusho/templates/widgets/autocomplete-foreign-key.html

@@ -1,7 +1,12 @@
 <div class="dropdown">
+    <div class="input-group">
+        {% if custom.prepend %}
+        <span class="input-group-text" id="{{widget.attrs.id}}-prepend">{{custom.prepend}}</span>
+        {% endif %}
         <input autocomplete="off" id="{{widget.attrs.id}}" data-ac-url="{{custom.search_url}}" data-ac-name-field="{{custom.name_field_name}}" {% if custom.image_field_name %}data-ac-image-field="{{custom.image_field_name}}"{% endif %} data-bs-toggle="dropdown" type="text" placeholder="Search..." class="{{widget.attrs.class}}">
         <ul id="{{widget.attrs.id}}-ac-ul" class="dropdown-menu">
         </ul>
+    </div>
     <div class="d-flex align-items-center mt-3 mb-3" id="{{widget.attrs.id}}-dflex-container">
         {% if custom.current_instance %}
         {% if custom.image_field_name %}

start_docker_compose_interactive.sh

@@ -0,0 +1,7 @@
+#!/bin/bash
+
+# Startup the db container
+docker compose start shimatta-kenkyusho-db
+
+# Override entrypoint to get interactive shell
+docker compose run --entrypoint="/bin/sh" -p 8000:8000 shimatta-kenkyusho-web

start_server.sh

@@ -0,0 +1 @@
+podman run -it -e DJANGO_SECRET_KEY=<secret_key> -e DJANGO_ALLOWED_HOST=parts.shimatta.net -e DJANGO_STATIC_ROOT=/var/static -e DJANGO_MEDIA_URL=media -e DJANGO_MEDIA_ROOT=/var/media -e DJANGO_POSTGRESQL_SOCKET=host.docker.internal -e DJANGO_POSTGRESQL_PORT=2345 -e DJANGO_POSTGRESQL_USER=<db_user> -e DJANGO_POSTGRESQL_PW=<db_pass> -v /var/parts/static:/var/static -v /var/parts/media:/var/media -p 8000:8000 --entrypoint /bin/sh localhost/kenkyusho:0.1