From 5323991d06a1c0bf98195a4634cf25292474a447 Mon Sep 17 00:00:00 2001 From: Andrey Rusakov Date: Mon, 30 Sep 2024 18:55:42 +0200 Subject: [PATCH] Wrap needed with get_or_create --- api/management/commands/prepareusertests.py | 2 +- api/management/commands/seed.py | 357 +++++++++++++++++++- 2 files changed, 352 insertions(+), 7 deletions(-) diff --git a/api/management/commands/prepareusertests.py b/api/management/commands/prepareusertests.py index 56ad919d..fd1cae41 100644 --- a/api/management/commands/prepareusertests.py +++ b/api/management/commands/prepareusertests.py @@ -20,7 +20,7 @@ class Command(BaseCommand): """ def handle(self, *args, **options): # Set all ready orders to ARCHIVED - orders_ready = Order.objects.filter(status=Order.OrderStatus.READY).all() + orders_ready = Order.objects.filter(order_status=Order.OrderStatus.READY).all() for order in orders_ready: order.order_status = Order.OrderStatus.ARCHIVED order.save() diff --git a/api/management/commands/seed.py b/api/management/commands/seed.py index 9390d0aa..925930c1 100644 --- a/api/management/commands/seed.py +++ b/api/management/commands/seed.py @@ -1,15 +1,360 @@ +import os +import datetime +from pathlib import Path +from django.db import models +from django.utils import timezone +from django.conf import settings +from django.core.files.uploadedfile import SimpleUploadedFile from django.core.management.base import BaseCommand -import random +from django.contrib.auth import get_user_model +from django.contrib.auth.models import User +from api.models import Contact, Order, OrderItem, OrderType, Product, DataFormat, Identity, Metadata, Pricing, Money +from django.contrib.gis.geos import Polygon +from api.helpers import _zip_them_all +from typing import TypeVar, Generic, Type +from collections.abc import MutableMapping + +T = TypeVar('T', bound=models.Model) +UserModel = get_user_model() class Command(BaseCommand): help = "seed database for testing and development." def handle(self, *args, **options): - print('Seeding data...') - seed(self) - print('Done.') + self.stdout.write("Seeding data...\n") + self.seed() + self.stdout.write("Done.\n") + + def notice(self, msg: str): + self.stdout.write(self.style.NOTICE(msg)) + + def success(self, msg: str): + self.stdout.write(self.style.SUCCESS(msg)) + + def error(self, msg: str): + self.stdout.write(self.style.ERROR(msg)) + + def getOrCreate(self, model: Type[T], defaults: MutableMapping[str, any] | None = ..., **kwargs) -> T: + (value, created) = model.objects.get_or_create(defaults, **kwargs) + typeName = model.__name__ + if created: + self.success(f"Created {typeName} \"{value}\"") + else: + self.notice(f"Not creating {typeName}: \"{value}\" - already exists") + return value + + def addUser(self, username: str, email: str, password: str) -> User: + return self.getOrCreate(UserModel, username=username, defaults={email: email, password: password}) + + def setIdentity(self, user: User, params: dict[str, any]) -> Identity: + for k, v in params.items(): + setattr(user.identity, k, v) + user.identity.save() + self.success(f"Updated identity for user '{user.username}'") + return user.identity + + def addContact(self, user: User, params: dict[str, any]) -> Contact: + params["belongs_to"] = user + contact = self.addContact(**params) + self.success(f"Created contact \"{contact.first_name} {contact.last_name}\"") + return contact + + def addOrderType(self, name: str) -> OrderType: + return self.getOrCreate(OrderType, name=name) + + def addProduct(self, user: User, label: str) -> OrderType: + return self.getOrCreate(Product, label=label, defaults={ + "metadata": self.getOrCreate(Metadata, name="metadata", defaults={"modified_user_id": user.id}), + "pricing": self.getOrCreate(Pricing, name="Free", defaults={"pricing_type": Pricing.PricingType.FREE}), + "provider": user + }) + + def addDataFormat(self, name: str) -> DataFormat: + return self.getOrCreate(DataFormat, name=name) + + def seed(self): + # Create users + rincevent = self.addUser( + 'rincevent', 'rincevent@mail.com', 'rincevent') + self.setIdentity(rincevent, { + "email": os.environ.get('EMAIL_TEST_TO', 'admin@admin.com'), + "first_name": 'Jean', + "last_name": 'Michoud', + "street": 'Rue de Tivoli 22', + "postcode": '2000', + "city": 'Neuchâtel', + "country": 'Suisse', + "company_name": 'Service du Registre Foncier et de la Géomatique - SITN', + "phone": '+41 32 000 00 00'}) + + mmi = self.addUser("mmi", "mmi@mmi.com", "mmi") + self.setIdentity(mmi, { + "email": os.environ.get('EMAIL_TEST_TO_ARXIT', 'admin@admin.com'), + "first_name": 'Jeanne', + "last_name": 'Paschoud', + "street": 'Rue de Tivoli 22', + "postcode": '2000', + "city": 'Neuchâtel', + "country": 'Suisse', + "company_name": 'Service du Registre Foncier et de la Géomatique - SITN', + "phone": '+41 32 000 00 00'}) + + mma = self.addUser("mma", "mma@mma.com", "mma") + self.setIdentity(mma, { + "email": 'mma-email@admin.com', + "first_name": 'Jean-René', + "last_name": 'Humbert-Droz L\'Authentique', + "street": 'Rue de Tivoli 22', + "postcode": '2000', + "city": 'Neuchâtel', + "country": 'Suisse', + "company_name": 'Service du Registre Foncier et de la Géomatique - SITN', + "phone": '+41 32 000 00 00'}) + + mka2 = self.addUser("mka", "mka@mka.com", "mka") + self.setIdentity(mka2, { + "email": 'mka2-email@ne.ch', + "first_name": 'Michaël', + "last_name": 'Kalbermatten', + "street": 'Rue de Tivoli 22', + "postcode": '2000', + "city": 'Neuchâtel', + "country": 'Suisse', + "company_name": 'Service du Registre Foncier et de la Géomatique - SITN', + "phone": '+41 32 000 00 00', + "subscribed": True}) + + # contacts + contact1 = Contact.objects.create( + first_name='Marc', + last_name='Riedo', + email='test@admin.com', + postcode=2000, + city='Neuchâtel', + country='Suisse', + company_name='SITN', + phone='+41 00 787 45 15', + belongs_to=mmi + ) + contact1.save() + contact2 = Contact.objects.create( + first_name='Marcelle', + last_name='Rieda', + email='test2@admin.com', + postcode=2000, + city='Neuchâtel', + country='Suisse', + company_name='SITN', + phone='+41 00 787 45 16', + belongs_to=mmi + ) + contact2.save() + contact3 = Contact.objects.create( + first_name='Jean', + last_name='Doe', + email='test3@admin.com', + postcode=2000, + city='Lausanne', + country='Suisse', + company_name='Marine de Colombier', + phone='+41 00 787 29 16', + belongs_to=mmi + ) + contact3.save() + contact_mka2 = Contact.objects.create( + first_name='Jean', + last_name='Doe', + email='test3@admin.com', + postcode=2000, + city='Lausanne', + country='Suisse', + company_name='Marine de Colombier', + phone='+41 00 787 29 16', + belongs_to=mka2 + ) + contact_mka2.save() + + order_geom = Polygon(( + ( + 2528577.8382161376, + 1193422.4003930448 + ), + ( + 2542482.6542869355, + 1193422.4329014618 + ), + ( + 2542482.568523701, + 1199018.36469272 + ), + ( + 2528577.807487005, + 1199018.324372703 + ), + ( + 2528577.8382161376, + 1193422.4003930448 + ) + )) + + order_type_prive = self.addOrderType('Privé') + + # Create orders + order1 = Order.objects.create( + title='Plan de situation pour enquête', + description='C\'est un test', + order_type=order_type_prive, + client=rincevent, + geom=order_geom, + invoice_reference='Dossier n°545454', + date_ordered=timezone.now()) + order1.save() + + order2 = Order.objects.create( + title='Plan de situation pour enquête', + description='C\'est un test', + order_type=order_type_prive, + client=rincevent, + geom=order_geom, + invoice_reference='Dossier n°545454', + date_ordered=timezone.now()) + order2.save() + + order3 = Order.objects.create( + title='Plan de situation pour enquête', + description='C\'est un test', + order_type=order_type_prive, + client=rincevent, + geom=order_geom, + invoice_reference='Dossier n°545454', + date_ordered=timezone.now()) + order3.save() + + order4 = Order.objects.create( + title='Plan de situation pour enquête', + description='C\'est un test', + order_type=order_type_prive, + client=mma, + geom=order_geom, + invoice_reference='Dossier n°545454', + date_ordered=timezone.now()) + order4.save() + + order_mka2 = Order.objects.create( + title='Plan de situation pour enquête', + description='C\'est un test', + order_type=order_type_prive, + client=mka2, + geom=order_geom, + invoice_reference='Dossier n°545454', + date_ordered=timezone.now()) + order_mka2.save() + + order_download = Order.objects.create( + title='Commande prête à être téléchargée', + description='C\'est un test', + order_type=order_type_prive, + client=mmi, + geom=order_geom, + invoice_reference='Dossier 8', + date_ordered=timezone.now()) + order_download.save() + + order_quoted = Order.objects.create( + title='Commande devisée pour test', + description='C\'est un test', + order_type=order_type_prive, + client=mmi, + geom=order_geom, + invoice_reference='Dossier n°545454', + date_ordered=timezone.now()) + order_quoted.save() + + # Products + product1 = self.addProduct(mma, 'MO - Cadastre complet') + product2 = self.addProduct(mma, 'Maquette 3D') + product_deprecated = self.addProduct(mma, + 'MO07 - Objets divers et éléments linéaires - linéaires') + + data_format = self.addDataFormat('Geobat NE complet (DXF)') + data_format_maquette = self.addDataFormat('3dm (Fichier Rhino)') + + orderitems = [ + OrderItem.objects.create(order=order1, product=product1), + OrderItem.objects.create(order=order1, product=product2), + OrderItem.objects.create(order=order_download, product=product1), + OrderItem.objects.create(order=order2, product=product1), + OrderItem.objects.create( + order=order3, product=product1, data_format=data_format), + OrderItem.objects.create(order=order4, product=product2), + OrderItem.objects.create( + order=order_mka2, product=product1, data_format=data_format) + ] + for order_item in orderitems: + order_item.set_price() + order_item.save() + order_item_deprecated = OrderItem.objects.create( + order=order_mka2, product=product_deprecated, data_format=data_format) + order_item_deprecated.set_price(price=Money( + 400, 'CHF'), base_fee=Money(150, 'CHF')) + order_item_deprecated.price_status = OrderItem.PricingStatus.CALCULATED + order_item_deprecated.save() + + order_item_download = OrderItem.objects.create( + order=order_download, product=product2, data_format=data_format_maquette) + order_item_download.set_price(price=Money( + 400, 'CHF'), base_fee=Money(150, 'CHF')) + order_item_download.price_status = OrderItem.PricingStatus.CALCULATED + order_item_download.save() + + order_item_quoted = OrderItem.objects.create( + order=order_quoted, product=product2, data_format=data_format_maquette) + order_item_quoted.set_price(price=Money( + 400, 'CHF'), base_fee=Money(150, 'CHF')) + order_item_quoted.price_status = OrderItem.PricingStatus.CALCULATED + order_item_quoted.save() + + order2.set_price() + order2.save() + + order3.set_price() + order3.confirm() + order3.invoice_contact = contact1 + order3.save() + + order4.save() + + order_mka2.invoice_contact = contact_mka2 + order_mka2.set_price() + order_mka2.date_ordered = datetime.datetime( + 2018, 12, 1, 8, 20, 3, 0, tzinfo=datetime.timezone.utc) + order_mka2.order_status = Order.OrderStatus.ARCHIVED + order_mka2.save() + + order_download.set_price() + empty_zip_data = b'PK\x05\x06\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00' + extract_file = SimpleUploadedFile( + "result.zip", empty_zip_data, content_type="multipart/form-data") + for order_item in order_download.items.all(): + order_item.extract_result = extract_file + order_item.status = OrderItem.OrderItemStatus.PROCESSED + order_item.save() + order_download.order_status = Order.OrderStatus.PROCESSED + # Creating zip with all zips without background process unsupported by manage.py + zip_list_path = list(order_download.items.all( + ).values_list('extract_result', flat=True)) + today = timezone.now() + zip_path = Path( + 'extract', + str(today.year), str(today.month), + "{}{}.zip".format('0a2ebb0a-', str(order_download.id))) + order_download.extract_result.name = zip_path.as_posix() + full_zip_path = Path(settings.MEDIA_ROOT, zip_path) + _zip_them_all(full_zip_path, zip_list_path) + order_download.save() -def seed(cmd: Command): - pass \ No newline at end of file + order_quoted.set_price() + order_quoted.quote_done() + order_quoted.save()