small cosmetic improvements

[swftools.git] / wx / gui / options / viewer.py

diff --git a/wx/gui/options/viewer.py b/wx/gui/options/viewer.py
index e6693e6..0432cac 100644 (file)
--- a/wx/gui/options/viewer.py
+++ b/wx/gui/options/viewer.py
@@ -25,53 +25,58 @@
 import os
 import wx
 import operator
-#import imp
+import imp
 from lib.wordwrap import wordwrap
 from gui.boldstatictext import BoldStaticText
 import viewers
 import gui.fields
+
+# this two are only to satisfy some plugins requirements
 import gui.plugin
+import shutil
 
-import viewers.raw
-import viewers.simple
-import viewers.rfx
-import viewers.flexpaper
-import viewers.technoargia
+#import viewers.raw
+#import viewers.simple
+#import viewers.rfx
+#import viewers.flexpaper
+#import viewers.technoargia
 
 class Viewers:
     def __init__(self):
-        self.viewers = [
-            'raw',
-            'simple',
-            'rfx',
-            'flexpaper',
-            'technoargia',
-        ]
-        self.modules = [
-            viewers.raw,
-            viewers.simple,
-            viewers.rfx,
-            viewers.flexpaper,
-            viewers.technoargia,
-        ]
+        #self.viewers = [
+        #    'raw',
+        #    'simple',
+        #    'rfx',
+        #    'flexpaper',
+        #    'technoargia',
+        #]
+        self.viewers = []
+        #self.modules = [
+        #    viewers.raw,
+        #    viewers.simple,
+        #    viewers.rfx,
+        #    viewers.flexpaper,
+        #    viewers.technoargia,
+        #]
+        self.modules = []
  
-        #self.list_viewers()
-        #self.import_viewers()
+        self.list_viewers()
+        self.import_viewers()
 
-    #def list_viewers(self):
-    #    for file in os.listdir('viewers'):
-    #        if (file.startswith('.') or file.startswith('_')
-    #            or file.endswith(".pyc") or not file.endswith('.py')):
-    #            continue
-    #        self.viewers.append(os.path.splitext(file)[0])
+    def list_viewers(self):
+        for file in os.listdir('viewers'):
+            if (file.startswith('.') or file.startswith('_')
+                or file.endswith(".pyc") or not file.endswith('.py')):
+                continue
+            self.viewers.append(os.path.splitext(file)[0])
 
-    #def import_viewers(self):
-    #    for file in self.viewers:
-    #        try:
-    #            _temp = imp.load_source("viewers.%s" % file, os.path.join(os.getcwdu(), "viewers/%s.py" % file))
-    #            self.modules.append(_temp)
-    #        except:
-    #            print "Could not load %s" % file
+    def import_viewers(self):
+        for file in self.viewers:
+            try:
+                _temp = imp.load_source("viewers.%s" % file, os.path.join(os.getcwdu(), "viewers/%s.py" % file))
+                self.modules.append(_temp)
+            except Exception, e:
+                print "Could not load %s (%s)" % (file, e)
 
 
 
@@ -99,6 +104,8 @@ class ViewerBook(wx.Listbook):
 
             win = self.makePanel(mod)
             self.AddPage(win, mod, imageId=idx)
+            if hasattr(mod, "default"):
+                self.SetSelection(idx)
 
 
     def makePanel(self, mod):