# HG changeset patch
# User Andrey Skvortsov <andrej.skvortzov@gmail.com>
# Date 1505496624 -10800
# Node ID fc387c4fc1d46ef89b3c89bace022014f36b164c
# Parent  c18adf359d218e02fbb783daa5b599f3af25e780
rewrite PLCOpenEditor. Add PLCOpenEditorApp class for easier customizations.

diff -r c18adf359d21 -r fc387c4fc1d4 PLCOpenEditor.py
--- a/PLCOpenEditor.py	Fri Sep 15 20:28:54 2017 +0300
+++ b/PLCOpenEditor.py	Fri Sep 15 20:30:24 2017 +0300
@@ -32,69 +32,23 @@
 import util.paths as paths
 import util.ExceptionHandler
 
-
-beremiz_dir = paths.AbsDir(__file__)
-
-
-if __name__ == '__main__':
-    # Usage message displayed when help request or when error detected in
-    # command line
-    def usage():
-        print "\nUsage of PLCOpenEditor.py :"
-        print "\n   %s [Filepath]\n" % sys.argv[0]
-
-    # Parse options given to PLCOpenEditor in command line
-    try:
-        opts, args = getopt.getopt(sys.argv[1:], "h", ["help"])
-    except getopt.GetoptError:
-        # print help information and exit:
-        usage()
-        sys.exit(2)
-
-    # Extract if help has been requested
-    for o, a in opts:
-        if o in ("-h", "--help"):
-            usage()
-            sys.exit()
-
-    # Extract the optional filename to open
-    fileOpen = None
-    if len(args) > 1:
-        usage()
-        sys.exit()
-    elif len(args) == 1:
-        fileOpen = args[0]
-
-    # Create wxApp (Need to create App before internationalization because of
-    # Windows)
-    if wx.VERSION >= (3, 0, 0):
-        app = wx.App()
-    else:
-        app = wx.PySimpleApp()
-
-    from util.misc import InstallLocalRessources
-    InstallLocalRessources(beremiz_dir)
-
-    # these imports require wx.GetApp to return
-    # a valid application instance
-
-    from IDEFrame import IDEFrame, AppendMenu
-    from IDEFrame import \
-        TITLE, \
-        EDITORTOOLBAR, \
-        FILEMENU, \
-        EDITMENU, \
-        DISPLAYMENU, \
-        PROJECTTREE, \
-        POUINSTANCEVARIABLESPANEL, \
-        LIBRARYTREE, \
-        PAGETITLES
-
-    from IDEFrame import EncodeFileSystemPath, DecodeFileSystemPath
-    from editors.Viewer import Viewer
-    from PLCControler import PLCControler
-    from dialogs import ProjectDialog
-    from dialogs.AboutDialog import ShowAboutDialog
+from IDEFrame import IDEFrame, AppendMenu
+from IDEFrame import \
+    TITLE, \
+    EDITORTOOLBAR, \
+    FILEMENU, \
+    EDITMENU, \
+    DISPLAYMENU, \
+    PROJECTTREE, \
+    POUINSTANCEVARIABLESPANEL, \
+    LIBRARYTREE, \
+    PAGETITLES
+from IDEFrame import EncodeFileSystemPath, DecodeFileSystemPath
+from editors.Viewer import Viewer
+from PLCControler import PLCControler
+from dialogs import ProjectDialog
+from dialogs.AboutDialog import ShowAboutDialog
+from util.misc import InstallLocalRessources
 
 
 # -------------------------------------------------------------------------------
@@ -107,6 +61,9 @@
 ] = [wx.NewId() for _init_coll_FileMenu_Items in range(1)]
 
 
+beremiz_dir = paths.AbsDir(__file__)
+
+
 class PLCOpenEditor(IDEFrame):
 
     # Compatibility function for wx versions < 2.6
@@ -414,14 +371,50 @@
         dialog.Destroy()
 
 
+class PLCOpenEditorApp(wx.App):
+    # def SetOpenFile(
+
+    def PrintUsage(self):
+        print "\nUsage of PLCOpenEditor.py :"
+        print "\n   %s [Filepath]\n" % sys.argv[0]
+
+    def ParseCommandLine(self):
+        # Parse options given to PLCOpenEditor in command line
+        try:
+            opts, args = getopt.getopt(sys.argv[1:], "h", ["help"])
+        except getopt.GetoptError:
+            # print help information and exit:
+            self.PrintUsage()
+            sys.exit(2)
+
+        # Extract if help has been requested
+        for o, a in opts:
+            if o in ("-h", "--help"):
+                self.PrintUsage()
+                sys.exit()
+
+        # Extract the optional filename to open
+        self.fileOpen = None
+        if len(args) > 1:
+            self.PrintUsage()
+            sys.exit()
+        elif len(args) == 1:
+            self.fileOpen = args[0]
+
+    def OnInit(self):
+        self.ParseCommandLine()
+        InstallLocalRessources(beremiz_dir)
+        if wx.VERSION < (3, 0, 0):
+            wx.InitAllImageHandlers()
+        util.ExceptionHandler.AddExceptHook(version.app_version)
+        self.frame = PLCOpenEditor(None, fileOpen=self.fileOpen)
+        return True
+
+    def Show(self):
+        self.frame.Show()
+
+
 if __name__ == '__main__':
-    if wx.VERSION < (3, 0, 0):
-        wx.InitAllImageHandlers()
-
-    # Install a exception handle for bug reports
-    util.ExceptionHandler.AddExceptHook(version.app_version)
-
-    frame = PLCOpenEditor(None, fileOpen=fileOpen)
-
-    frame.Show()
+    app = PLCOpenEditorApp()
+    app.Show()
     app.MainLoop()