github.com_openSUSE_osc/osc/babysitter.py

# Copyright (C) 2008 Novell Inc.  All rights reserved.
# This program is free software; it may be used, copied, modified
# and distributed under the terms of the GNU General Public Licence,
# either version 2, or (at your option) any later version.

import errno
import os.path
import pdb
import sys
import signal
import traceback

from osc import oscerr
from .oscsslexcp import NoSecureSSLError
from osc.util.cpio import CpioError
from osc.util.packagequery import PackageError

try:
    from M2Crypto.SSL.Checker import SSLVerificationError
    from M2Crypto.SSL import SSLError as SSLError
except:
    SSLError = None
    SSLVerificationError = None

try:
    # import as RPMError because the class "error" is too generic
    from rpm import error as RPMError
except:
    # if rpm-python isn't installed (we might be on a debian system):
    RPMError = None

try:
    from http.client import HTTPException, BadStatusLine
except ImportError:
    #python 2.x
    from httplib import HTTPException, BadStatusLine
from urllib2 import URLError, HTTPError

# the good things are stolen from Matt Mackall's mercurial


def catchterm(*args):
    raise oscerr.SignalInterrupt

for name in 'SIGBREAK', 'SIGHUP', 'SIGTERM':
    num = getattr(signal, name, None)
    if num:
        signal.signal(num, catchterm)


def run(prg):
    try:
        try:
            if '--debugger' in sys.argv:
                pdb.set_trace()
            # here we actually run the program:
            return prg.main()
        except:
            # look for an option in the prg.options object and in the config
            # dict print stack trace, if desired
            if getattr(prg.options, 'traceback', None) or getattr(prg.conf, 'config', {}).get('traceback', None) or \
               getattr(prg.options, 'post_mortem', None) or getattr(prg.conf, 'config', {}).get('post_mortem', None):
                traceback.print_exc(file=sys.stderr)
                # we could use http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/52215
            # enter the debugger, if desired
            if getattr(prg.options, 'post_mortem', None) or getattr(prg.conf, 'config', {}).get('post_mortem', None):
                if sys.stdout.isatty() and not hasattr(sys, 'ps1'):
                    pdb.post_mortem(sys.exc_info()[2])
                else:
                    print >>sys.stderr, 'sys.stdout is not a tty. Not jumping into pdb.'
            raise
    except oscerr.SignalInterrupt:
        print >>sys.stderr, 'killed!'
        return 1
    except KeyboardInterrupt:
        print >>sys.stderr, 'interrupted!'
        return 1
    except oscerr.UserAbort:
        print >>sys.stderr, 'aborted.'
        return 1
    except oscerr.APIError as e:
        print >>sys.stderr, 'BuildService API error:', e.msg
        return 1
    except oscerr.LinkExpandError as e:
        print >>sys.stderr, 'Link "%s/%s" cannot be expanded:\n' % (e.prj, e.pac), e.msg
        print >>sys.stderr, 'Use "osc repairlink" to fix merge conflicts.\n'
        return 1
    except oscerr.WorkingCopyWrongVersion as e:
        print >>sys.stderr, e
        return 1
    except oscerr.NoWorkingCopy as e:
        print >>sys.stderr, e
        if os.path.isdir('.git'):
            print >>sys.stderr, "Current directory looks like git."
        if os.path.isdir('.hg'):
            print >>sys.stderr, "Current directory looks like mercurial."
        if os.path.isdir('.svn'):
            print >>sys.stderr, "Current directory looks like svn."
        if os.path.isdir('CVS'):
            print >>sys.stderr, "Current directory looks like cvs."
        return 1
    except HTTPError as e:
        print >>sys.stderr, 'Server returned an error:', e
        if hasattr(e, 'osc_msg'):
            print >>sys.stderr, e.osc_msg

        try:
            body = e.read()
        except AttributeError:
            body = ''

        if getattr(prg.options, 'debug', None) or \
           getattr(prg.conf, 'config', {}).get('debug', None):
            print >>sys.stderr, e.hdrs
            print >>sys.stderr, body

        if e.code in [400, 403, 404, 500]:
            if '<summary>' in body:
                msg = body.split('<summary>')[1]
                msg = msg.split('</summary>')[0]
                print >>sys.stderr, msg
        return 1
    except BadStatusLine as e:
        print >>sys.stderr, 'Server returned an invalid response:', e
        print >>sys.stderr, e.line
        return 1
    except HTTPException as e:
        print >>sys.stderr, e
        return 1
    except URLError as e:
        print >>sys.stderr, 'Failed to reach a server:\n', e.reason
        return 1
    except IOError as e:
        # ignore broken pipe
        if e.errno != errno.EPIPE:
            raise
        return 1
    except OSError as e:
        if e.errno != errno.ENOENT:
            raise
        print >>sys.stderr, e
        return 1
    except (oscerr.ConfigError, oscerr.NoConfigfile) as e:
        print >>sys.stderr, e.msg
        return 1
    except oscerr.OscIOError as e:
        print >>sys.stderr, e.msg
        if getattr(prg.options, 'debug', None) or \
           getattr(prg.conf, 'config', {}).get('debug', None):
            print >>sys.stderr, e.e
        return 1
    except (oscerr.WrongOptions, oscerr.WrongArgs) as e:
        print >>sys.stderr, e
        return 2
    except oscerr.ExtRuntimeError as e:
        print >>sys.stderr, e.file + ':', e.msg
        return 1
    except oscerr.WorkingCopyOutdated as e:
        print >>sys.stderr, e
        return 1
    except (oscerr.PackageExists, oscerr.PackageMissing, oscerr.WorkingCopyInconsistent) as e:
        print >>sys.stderr, e.msg
        return 1
    except oscerr.PackageInternalError as e:
        print >>sys.stderr, 'a package internal error occured\n' \
            'please file a bug and attach your current package working copy ' \
            'and the following traceback to it:'
        print >>sys.stderr, e.msg
        traceback.print_exc(file=sys.stderr)
        return 1
    except oscerr.PackageError as e:
        print >>sys.stderr, e.msg
        return 1
    except PackageError as e:
        print >>sys.stderr, '%s:' % e.fname, e.msg
        return 1
    except RPMError as e:
        print >>sys.stderr, e
        return 1
    except SSLError as e:
        print >>sys.stderr, "SSL Error:", e
        return 1
    except SSLVerificationError as e:
        print >>sys.stderr, "Certificate Verification Error:", e
        return 1
    except NoSecureSSLError as e:
        print >>sys.stderr, e
        return 1
    except CpioError as e:
        print >>sys.stderr, e
        return 1
    except oscerr.OscBaseError as e:
        print >>sys.stderr, '*** Error:', e
        return 1

# vim: sw=4 et