]> git.parisson.com Git - telemeta.git/commitdiff
only import files
authorGuillaume Pellerin <yomguy@parisson.com>
Wed, 1 Dec 2010 10:54:35 +0000 (11:54 +0100)
committerGuillaume Pellerin <yomguy@parisson.com>
Wed, 1 Dec 2010 10:54:35 +0000 (11:54 +0100)
tools/scripts/telemeta-media-import.py

index f9a5d1c5c19ff20f6c8dde599f41e7497f940c86..bd998e741c39fb495a55defd29012b4bc535ddab 100755 (executable)
@@ -41,9 +41,17 @@ class TelemetaMediaImport:
     def __init__(self, media_dir, log_file):
         self.logger = Logger(log_file)
         self.media_dir = media_dir + os.sep + 'items'
-        self.medias = os.listdir(self.media_dir)
+        self.medias = self.get_medias()
         self.buffer_size = 0x1000
         self.media_item_dir = 'items/'
+    
+    def get_medias(self):
+        medias = []
+        list = os.listdir(self.media_dir)
+        for element in list:
+            if os.path.isfile(self.media_dir+os.sep+element):
+                medias.append(element)
+        return medias
         
     def set_collection(self, collection_name):
         import telemeta.models