add single import function

diff --git a/tests/wav_import.py b/tests/wav_import.py
index 7ee0a57f07b77b96a1c6d51539e3bccdc394ff0a..aee3c0a3644b0f8551467f3ef5fc0516cd23c435 100644 (file)
--- a/tests/wav_import.py
+++ b/tests/wav_import.py
@@ -17,6 +17,16 @@
 # >>> source_dir = '/home/momo/music/wav/CNRSMH_2006_002_GUYANE'
 # >>> t = TelemetaWavImport(id_string, source_dir)
 # >>> t.main()
+#
+# or :
+#
+# $ python manage.py shell
+# >>> from wav_import import TelemetaWavImport
+# >>> id_string = 'BM.2006.002.001--25__01-10'
+# >>> source_dir = '/home/momo/music/wav/CNRSMH_2006_002_GUYANE'
+# >>> source_file = 'CNRSMH_2006_002_001_10.wav'
+# >>> t = TelemetaWavImport(id_string, source_dir, source_file, source_file)
+# >>> t.main()
 
 
 import os
@@ -28,31 +38,37 @@ from django.conf import settings
 
 class TelemetaWavImport:
 
-    def __init__(self, id_string, source_dir):
+    def __init__(self, id_string, source_dir, source_file=None):
         self.item_media_root_dir = settings.MEDIA_ROOT + os.sep + 'items' + os.sep
         self.source_dir = source_dir
         self.source_files = os.listdir(self.source_dir)
+        self.source_file = source_file
         self.item_list = MediaItem.objects.filter(id__startswith=id_string)
         self.collection_id = self.item_list[0].collection_id
         self.year = datetime.datetime.now().strftime("%Y")
-        self.dest_dir = self.item_media_root_dir + self.year + os.sep + self.collection_id
+        self.dest_dir = self.item_media_root_dir + self.year + os.sep + self.collection_id + os.sep
   
     def copy_files(self):
         if not os.path.exists(self.dest_dir):
             os.makedirs(self.dest_dir)
         for file in self.source_files:
-            shutil.copy(self.source_dir + os.sep + file, self.dest_dir)
+            if not os.path.exists(self.dest_dir + file):
+                shutil.copy(self.source_dir + os.sep + file, self.dest_dir)
         
     def wav_import(self):
-        self.files = os.listdir(self.dest_dir)
-        self.files.sort()
-        #print self.files
+        if not self.source_file:
+            self.files = os.listdir(self.dest_dir)
+            self.files.sort()
+        else:
+            self.files = [self.source_file]
+        
+        print self.files
         i = 0
-        if len(self.files) > len(self.item_list):
+        if len(self.files) >= len(self.item_list):
             for item in self.item_list:
                 #item = MediaItem.objects.get(id=object.id)
                 print item.id + " : " + item.title + " : "
-                item.file._name = unicode(self.dest_dir + os.sep + self.files[i])
+                item.file._name = unicode(self.dest_dir + self.files[i])
                 item.save()
                 print item.file._name
                 i += 1