commands: ajout d’une commande pour lister les propriétés des cours

Pour pouvoir analyser plus facilement les attributs d’un cours, un
nouvel outil est rajouté pour lister tous les attributs d’un ou
plusieurs cours.  Il fait appel au parseur et permet de sélectionner
une source, et de limiter le nombre de cours affichés.

Signed-off-by: Alban Gruin <alban at pa1ch dot fr>

1 files changed, 45 insertions, 0 deletions

diff --git a/management/commands/printvalues.py b/management/commands/printvalues.py
new file mode 100644
index 0000000..91dd18b
--- /dev/null
+++ b/management/commands/printvalues.py
@@ -0,0 +1,45 @@
+#    Copyright (C) 2019  Alban Gruin
+#
+#    celcatsanitizer is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU Affero General Public License as published
+#    by the Free Software Foundation, either version 3 of the License, or
+#    (at your option) any later version.
+#
+#    celcatsanitizer is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU Affero General Public License for more details.
+#
+#    You should have received a copy of the GNU Affero General Public License
+#    along with celcatsanitizer.  If not, see <http://www.gnu.org/licenses/>.
+
+from django.core.management.base import BaseCommand
+
+from ...models import Source
+from .__parsercommand import ParserCommand
+
+import json
+
+
+class Command(BaseCommand, ParserCommand):
+    help = "List values from courses from a source"
+
+    def add_arguments(self, parser):
+        parser.add_argument("--source", type=int, nargs=1, required=True)
+        parser.add_argument("--limit", type=int, nargs=1)
+
+    def handle(self, *args, **options):
+        source = Source.objects.get(pk=options["source"][0])
+        parser = self.get_parser()(source)
+        events = [event for month in parser.get_source() for event in month]
+
+        i = 0
+        limit = len(events)
+        if options["limit"] is not None:
+            limit = min(options["limit"][0], limit)
+
+        while i < limit:
+            self.stdout.write(json.dumps(events[i], indent=4, sort_keys=True))
+            i += 1
+
+        self.stdout.write(self.style.SUCCESS("Done."))