1
0
mirror of https://gitlab.crans.org/bde/nk20 synced 2025-04-23 15:12:37 +00:00

Compare commits

...

5 Commits

Author SHA1 Message Date
aeltheos
9816be1df5 Merge branch 'nix-shell' into 'main'
Nix shell

See merge request bde/nk20!201
2025-03-19 10:32:52 +01:00
quark
5ef019c5c2 Merge branch 'notekfet_wrapped' into 'main'
Rewrite script and add test

See merge request bde/nk20!300
2025-03-18 16:11:33 +01:00
quark
8da62e62fb Rewrite script and add test 2025-03-18 15:53:02 +01:00
Yoann Beaugnon
dde1baa25c typo 2022-08-21 19:50:53 +02:00
Yoann Beaugnon
7a7ee47e0b Add two shell.nix to enable easier development on nixos. 2022-08-21 19:46:11 +02:00
6 changed files with 216 additions and 77 deletions

1
.gitignore vendored
View File

@ -48,7 +48,6 @@ backups/
env/ env/
venv/ venv/
db.sqlite3 db.sqlite3
shell.nix
# ansibles customs host # ansibles customs host
ansible/host_vars/*.yaml ansible/host_vars/*.yaml

View File

@ -38,7 +38,7 @@ class Command(BaseCommand):
required=False, required=False,
help="""User will have their(s) wrapped generated, help="""User will have their(s) wrapped generated,
all = all users all = all users
adh = all users who have a valid memberships to BDE during the BDE considered adh = all users who have a valid cd memberships to BDE during the BDE considered
supersuser = all superusers supersuser = all superusers
custom user1,user2,... = a list of username, custom user1,user2,... = a list of username,
custom_id id1,id2,... = a list of user id""", custom_id id1,id2,... = a list of user id""",
@ -70,15 +70,7 @@ class Command(BaseCommand):
dest='create', dest='create',
) )
def handle(self, *args, **options): def handle(self, *args, **options): # NOQA
# useful string for output
red = '\033[31;1m'
yellow = '\033[33;1m'
green = '\033[32;1m'
abort = red + 'ABORT'
warning = yellow + 'WARNING'
success = green + 'SUCCESS'
# Traitement des paramètres # Traitement des paramètres
verb = options['verbosity'] verb = options['verbosity']
bde = [] bde = []
@ -89,11 +81,11 @@ class Command(BaseCommand):
if options['bde_id']: if options['bde_id']:
if bde: if bde:
if verb >= 1: if verb >= 1:
print(warning) self.stdout.write(self.style.WARNING(
print(yellow + 'You already defined bde with their name !') "WARNING\nYou already defined bde with their name !"))
if verb >= 0: if verb >= 0:
print(abort) self.stdout.write(self.style.ERROR("ABORT"))
return exit(1)
bde_id = options['bde_id'].split(',') bde_id = options['bde_id'].split(',')
bde = [Bde.objects.get(pk=i) for i in bde_id] bde = [Bde.objects.get(pk=i) for i in bde_id]
@ -113,11 +105,11 @@ class Command(BaseCommand):
user = ['custom_id', [User.objects.get(pk=u) for u in user_id]] user = ['custom_id', [User.objects.get(pk=u) for u in user_id]]
else: else:
if verb >= 1: if verb >= 1:
print(warning) self.sdtout.write(self.style.WARNING(
print(yellow + 'You user option is not recognized') "WARNING\nYou user option is not recognized"))
if verb >= 0: if verb >= 0:
print(abort) self.stdout.write(self.style.ERROR("ABORT"))
return exit(1)
club = [] club = []
if options['club']: if options['club']:
@ -133,11 +125,11 @@ class Command(BaseCommand):
club = ['custom_id', [Club.objects.get(pk=c) for c in club_id]] club = ['custom_id', [Club.objects.get(pk=c) for c in club_id]]
else: else:
if verb >= 1: if verb >= 1:
print(warning) self.stdout.write(self.style.WARNING(
print(yellow + 'You club option is not recognized') "WARNING\nYou club option is not recognized"))
if verb >= 0: if verb >= 0:
print(abort) self.stdout.write(self.style.ERROR("ABORT"))
return exit(1)
change = options['change'] change = options['change']
create = options['create'] create = options['create']
@ -145,72 +137,75 @@ class Command(BaseCommand):
# check if parameters are sufficient for generate wrapped with the desired option # check if parameters are sufficient for generate wrapped with the desired option
if not bde: if not bde:
if verb >= 1: if verb >= 1:
print(warning) self.stdout.write(self.style.WARNING(
print(yellow + 'You have not selectionned a BDE !') "WARNING\nYou have not selectionned a BDE !"))
if verb >= 0: if verb >= 0:
print(abort) self.stdout.write(self.style.ERROR("ABORT"))
return exit(1)
if not (user or club): if not (user or club):
if verb >= 1: if verb >= 1:
print(warning) self.stdout.write(self.style.WARNING(
print(yellow + 'No club or user selected !') "WARNING\nNo club or user selected !"))
if verb >= 0: if verb >= 0:
print(abort) self.stdout.write(self.style.ERROR("ABORT"))
return exit(1)
if verb >= 3: if verb >= 3:
print('\033[1mOptions:\033[m') self.stdout.write("Options:")
bde_str = '' bde_str = ''
for b in bde: for b in bde:
bde_str += str(b) bde_str += str(b) + '\n'
print('BDE: ' + bde_str) self.stdout.write("BDE: " + bde_str)
if user: if user:
print('User: ' + user[0]) self.stdout.write('User: ' + user[0])
if club: if club:
print('Club: ' + club[0]) self.stdout.write('Club: ' + club[0])
print('change: ' + str(change)) self.stdout.write('change: ' + str(change))
print('create: ' + str(create)) self.stdout.write('create: ' + str(create) + '\n')
print('')
if not (change or create): if not (change or create):
if verb >= 1: if verb >= 1:
print(warning) self.stdout.write(self.style.WARNING(
print(yellow + 'change and create is set to false, none wrapped will be created') "WARNING\nchange and create is set to false, none wrapped will be created"))
if verb >= 0: if verb >= 0:
print(abort) self.stdout.write(self.style.ERROR("ABORT"))
return exit(1)
if verb >= 1 and change: if verb >= 1 and change:
print(warning) self.stdout.write(self.style.WARNING(
print(yellow + 'change is set to true, some wrapped may be replaced !') "WARNING\nchange is set to true, some wrapped may be replaced !"))
if verb >= 1 and not create: if verb >= 1 and not create:
print(warning) self.stdout.write(self.style.WARNING(
print(yellow + 'create is set to false, wrapped will not be created !') "WARNING\ncreate is set to false, wrapped will not be created !"))
if verb >= 3 or change or not create: if verb >= 3 or change or not create:
a = str(input('\033[mContinue ? (y/n) ')).lower() a = str(input('\033[mContinue ? (y/n) ')).lower()
if a in ['n', 'no', 'non', '0']: if a in ['n', 'no', 'non', '0']:
if verb >= 0: if verb >= 0:
print(abort) self.stdout.write(self.style.ERROR("ABORT"))
return exit(1)
note = self.convert_to_note(change, create, bde=bde, user=user, club=club, verb=verb) note = self.convert_to_note(change, create, bde=bde, user=user, club=club, verb=verb)
if verb >= 1: if verb >= 1:
print("\033[32mUser and/or Club given has successfully convert in their note\033[m") self.stdout.write(self.style.SUCCESS(
"User and/or Club given has successfully convert in their note"))
global_data = self.global_data(bde, verb=verb) global_data = self.global_data(bde, verb=verb)
if verb >= 1: if verb >= 1:
print("\033[32mGlobal data has been successfully generated\033[m") self.stdout.write(self.style.SUCCESS(
"Global data has been successfully generated"))
unique_data = self.unique_data(bde, note, global_data=global_data, verb=verb) unique_data = self.unique_data(bde, note, global_data=global_data, verb=verb)
if verb >= 1: if verb >= 1:
print("\033[32mUnique data has been successfully generated\033[m") self.stdout.write(self.style.SUCCESS(
"Unique data has been successfully generated"))
self.make_wrapped(unique_data, note, bde, change, create, verb=verb) self.make_wrapped(unique_data, note, bde, change, create, verb=verb)
if verb >= 1: if verb >= 1:
print(green + "The wrapped has been generated !") self.stdout.write(self.style.SUCCESS(
"The wrapped has been generated !"))
if verb >= 0: if verb >= 0:
print(success) self.stdout.write(self.style.SUCCESS("SUCCESS"))
exit(0)
return def convert_to_note(self, change, create, bde=None, user=None, club=None, verb=1): # NOQA
def convert_to_note(self, change, create, bde=None, user=None, club=None, verb=1):
notes = [] notes = []
for b in bde: for b in bde:
note_for_bde = Note.objects.filter(pk__lte=-1) note_for_bde = Note.objects.filter(pk__lte=-1)
@ -253,17 +248,17 @@ class Command(BaseCommand):
note_for_bde = self.filter_note(b, note_for_bde, change, create, verb=verb) note_for_bde = self.filter_note(b, note_for_bde, change, create, verb=verb)
notes.append(note_for_bde) notes.append(note_for_bde)
if verb >= 2: if verb >= 2:
print("\033[m{nb} note selectionned for bde {bde}".format(nb=len(note_for_bde), bde=b.name)) self.stdout.write(f"{len(note_for_bde)} note selectionned for bde {b.name}")
return notes return notes
def global_data(self, bde, verb=1): def global_data(self, bde, verb=1): # NOQA
data = {} data = {}
for b in bde: for b in bde:
if b.name == 'Rave Part[list]': if b.name == 'Rave Part[list]':
if verb >= 2: if verb >= 2:
print("Begin to make global data") self.stdout.write("Begin to make global data")
if verb >= 3: if verb >= 3:
print('nb_transaction') self.stdout.write("nb_transaction")
# nb total de transactions # nb total de transactions
data['nb_transaction'] = Transaction.objects.filter( data['nb_transaction'] = Transaction.objects.filter(
created_at__gte=b.date_start, created_at__gte=b.date_start,
@ -271,7 +266,7 @@ class Command(BaseCommand):
valid=True).count() valid=True).count()
if verb >= 3: if verb >= 3:
print('nb_vieux_con') self.stdout.write("nb_vieux_con")
# nb total de vielleux con·ne·s derrière le bar # nb total de vielleux con·ne·s derrière le bar
button_id = [2884, 2585] button_id = [2884, 2585]
transactions = Transaction.objects.filter( transactions = Transaction.objects.filter(
@ -286,7 +281,7 @@ class Command(BaseCommand):
data['nb_vieux_con'] = q data['nb_vieux_con'] = q
if verb >= 3: if verb >= 3:
print('nb_soiree') self.stdout.write("nb_soiree")
# nb total de soirée # nb total de soirée
a_type_id = [1, 2, 4, 5, 7, 10] a_type_id = [1, 2, 4, 5, 7, 10]
data['nb_soiree'] = Activity.objects.filter( data['nb_soiree'] = Activity.objects.filter(
@ -296,7 +291,7 @@ class Command(BaseCommand):
activity_type__pk__in=a_type_id).count() activity_type__pk__in=a_type_id).count()
if verb >= 3: if verb >= 3:
print('pots, nb_entree_pot') self.stdout.write('pots, nb_entree_pot')
# nb d'entrée totale aux pots # nb d'entrée totale aux pots
pot_id = [1, 4, 10] pot_id = [1, 4, 10]
pots = Activity.objects.filter( pots = Activity.objects.filter(
@ -310,7 +305,7 @@ class Command(BaseCommand):
data['nb_entree_pot'] += Entry.objects.filter(activity=pot).count() data['nb_entree_pot'] += Entry.objects.filter(activity=pot).count()
if verb >= 3: if verb >= 3:
print('top3_buttons') self.stdout.write('top3_buttons')
# top 3 des boutons les plus cliqués # top 3 des boutons les plus cliqués
transactions = Transaction.objects.filter( transactions = Transaction.objects.filter(
created_at__gte=b.date_start, created_at__gte=b.date_start,
@ -329,7 +324,7 @@ class Command(BaseCommand):
data['top3_buttons'] = list(sorted(d.items(), key=lambda item: item[1], reverse=True))[:3] data['top3_buttons'] = list(sorted(d.items(), key=lambda item: item[1], reverse=True))[:3]
if verb >= 3: if verb >= 3:
print('class_conso_all') self.stdout.write('class_conso_all')
# le classement des plus gros consommateurs (BDE + club) # le classement des plus gros consommateurs (BDE + club)
transactions = Transaction.objects.filter( transactions = Transaction.objects.filter(
created_at__gte=b.date_start, created_at__gte=b.date_start,
@ -348,7 +343,7 @@ class Command(BaseCommand):
data['class_conso_all'] = dict(sorted(d.items(), key=lambda item: item[1], reverse=True)) data['class_conso_all'] = dict(sorted(d.items(), key=lambda item: item[1], reverse=True))
if verb >= 3: if verb >= 3:
print('class_conso_bde') self.stdout.write('class_conso_bde')
# le classement des plus gros consommateurs BDE # le classement des plus gros consommateurs BDE
transactions = Transaction.objects.filter( transactions = Transaction.objects.filter(
created_at__gte=b.date_start, created_at__gte=b.date_start,
@ -368,11 +363,10 @@ class Command(BaseCommand):
else: else:
# make your wrapped or reuse previous wrapped # make your wrapped or reuse previous wrapped
raise NotImplementedError("The BDE: {bde_name} has not personalized wrapped, make it !" raise NotImplementedError(f"The BDE: {b.name} has not personalized wrapped, make it !")
.format(bde_name=b.name))
return data return data
def unique_data(self, bde, note, global_data=None, verb=1): def unique_data(self, bde, note, global_data=None, verb=1): # NOQA
data = [] data = []
for i in range(len(bde)): for i in range(len(bde)):
data_bde = [] data_bde = []
@ -380,8 +374,7 @@ class Command(BaseCommand):
if verb >= 3: if verb >= 3:
total = len(note[i]) total = len(note[i])
current = 0 current = 0
print('Make {nb} data for wrapped sponsored by {bde}' self.stdout.write(f"Make {total} data for wrapped sponsored by {bde[i].name}")
.format(nb=total, bde=bde[i].name))
for n in note[i]: for n in note[i]:
d = {} d = {}
if 'user' in n.__dir__(): if 'user' in n.__dir__():
@ -542,12 +535,11 @@ class Command(BaseCommand):
data_bde.append(json.dumps(d)) data_bde.append(json.dumps(d))
if verb >= 3: if verb >= 3:
current += 1 current += 1
print('\033[2K' + '({c}/{t})'.format(c=current, t=total) + '\033[1A') self.stdout.write("\033[2K" + f"({current}/{total})" + "\033[1A")
else: else:
# make your wrapped or reuse previous wrapped # make your wrapped or reuse previous wrapped
raise NotImplementedError("The BDE: {bde_name} has not personalized wrapped, make it !" raise NotImplementedError(f"The BDE: {bde[i].name} has not personalized wrapped, make it !")
.format(bde_name=bde[i].name))
data.append(data_bde) data.append(data_bde)
return data return data
@ -557,7 +549,7 @@ class Command(BaseCommand):
total = 0 total = 0
for n in note: for n in note:
total += len(n) total += len(n)
print('\033[mMake {nb} wrapped'.format(nb=total)) self.stdout.write(f"Make {total} wrapped")
for i in range(len(bde)): for i in range(len(bde)):
for j in range(len(note[i])): for j in range(len(note[i])):
if create and not Wrapped.objects.filter(bde=bde[i], note=note[i][j]): if create and not Wrapped.objects.filter(bde=bde[i], note=note[i][j]):
@ -572,7 +564,7 @@ class Command(BaseCommand):
w.save() w.save()
if verb >= 3: if verb >= 3:
current += 1 current += 1
print('\033[2K' + '({c}/{t})'.format(c=current, t=total) + '\033[1A') self.stdout.write("\033[2K" + f"({current}/{total})" + "\033[1A")
return return
def filter_note(self, bde, note, change, create, verb=1): def filter_note(self, bde, note, change, create, verb=1):

View File

View File

@ -0,0 +1,91 @@
# Copyright (C) 2018-2025 by BDE ENS Paris-Saclay
# SPDX-License-Identifier: GPL-3.0-or-later
from datetime import timedelta
from api.tests import TestAPI
from django.contrib.auth.models import User
from django.test import TestCase
from django.urls import reverse
from django.utils import timezone
from ..api.views import WrappedViewSet, BdeViewSet
from ..models import Bde, Wrapped
class TestWrapped(TestCase):
"""
Test activities
"""
fixtures = ('initial',)
def setUp(self):
self.user = User.objects.create_superuser(
username="admintoto",
password="tototototo",
email="toto@example.com"
)
self.client.force_login(self.user)
sess = self.client.session
sess["permission_mask"] = 42
sess.save()
self.bde = Bde.objects.create(
name="The best BDE",
date_start=timezone.now() - timedelta(days=365),
date_end=timezone.now(),
)
self.wrapped = Wrapped.objects.create(
generated=True,
public=False,
bde=self.bde,
note=self.user.note,
data_json="{}",
)
def test_wrapped_list(self):
"""
Display the list of all wrapped
"""
response = self.client.get(reverse("wrapped:wrapped_list"))
self.assertEqual(response.status_code, 200)
def test_wrapped_detail(self):
"""
Display the detail of an wrapped
"""
response = self.client.get(reverse("wrapped:wrapped_detail", args=(self.wrapped.pk,)))
self.assertEqual(response.status_code, 200)
class TestWrappedAPI(TestAPI):
def setUp(self) -> None:
super().setUp()
self.bde = Bde.objects.create(
name="The best BDE",
date_start=timezone.now() - timedelta(days=365),
date_end=timezone.now(),
)
self.wrapped = Wrapped.objects.create(
generated=True,
public=False,
bde=self.bde,
note=self.user.note,
data_json="{}",
)
def test_bde_api(self):
"""
Load Bde API page and test all filters and permissions
"""
self.check_viewset(BdeViewSet, "/api/wrapped/bde/")
def test_wrapped_api(self):
"""
Load Wrapped API page and test all filters and permissions
"""
self.check_viewset(WrappedViewSet, "/api/wrapped/wrapped/")

34
shell-static.nix Executable file
View File

@ -0,0 +1,34 @@
# This is a workaround meant for use with the nix package manager. If you don't know what it is or don't use it, please ignore this file.
#
# The nk20 javascript static location are hardcoded for imperative system.
# This make ./manage.py collectstatic hard to use with nixos.
#
# A workaround is to enter a FHSUserEnv with the static placed under /share/javascript/<static>.
# This emulate a debian like system and enable collecting static normally with ./manage.py collectstatics.
# The regular shell.nix should be enough for other configurations.
#
# Warning, you are still supposed to use pip package with a venv !
{ pkgs ? import <nixpkgs> {} }:
(pkgs.buildFHSUserEnv {
name = "pipzone";
targetPkgs = pkgs: (with pkgs;
let
fhs-static = stdenv.mkDerivation {
name = "fhs-static";
buildCommand = ''
mkdir -p $out/share/javascript/bootstrap4
mkdir -p $out/share/javascript/jquery
ln -s ${python39Packages.xstatic-bootstrap}/lib/python3.9/site-packages/xstatic/pkg/bootstrap/data/* $out/share/javascript/bootstrap4
ln -s ${python39Packages.xstatic-jquery}/lib/python3.9/site-packages/xstatic/pkg/jquery/data/* $out/share/javascript/jquery
'';
};
in [
fhs-static
python39
gettext
python39Packages.pip
python39Packages.virtualenv
python39Packages.setuptools
]);
runScript = "bash";
}).env

23
shell.nix Executable file
View File

@ -0,0 +1,23 @@
# This is meant for use with the nix package manager. If you don't know what it is or don't use it, please ignore this file.
#
# This shell.nix contains all dependencies require to create a venv and pip install -r requirements.txt.
#
# Please check shell-static.nix for running ./manage.py collectstatics.
{ pkgs ? import <nixpkgs> {} }:
pkgs.mkShell {
buildInputs = with pkgs; [
python39
python39Packages.pip
python39Packages.setuptools
gettext
];
shellHook = ''
# Tells pip to put packages into $PIP_PREFIX instead of the usual locations.
# See https://pip.pypa.io/en/stable/user_guide/#environment-variables.
export PIP_PREFIX=$(pwd)/_build/pip_packages
export PYTHONPATH="$PIP_PREFIX/${pkgs.python39.sitePackages}:$PYTHONPATH"
export PATH="$PIP_PREFIX/bin:$PATH"
unset SOURCE_DATE_EPOCH
'';
}