1

#!/usr/bin/env python

     2

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

     3

     4

import sys

     5

import new

     6

from types import ModuleType

     7

     8

# TODO use gettext instead

     9

def get_translation(txt):

    10

    return txt

    11

    12

    13

class FakeObject:

    14

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

    15

        self.__classname__ = kwargs["__classname__"]

    16

    17

    def __getattr__(self,name):

    18

        if name.startswith('__'):

    19

            raise AttributeError, name

    20

        return FakeObject(__classname__=self.__classname__+"."+name)

    21

    22

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

    23

        return FakeObject(__classname__=self.__classname__+"()")

    24

    25

    def __getitem__(self, key):

    26

        raise IndexError, key

    27

    28

    def __str__(self):

    29

        return self.__classname__

    30

    31

    def __or__(self, other):

    32

        return FakeObject(__classname__=self.__classname__+"|"+other.__classname__)

    33

    34

    35

class FakeClass:

    36

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

    37

        print("DUMMY Class __init__ !",self.__name__,args,kwargs)

    38

    39

    40

class FakeModule(ModuleType):

    41

    def __init__(self, name, classes):

    42

        self.__modname__ = name

    43

        self.__objects__ = dict(map(lambda desc: 

    44

            (desc, new.classobj(desc, (FakeClass,), {}))

    45

            if type(desc)==str else desc, classes))

    46

        ModuleType(name)

    47

    48

    def __getattr__(self,name):

    49

        if name.startswith('__'):

    50

            raise AttributeError, name

    51

    52

        if self.__objects__.has_key(name):

    53

            return self.__objects__[name]

    54

    55

        obj = FakeObject(__classname__=self.__modname__+"."+name)

    56

        self.__objects__[name] = obj

    57

        return obj

    58

    59

    60

# Keep track of already faked modules to catch those

    61

# that are already present in sys.modules from start

    62

# (i.e. mpl_toolkits for exemple)

    63

already_patched = {}

    64

    65

for name, classes in [

    66

    # list given for each module name contains name string for a FakeClass,

    67

    # otherwise a tuple (name, object) for arbitrary object/function/class

    68

    ('wx',[

    69

        'Panel', 'PyCommandEvent', 'Dialog', 'PopupWindow', 'TextEntryDialog',

    70

        'Notebook', 'ListCtrl', 'TextDropTarget', 'PyControl', 'TextCtrl', 

    72

        ('GetTranslation', get_translation)]),

    73

    ('wx.lib.agw.advancedsplash',[]),

    75

    ('wx.lib.buttons',['GenBitmapTextButton']),

    76

    ('wx.adv',['EditableListBox']),

    77

    ('wx.grid',[

    78

        'Grid', 'PyGridTableBase', 'GridCellEditor', 'GridCellTextEditor',

    79

        'GridCellChoiceEditor']),

    80

    ('wx.lib.agw.customtreectrl',['CustomTreeCtrl']),

    82

    ('wx.lib.intctrl',['IntCtrl']),

    83

    ('matplotlib.pyplot',[]),

    84

    ('matplotlib.backends.backend_wxagg',['FigureCanvasWxAgg']),

    85

    ('wx.stc',['StyledTextCtrl']),

    86

    ('wx.lib.scrolledpanel',[]),

    87

    ('wx.lib.mixins.listctrl',['ColumnSorterMixin', 'ListCtrlAutoWidthMixin']),

    88

    ('wx.dataview',['PyDataViewIndexListModel']),

    89

    ('matplotlib.backends.backend_agg',[]),

    90

    ('wx.aui',[]),

    91

    ('mpl_toolkits.mplot3d',[])]:

    92

    modpath = None

    93

    parentmod = None

    94

    for identifier in name.split("."):

    95

        modpath = (modpath + "." + identifier) if modpath else identifier

    96

        mod = sys.modules.get(modpath, None)

    97

    98

        if mod is None or modpath not in already_patched:

    99

            mod = FakeModule(modpath, classes)

   100

            sys.modules[modpath] = mod

   101

            already_patched[modpath] = True

   102

   103

        if parentmod is not None:

   104

            parentmod.__objects__[identifier] = mod

   105

   106

        parentmod = mod

   107

   108

from six.moves import builtins

   109

   110

builtins.__dict__['_'] = get_translation

   111