Application: gvSIG desktop » gvSIG scripting

# copyright 2003-2014 LOGILAB S.A. (Paris, FRANCE), all rights reserved.

# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr

#

# This file is part of astroid.

#

# astroid is free software: you can redistribute it and/or modify it under

# the terms of the GNU Lesser General Public License as published by the Free

# Software Foundation, either version 2.1 of the License, or (at your option) any

# later version.

#

# astroid is distributed in the hope that it will be useful, but WITHOUT

# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS

# FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more

# details.

#

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

# with astroid.  If not, see <http://www.gnu.org/licenses/>.

"""Astroid hooks for six.moves."""

import sys

from textwrap import dedent

from astroid import MANAGER, register_module_extender

from astroid.builder import AstroidBuilder

from astroid.exceptions import AstroidBuildingException, InferenceError

from astroid import nodes

SIX_ADD_METACLASS = 'six.add_metaclass'

def _indent(text, prefix, predicate=None):

    """Adds 'prefix' to the beginning of selected lines in 'text'.

    If 'predicate' is provided, 'prefix' will only be added to the lines

    where 'predicate(line)' is True. If 'predicate' is not provided,

    it will default to adding 'prefix' to all non-empty lines that do not

    consist solely of whitespace characters.

    """

    if predicate is None:

        predicate = lambda line: line.strip()

    def prefixed_lines():

        for line in text.splitlines(True):

            yield prefix + line if predicate(line) else line

    return ''.join(prefixed_lines())

if sys.version_info[0] == 2:

    _IMPORTS_2 = """

    import BaseHTTPServer

    import CGIHTTPServer

    import SimpleHTTPServer

    from StringIO import StringIO

    from cStringIO import StringIO as cStringIO

    from UserDict import UserDict

    from UserList import UserList

    from UserString import UserString

    import __builtin__ as builtins

    import thread as _thread

    import dummy_thread as _dummy_thread

    import ConfigParser as configparser

    import copy_reg as copyreg

    from itertools import (imap as map,

                           ifilter as filter,

                           ifilterfalse as filterfalse,

                           izip_longest as zip_longest,

                           izip as zip)

    import htmlentitydefs as html_entities

    import HTMLParser as html_parser

    import httplib as http_client

    import cookielib as http_cookiejar

    import Cookie as http_cookies

    import Queue as queue

    import repr as reprlib

    from pipes import quote as shlex_quote

    import SocketServer as socketserver

    import SimpleXMLRPCServer as xmlrpc_server

    import xmlrpclib as xmlrpc_client

    import _winreg as winreg

    import robotparser as urllib_robotparser

    import Tkinter as tkinter

    import tkFileDialog as tkinter_tkfiledialog

    input = raw_input

    intern = intern

    range = xrange

    xrange = xrange

    reduce = reduce

    reload_module = reload

    class UrllibParse(object):

        import urlparse as _urlparse

        import urllib as _urllib

        ParseResult = _urlparse.ParseResult

        SplitResult = _urlparse.SplitResult

        parse_qs = _urlparse.parse_qs

        parse_qsl = _urlparse.parse_qsl

        urldefrag = _urlparse.urldefrag

        urljoin = _urlparse.urljoin

        urlparse = _urlparse.urlparse

        urlsplit = _urlparse.urlsplit

        urlunparse = _urlparse.urlunparse

        urlunsplit = _urlparse.urlunsplit

        quote = _urllib.quote

        quote_plus = _urllib.quote_plus

        unquote = _urllib.unquote

        unquote_plus = _urllib.unquote_plus

        urlencode = _urllib.urlencode

        splitquery = _urllib.splitquery

        splittag = _urllib.splittag

        splituser = _urllib.splituser

        uses_fragment = _urlparse.uses_fragment       

        uses_netloc = _urlparse.uses_netloc

        uses_params = _urlparse.uses_params

        uses_query = _urlparse.uses_query

        uses_relative = _urlparse.uses_relative

    class UrllibError(object):

        import urllib2 as _urllib2

        import urllib as _urllib

        URLError = _urllib2.URLError

        HTTPError = _urllib2.HTTPError

        ContentTooShortError = _urllib.ContentTooShortError

    class DummyModule(object):

        pass

    class UrllibRequest(object):

        import urlparse as _urlparse

        import urllib2 as _urllib2

        import urllib as _urllib

        urlopen = _urllib2.urlopen

        install_opener = _urllib2.install_opener

        build_opener = _urllib2.build_opener

        pathname2url = _urllib.pathname2url

        url2pathname = _urllib.url2pathname

        getproxies = _urllib.getproxies

        Request = _urllib2.Request

        OpenerDirector = _urllib2.OpenerDirector

        HTTPDefaultErrorHandler = _urllib2.HTTPDefaultErrorHandler

        HTTPRedirectHandler = _urllib2.HTTPRedirectHandler

        HTTPCookieProcessor = _urllib2.HTTPCookieProcessor

        ProxyHandler = _urllib2.ProxyHandler

        BaseHandler = _urllib2.BaseHandler

        HTTPPasswordMgr = _urllib2.HTTPPasswordMgr

        HTTPPasswordMgrWithDefaultRealm = _urllib2.HTTPPasswordMgrWithDefaultRealm

        AbstractBasicAuthHandler = _urllib2.AbstractBasicAuthHandler

        HTTPBasicAuthHandler = _urllib2.HTTPBasicAuthHandler

        ProxyBasicAuthHandler = _urllib2.ProxyBasicAuthHandler

        AbstractDigestAuthHandler = _urllib2.AbstractDigestAuthHandler

        HTTPDigestAuthHandler = _urllib2.HTTPDigestAuthHandler

        ProxyDigestAuthHandler = _urllib2.ProxyDigestAuthHandler

        HTTPHandler = _urllib2.HTTPHandler

        HTTPSHandler = _urllib2.HTTPSHandler

        FileHandler = _urllib2.FileHandler

        FTPHandler = _urllib2.FTPHandler

        CacheFTPHandler = _urllib2.CacheFTPHandler

        UnknownHandler = _urllib2.UnknownHandler

        HTTPErrorProcessor = _urllib2.HTTPErrorProcessor

        urlretrieve = _urllib.urlretrieve

        urlcleanup = _urllib.urlcleanup

        proxy_bypass = _urllib.proxy_bypass

    urllib_parse = UrllibParse()

    urllib_error = UrllibError()

    urllib = DummyModule()

    urllib.request = UrllibRequest()

    urllib.parse = UrllibParse()

    urllib.error = UrllibError()

    """

else:

    _IMPORTS_3 = """

    import _io

    cStringIO = _io.StringIO

    filter = filter

    from itertools import filterfalse

    input = input

    from sys import intern

    map = map

    range = range

    from imp import reload as reload_module

    from functools import reduce

    from shlex import quote as shlex_quote

    from io import StringIO

    from collections import UserDict, UserList, UserString

    xrange = range

    zip = zip

    from itertools import zip_longest

    import builtins

    import configparser

    import copyreg

    import _dummy_thread

    import http.cookiejar as http_cookiejar

    import http.cookies as http_cookies

    import html.entities as html_entities

    import html.parser as html_parser

    import http.client as http_client

    import http.server

    BaseHTTPServer = CGIHTTPServer = SimpleHTTPServer = http.server

    import pickle as cPickle

    import queue

    import reprlib

    import socketserver

    import _thread

    import winreg

    import xmlrpc.server as xmlrpc_server

    import xmlrpc.client as xmlrpc_client

    import urllib.robotparser as urllib_robotparser

    import email.mime.multipart as email_mime_multipart

    import email.mime.nonmultipart as email_mime_nonmultipart

    import email.mime.text as email_mime_text

    import email.mime.base as email_mime_base

    import urllib.parse as urllib_parse

    import urllib.error as urllib_error

    import tkinter

    import tkinter.dialog as tkinter_dialog

    import tkinter.filedialog as tkinter_filedialog

    import tkinter.scrolledtext as tkinter_scrolledtext

    import tkinter.simpledialog as tkinder_simpledialog

    import tkinter.tix as tkinter_tix

    import tkinter.ttk as tkinter_ttk

    import tkinter.constants as tkinter_constants

    import tkinter.dnd as tkinter_dnd

    import tkinter.colorchooser as tkinter_colorchooser

    import tkinter.commondialog as tkinter_commondialog

    import tkinter.filedialog as tkinter_tkfiledialog

    import tkinter.font as tkinter_font

    import tkinter.messagebox as tkinter_messagebox

    import urllib.request

    import urllib.robotparser as urllib_robotparser

    import urllib.parse as urllib_parse

    import urllib.error as urllib_error

    """

if sys.version_info[0] == 2:

    _IMPORTS = dedent(_IMPORTS_2)

else:

    _IMPORTS = dedent(_IMPORTS_3)

def six_moves_transform():

    code = dedent('''

    class Moves(object):

    {}

    moves = Moves()

    ''').format(_indent(_IMPORTS, "    "))

    module = AstroidBuilder(MANAGER).string_build(code)

    module.name = 'six.moves'

    return module

def _six_fail_hook(modname):

    if modname != 'six.moves':

        raise AstroidBuildingException

    module = AstroidBuilder(MANAGER).string_build(_IMPORTS)

    module.name = 'six.moves'

    return module

def transform_six_add_metaclass(node):

    """Check if the given class node is decorated with *six.add_metaclass*

    If so, inject its argument as the metaclass of the underlying class.

    """

    if not node.decorators:

        return

    for decorator in node.decorators.nodes:

        if not isinstance(decorator, nodes.Call):

            continue

        try:

            func = next(decorator.func.infer())

        except InferenceError:

            continue

        if func.qname() == SIX_ADD_METACLASS and decorator.args:

            metaclass = decorator.args[0]

            node._metaclass = metaclass

            return node

register_module_extender(MANAGER, 'six', six_moves_transform)

register_module_extender(MANAGER, 'requests.packages.urllib3.packages.six',

                         six_moves_transform)

MANAGER.register_failed_import_hook(_six_fail_hook)

MANAGER.register_transform(nodes.ClassDef, transform_six_add_metaclass)