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) 2007:      Edouard TISSERANT and Laurent BESSARD

     8

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

     9

#

    10

# See COPYING file for copyrights details.

    11

#

    12

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

    13

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

    14

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

    15

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

    16

#

    17

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

    18

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

    19

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    20

# GNU General Public License for more details.

    21

#

    22

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

    23

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

    24

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

    25

    26

    27

from __future__ import absolute_import

    28

import os

    29

import sys

    30

import time

    31

import tempfile

    32

import platform

    33

import traceback

    34

import threading

    35

import wx

    36

    37

Max_Traceback_List_Size = 20

    38

    39

    40

def Display_Exception_Dialog(e_type, e_value, e_tb, bug_report_path, exit):

    41

    trcbck_lst = []

    42

    for i, line in enumerate(traceback.extract_tb(e_tb)):

    43

        trcbck = " " + str(i+1) + ". "

    44

        if line[0].find(os.getcwd()) == -1:

    45

            trcbck += "file : " + str(line[0]) + ",   "

    46

        else:

    47

            trcbck += "file : " + str(line[0][len(os.getcwd()):]) + ",   "

    48

        trcbck += "line : " + str(line[1]) + ",   " + "function : " + str(line[2])

    49

        trcbck_lst.append(trcbck)

    50

    51

    # Allow clicking....

    52

    cap = wx.Window.GetCapture()

    53

    if cap:

    54

        cap.ReleaseMouse()

    55

    56

    dlg = wx.SingleChoiceDialog(

    57

        None,

    58

        _("""

    59

An unhandled exception (bug) occured. Bug report saved at :

    60

(%s)

    61

    62

Please be kind enough to send this file to:

    63

beremiz-devel@lists.sourceforge.net

    64

    65

You should now restart program.

    66

    67

Traceback:

    68

""") % bug_report_path +

    69

        repr(e_type) + " : " + repr(e_value),

    70

        _("Error"),

    71

        trcbck_lst)

    72

    try:

    73

        res = (dlg.ShowModal() == wx.ID_OK)

    74

    finally:

    75

        dlg.Destroy()

    76

    77

    if exit:

    78

        sys.exit()  # wx.Exit()

    79

    80

    return res

    81

    82

    83

def get_last_traceback(tb):

    84

    while True:

    85

        if not hasattr(tb, "tb_next"):

    86

            break

    87

        if tb.tb_next:

    88

            tb = tb.tb_next

    89

        else:

    90

            break

    91

    92

    return tb

    93

    94

    95

def format_namespace(d, indent='    '):

    96

    return '\n'.join(['%s%s: %s' % (indent, k, repr(v)[:10000]) for k, v in d.iteritems()])

    97

    98

    99

ignored_exceptions = []  # a problem with a line in a module is only reported once per session

   100

   101

   102

def AddExceptHook(app_version='[No version]', logf = None):

   103

   104

    def save_bug_report(e_type, e_value, e_traceback, bug_report_path, date):

   105

        info = {

   106

            'app-title': wx.GetApp().GetAppName(),

   107

            'app-version': app_version,

   108

            'wx-version': wx.VERSION_STRING,

   109

            'wx-platform': wx.Platform,

   110

            'python-version': platform.python_version(),

   111

            'platform': platform.platform(),

   112

            'e-type': e_type,

   113

            'e-value': e_value,

   114

            'date': date,

   115

            'cwd': os.getcwd(),

   116

        }

   117

        if e_traceback:

   118

            info['traceback'] = ''.join(traceback.format_tb(e_traceback)) + '%s: %s' % (e_type, e_value)

   119

            last_tb = get_last_traceback(e_traceback)

   120

            # save locals at the level of the stack trace where the exception actually occurred

   121

            exception_locals = last_tb.tb_frame.f_locals if last_tb else {};

   122

            info['locals'] = format_namespace(exception_locals)

   123

            if 'self' in exception_locals:

   124

                try:

   125

                    info['self'] = format_namespace(exception_locals['self'].__dict__)

   126

                except Exception:

   127

                    pass

   128

        path = os.path.dirname(bug_report_path)

   129

        if not os.path.exists(path):

   130

            os.mkdir(path)

   131

        output = open(bug_report_path, 'w')

   132

        lst = info.keys()

   133

        lst.sort()

   134

        for a in lst:

   135

            line = a + ":\n" + str(info[a]) + "\n\n"

   136

            output.write(line)

   137

            if logf is not None:

   138

                logf.write(line)

   139

        output.close()

   140

   141

    def handle_exception(e_type, e_value, e_traceback, exit=False):

   142

        traceback.print_exception(e_type, e_value, e_traceback)  # this is very helpful when there's an exception in the rest of this func

   143

        last_tb = get_last_traceback(e_traceback)

   144

        ex = (last_tb.tb_frame.f_code.co_filename if last_tb else "unknown",

   145

              last_tb.tb_frame.f_lineno if last_tb else None)

   146

        if ex not in ignored_exceptions:

   147

            ignored_exceptions.append(ex)

   148

            date = time.ctime()

   149

            path = tempfile.gettempdir()+os.sep+wx.GetApp().GetAppName()

   150

            bug_report_path = path + os.sep + "bug_report_" + time.strftime("%Y_%m_%d__%H-%M-%S") + ".txt"

   151

            save_bug_report(e_type, e_value, e_traceback, bug_report_path, date)

   152

            wx.CallAfter(Display_Exception_Dialog, e_type, e_value, e_traceback, bug_report_path, exit)

   153

    # sys.excepthook = lambda *args: wx.CallAfter(handle_exception, *args)

   154

    sys.excepthook = handle_exception

   155

   156

    init_old = threading.Thread.__init__

   157

   158

    def init(self, *args, **kwargs):

   159

        init_old(self, *args, **kwargs)

   160

        run_old = self.run

   161

   162

        def run_with_except_hook(*args, **kw):

   163

            try:

   164

                run_old(*args, **kw)

   165

            except (KeyboardInterrupt, SystemExit):

   166

                raise

   167

            except Exception:

   168

                sys.excepthook(*sys.exc_info())

   169

        self.run = run_with_except_hook

   170

    threading.Thread.__init__ = init

   171

   172

    return handle_exception