kpsul/kfet/management/commands/loadkfetdevdata.py

162 lines
4.8 KiB
Python
Raw Normal View History

2017-02-08 02:22:31 +01:00
"""
Crée des utilisateurs, des articles et des opérations aléatoires
"""
import os
import random
from datetime import timedelta
from django.contrib.auth.models import Group, Permission, User
2017-03-19 20:03:39 +01:00
from django.core.management import call_command
from django.utils import timezone
2017-02-08 02:22:31 +01:00
from gestioncof.management.base import MyBaseCommand
from gestioncof.models import CofProfile
from kfet.models import (
Account,
Article,
Checkout,
CheckoutStatement,
Supplier,
SupplierArticle,
)
2017-02-08 02:22:31 +01:00
# Où sont stockés les fichiers json
DATA_DIR = os.path.join(os.path.dirname(os.path.dirname(__file__)), "data")
2017-02-08 02:22:31 +01:00
class Command(MyBaseCommand):
help = "Crée des utilisateurs, des articles et des opérations aléatoires"
def handle(self, *args, **options):
# ---
# Groupes
# ---
Group.objects.filter(name__icontains="K-Fêt").delete()
2017-02-08 02:22:31 +01:00
group_chef = Group(name="K-Fêt César")
group_boy = Group(name="K-Fêt Légionnaire")
group_chef.save()
group_boy.save()
2020-08-28 18:00:54 +02:00
permissions_chef = Permission.objects.filter(
content_type__app_label="kfet",
)
2017-02-08 02:22:31 +01:00
permissions_boy = Permission.objects.filter(
content_type__app_label="kfet", codename__in=["is_team", "perform_deposit"]
)
2017-02-08 02:22:31 +01:00
group_chef.permissions.add(*permissions_chef)
group_boy.permissions.add(*permissions_boy)
# ---
# Comptes
# ---
2017-02-08 18:54:49 +01:00
self.stdout.write("Création des comptes K-Fêt")
gaulois = CofProfile.objects.filter(user__last_name="Gaulois")
gaulois_trigramme = map("{:03d}".format, range(50))
2017-02-08 02:22:31 +01:00
romains = CofProfile.objects.filter(user__last_name="Romain")
romains_trigramme = map(lambda x: str(100 + x), range(99))
2017-02-08 02:22:31 +01:00
2017-02-08 18:54:49 +01:00
created_accounts = 0
team_accounts = 0
2017-02-08 02:22:31 +01:00
for (profile, trigramme) in zip(gaulois, gaulois_trigramme):
2017-02-08 18:54:49 +01:00
account, created = Account.objects.get_or_create(
2017-02-08 02:22:31 +01:00
trigramme=trigramme,
2017-02-08 18:11:38 +01:00
cofprofile=profile,
defaults={"balance": random.randint(1, 999) / 10},
2017-02-08 02:22:31 +01:00
)
2017-02-08 18:54:49 +01:00
created_accounts += int(created)
2017-02-08 02:22:31 +01:00
if profile.user.first_name == "Abraracourcix":
2017-02-08 02:22:31 +01:00
profile.user.groups.add(group_chef)
for (profile, trigramme) in zip(romains, romains_trigramme):
2017-02-08 18:54:49 +01:00
account, created = Account.objects.get_or_create(
2017-02-08 02:22:31 +01:00
trigramme=trigramme,
2017-02-08 18:11:38 +01:00
cofprofile=profile,
defaults={"balance": random.randint(1, 999) / 10},
2017-02-08 02:22:31 +01:00
)
2017-02-08 18:54:49 +01:00
created_accounts += int(created)
2017-02-08 02:22:31 +01:00
2017-02-08 18:54:49 +01:00
if random.random() > 0.75 and created:
2017-02-08 02:22:31 +01:00
profile.user.groups.add(group_boy)
2017-02-08 18:54:49 +01:00
team_accounts += 1
self.stdout.write(
"- {:d} comptes créés, {:d} dans l'équipe K-Fêt".format(
created_accounts, team_accounts
)
)
2017-02-08 02:22:31 +01:00
# Compte liquide
2017-03-18 23:02:08 +01:00
self.stdout.write("Création du compte liquide")
liq_user, _ = User.objects.get_or_create(username="liquide")
2017-02-08 02:22:31 +01:00
liq_profile, _ = CofProfile.objects.get_or_create(user=liq_user)
liq_account, _ = Account.objects.get_or_create(
cofprofile=liq_profile, trigramme="LIQ"
)
2017-02-08 02:22:31 +01:00
2017-03-18 22:02:33 +01:00
# Root account if existing
root_profile = CofProfile.objects.filter(user__username="root")
2017-03-18 22:02:33 +01:00
if root_profile.exists():
2017-03-18 23:02:08 +01:00
self.stdout.write("Création du compte K-Fêt root")
2017-03-18 22:02:33 +01:00
root_profile = root_profile.get()
Account.objects.get_or_create(cofprofile=root_profile, trigramme="AAA")
2017-03-18 22:02:33 +01:00
2017-02-08 18:26:56 +01:00
# ---
# Caisse
# ---
2017-03-18 22:02:23 +01:00
checkout, created = Checkout.objects.get_or_create(
created_by=Account.objects.get(trigramme="000"),
name="Chaudron",
2017-03-18 22:02:23 +01:00
defaults={
"valid_from": timezone.now(),
"valid_to": timezone.now() + timedelta(days=730),
2017-03-18 22:02:23 +01:00
},
2017-02-08 18:26:56 +01:00
)
2017-03-18 22:02:23 +01:00
if created:
CheckoutStatement.objects.create(
by=Account.objects.get(trigramme="000"),
2017-03-18 22:02:23 +01:00
checkout=checkout,
balance_old=0,
balance_new=0,
amount_taken=0,
amount_error=0,
2017-03-18 22:02:23 +01:00
)
2017-03-30 04:16:40 +02:00
# ---
# Fournisseur
# ---
supplier, created = Supplier.objects.get_or_create(name="Panoramix")
if created:
2017-03-30 04:22:18 +02:00
articles = random.sample(list(Article.objects.all()), 40)
2017-03-30 04:16:40 +02:00
to_create = []
for article in articles:
to_create.append(SupplierArticle(supplier=supplier, article=article))
2017-03-30 04:16:40 +02:00
SupplierArticle.objects.bulk_create(to_create)
2017-02-08 02:22:31 +01:00
# ---
# Opérations
# ---
call_command("createopes", "100", "7", "--transfers=20")
# ---
# Wagtail CMS
# ---
call_command("kfet_loadwagtail")