From: Guillaume Pellerin <guillaume.pellerin@ircam.fr>
Date: Tue, 22 Nov 2016 16:26:07 +0000 (+0100)
Subject: Add domain arg for import
X-Git-Url: https://git.parisson.com/?a=commitdiff_plain;h=43ce543ead955e034aed430ff80d446b0111d8dd;p=telemeta.git

Add domain arg for import
---

diff --git a/telemeta/management/commands/telemeta-import-collections-from-crem.py b/telemeta/management/commands/telemeta-import-collections-from-crem.py
index 58745b84..1383ea00 100644
--- a/telemeta/management/commands/telemeta-import-collections-from-crem.py
+++ b/telemeta/management/commands/telemeta-import-collections-from-crem.py
@@ -77,6 +77,9 @@ class Command(BaseCommand):
           make_option('-p', '--pattern',
             dest='pattern',
             help='define the pattern'),
+          make_option('-m', '--domain',
+            dest='domain',
+            help='define site domain'),
     )
 
     def write_file(self, item, media):
@@ -112,8 +115,16 @@ class Command(BaseCommand):
         self.source_dir = os.path.abspath(kwargs.get('source_dir'))
         self.dry_run =  kwargs.get('dry-run')
         self.force = kwargs.get('force')
+        self.domain = kwargs.get('domain')
+
+        site = Site.objects.all()[0]
+        if self.domain:
+            site.domain = self.domain
+            site.name = self.domain
+            site.save()
+        else:
+            self.domain = site.domain
 
-        self.domain = Site.objects.all()[0].domain
         self.user = User.objects.filter(username='admin')[0]
         self.collections = os.listdir(self.source_dir)