From: Guillaume Pellerin <yomguy@parisson.com>
Date: Wed, 1 Dec 2010 10:54:35 +0000 (+0100)
Subject: only import files
X-Git-Tag: 1.1~486^2~5
X-Git-Url: https://git.parisson.com/?a=commitdiff_plain;h=8534e5989ea5005e3e215def8286a2233a716ad4;p=telemeta.git

only import files
---

diff --git a/tools/scripts/telemeta-media-import.py b/tools/scripts/telemeta-media-import.py
index f9a5d1c5..bd998e74 100755
--- a/tools/scripts/telemeta-media-import.py
+++ b/tools/scripts/telemeta-media-import.py
@@ -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