1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

     4

# This file is part of Beremiz, a Integrated Development Environment for

     5

# programming IEC 61131-3 automates supporting plcopen standard and CanFestival.

     6

#

     7

# Copyright (C) 2016 - 2017: Andrey Skvortsov <andrej.skvortzov@gmail.com>

     8

#

     9

# See COPYING file for copyrights details.

    10

#

    11

# This program is free software; you can redistribute it and/or

    12

# modify it under the terms of the GNU General Public License

    13

# as published by the Free Software Foundation; either version 2

    14

# of the License, or (at your option) any later version.

    15

#

    16

# This program is distributed in the hope that it will be useful,

    17

# but WITHOUT ANY WARRANTY; without even the implied warranty of

    18

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    19

# GNU General Public License for more details.

    20

#

    21

# You should have received a copy of the GNU General Public License

    22

# along with this program; if not, write to the Free Software

    23

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

    24

    25

    26

import os

    27

import sys

    28

import getopt

    29

    30

import wx

    31

from wx.lib.agw.advancedsplash import AdvancedSplash, AS_NOTIMEOUT, AS_CENTER_ON_SCREEN

    32

    33

import util.paths as paths

    34

    35

    36

class BeremizIDELauncher(object):

    37

    def __init__(self):

    38

        self.app = None

    39

        self.frame = None

    40

        self.updateinfo_url = None

    41

        self.extensions = []

    42

        self.app_dir = paths.AbsDir(__file__)

    43

        self.projectOpen = None

    44

        self.buildpath = None

    45

        self.splash = None

    46

        self.splashPath = self.Bpath("images", "splash.png")

    47

        self.modules = ["BeremizIDE"]

    48

        self.debug = os.path.exists("BEREMIZ_DEBUG")

    49

        self.handle_exception = None

    50

        self.logf = None

    51

    52

    def Bpath(self, *args):

    53

        return os.path.join(self.app_dir, *args)

    54

    55

    def Usage(self):

    56

        print("Usage:")

    57

        print("%s [Options] [Projectpath] [Buildpath]" % sys.argv[0])

    58

        print("")

    59

        print("Supported options:")

    60

        print("-h --help                    Print this help")

    61

        print("-u --updatecheck URL         Retrieve update information by checking URL")

    62

        print("-e --extend PathToExtension  Extend IDE functionality by loading at start additional extensions")

    63

        print("-l --log path                write content of console tab to given file")

    64

        print("")

    65

        print("")

    66

    67

    def SetCmdOptions(self):

    68

        self.shortCmdOpts = "hu:e:l:"

    69

        self.longCmdOpts = ["help", "updatecheck=", "extend=", "log="]

    70

    71

    def ProcessOption(self, o, a):

    72

        if o in ("-h", "--help"):

    73

            self.Usage()

    74

            sys.exit()

    75

        if o in ("-u", "--updatecheck"):

    76

            self.updateinfo_url = a

    77

        if o in ("-e", "--extend"):

    78

            self.extensions.append(a)

    79

        if o in ("-l", "--log"):

    80

            self.logf = open(a, 'a')

    81

    82

    def ProcessCommandLineArgs(self):

    83

        self.SetCmdOptions()

    84

        try:

    85

            opts, args = getopt.getopt(sys.argv[1:], self.shortCmdOpts, self.longCmdOpts)

    86

        except getopt.GetoptError:

    87

            # print help information and exit:

    88

            self.Usage()

    89

            sys.exit(2)

    90

    91

        for o, a in opts:

    92

            self.ProcessOption(o, a)

    93

    94

        if len(args) > 2:

    95

            self.Usage()

    96

            sys.exit()

    97

    98

        elif len(args) == 1:

    99

            self.projectOpen = args[0]

   100

            self.buildpath = None

   101

        elif len(args) == 2:

   102

            self.projectOpen = args[0]

   103

            self.buildpath = args[1]

   104

   105

    def CreateApplication(self):

   106

   107

        BeremizAppType = wx.App

   108

   109

        class BeremizApp(BeremizAppType):

   110

            def OnInit(_self):  # pylint: disable=no-self-argument

   111

                self.ShowSplashScreen()

   112

                return True

   113

   114

        self.app = BeremizApp(redirect=self.debug)

   115

        self.app.SetAppName('beremiz')

   116

   117

    def ShowSplashScreen(self):

   118

        class Splash(AdvancedSplash):

   119

            Painted = False

   120

   121

            def OnPaint(_self, event):  # pylint: disable=no-self-argument

   122

                AdvancedSplash.OnPaint(_self, event)

   123

                if not _self.Painted:  # trigger app start only once

   124

                    _self.Painted = True

   125

                    wx.CallAfter(self.AppStart)

   126

        bmp = wx.Image(self.splashPath).ConvertToBitmap()

   127

        self.splash = Splash(None,

   128

                             bitmap=bmp,

   129

                             agwStyle=AS_NOTIMEOUT | AS_CENTER_ON_SCREEN)

   130

   131

    def BackgroundInitialization(self):

   132

        self.InitI18n()

   133

        self.CheckUpdates()

   134

        self.LoadExtensions()

   135

        self.ImportModules()

   136

   137

    def InitI18n(self):

   138

        from util.misc import InstallLocalRessources

   139

        InstallLocalRessources(self.app_dir)

   140

   141

    def globals(self):

   142

        """

   143

        allows customizations to specify what globals

   144

        are passed to extensions

   145

        """

   146

        return globals()

   147

   148

    def LoadExtensions(self):

   149

        for extfilename in self.extensions:

   150

            from util.TranslationCatalogs import AddCatalog

   151

            from util.BitmapLibrary import AddBitmapFolder

   152

            extension_folder = os.path.split(os.path.realpath(extfilename))[0]

   153

            sys.path.append(extension_folder)

   154

            AddCatalog(os.path.join(extension_folder, "locale"))

   155

            AddBitmapFolder(os.path.join(extension_folder, "images"))

   156

            exec(compile(open(extfilename, "rb").read(), extfilename, 'exec'), self.globals())

   157

   158

    def CheckUpdates(self):

   159

        if self.updateinfo_url is not None:

   160

            self.updateinfo = _("Fetching %s") % self.updateinfo_url

   161

   162

            def updateinfoproc():

   163

                try:

   164

                    import urllib.request, urllib.error, urllib.parse

   165

                    self.updateinfo = urllib.request.urlopen(self.updateinfo_url, None).read()

   166

                except Exception:

   167

                    self.updateinfo = _("update info unavailable.")

   168

   169

            from threading import Thread

   170

            self.splash.SetText(text=self.updateinfo)

   171

            updateinfoThread = Thread(target=updateinfoproc)

   172

            updateinfoThread.start()

   173

            updateinfoThread.join(2)

   174

            self.splash.SetText(text=self.updateinfo)

   175

   176

    def ImportModules(self):

   177

        for modname in self.modules:

   178

            mod = __import__(modname)

   179

            setattr(self, modname, mod)

   180

   181

    def InstallExceptionHandler(self):

   182

        import version

   183

        import util.ExceptionHandler

   184

        self.handle_exception = util.ExceptionHandler.AddExceptHook(version.app_version, logf=self.logf)

   185

   186

    def CreateUI(self):

   187

        self.frame = self.BeremizIDE.Beremiz(None, self.projectOpen, self.buildpath, logf=self.logf)

   188

   189

    def CloseSplash(self):

   190

        if self.splash:

   191

            self.splash.Close()

   192

   193

    def ShowUI(self):

   194

        self.frame.Show()

   195

   196

    def PreStart(self):

   197

        self.ProcessCommandLineArgs()

   198

        self.CreateApplication()

   199

   200

    def AppStart(self):

   201

        try:

   202

            self.BackgroundInitialization()

   203

            self.CreateUI()

   204

            self.CloseSplash()

   205

            self.ShowUI()

   206

        except (KeyboardInterrupt, SystemExit):

   207

            raise

   208

        except Exception:

   209

            if self.handle_exception is not None:

   210

                self.handle_exception(*sys.exc_info(), exit=True)

   211

            else:

   212

                raise

   213

   214

    def MainLoop(self):

   215

        self.app.MainLoop()

   216

   217

    def Start(self):

   218

        self.PreStart()

   219

        self.InstallExceptionHandler()

   220

        self.MainLoop()

   221

   222

   223

if __name__ == '__main__':

   224

    beremiz = BeremizIDELauncher()

   225

    beremiz.Start()