From 6b4cf8d0ba6ee46a6bb9c1b91996e6d72ef14a7e Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Wed, 6 Feb 2013 18:24:49 +0100 Subject: Work on imports --- archaeological_operations/management/__init__.py | 0 .../management/commands/import_operations.py | 35 ++++++++++++++++++++++ 2 files changed, 35 insertions(+) create mode 100644 archaeological_operations/management/__init__.py create mode 100755 archaeological_operations/management/commands/import_operations.py (limited to 'archaeological_operations/management') diff --git a/archaeological_operations/management/__init__.py b/archaeological_operations/management/__init__.py new file mode 100644 index 000000000..e69de29bb diff --git a/archaeological_operations/management/commands/import_operations.py b/archaeological_operations/management/commands/import_operations.py new file mode 100755 index 000000000..b43954e16 --- /dev/null +++ b/archaeological_operations/management/commands/import_operations.py @@ -0,0 +1,35 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- +# Copyright (C) 2012 Étienne Loks + +# This program 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. + +# This program 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 this program. If not, see . + +# See the file COPYING for details. + +from django.core.management.base import BaseCommand, CommandError +from archaeological_operations.utils import import_from_csv + +class Command(BaseCommand): + args = ' []' + help = "Import archaelogical operation from a CSV file." + + def handle(self, *args, **options): + if not args or not args[0]: + raise CommandError("No CSV file provided." % args[0]) + nb_ops, errors = import_from_csv(args[0], + update=len(args) > 1 and args[1], + stdout=self.stdout) + self.stdout.write('\n* %d operation treated\n' % nb_ops) + if errors: + self.stderr.write('\n'.join(errors)) -- cgit v1.2.3