cffi_generator.py

import os
import pickle

import etgtools.extractors as extractors
import etgtools.generators as generators
from etgtools.generators import nci, Utf8EncodingStream, textfile_open, wrapText

from buildtools.config import Config
cfg = Config(noWxConfig=True)

AUTO_PYNAME_TYPES = (extractors.ClassDef, extractors.FunctionDef,
                     extractors.EnumValueDef, extractors.VariableDef)

class CffiWrapperGenerator(generators.WrapperGeneratorBase):
    def generate(self, module):
        outfile = module.name + '.def'
        outfile = os.path.join(cfg.ROOT_DIR, 'cffi', 'def_gen', outfile)

        # We need to generate pyDocstring for some objects so later generators
        # (pi_generator) can access it. (The sip generator also does this)
        # For now just use a dummy value, this is pretty inconsequential
        for item in module.allItems():
            if not hasattr(item, 'pyDocstring'):
                item.pyDocstring = ''

        self.stripIgnoredItems(module.items)
        self.trimPrefixes(module.items)

        with open(outfile, 'wb') as f:
            pickle.dump(module, f, 2)

    @classmethod
    def stripIgnoredItems(cls, items):
        """
        Strip any ignored items; they aren't useful to the module generator and
        just waste space.
        """
        dummy = object()
        for i, e in enumerate(items):
            if e.ignored:
                items[i] = dummy

                if hasattr(e, 'overloads') and len(e.overloads) > 0:
                    # If a method is ignored, replace it with the first
                    # overload that isn't ignored
                    cls.stripIgnoredItems(e.overloads)
                    if len(e.overloads) > 0:
                        e.overloads[0].overloads = e.overloads[1:]
                        items[i] = e.overloads[0]

            else:
                cls.stripIgnoredItems(e.items)
                cls.stripIgnoredItems(getattr(e, 'overloads', []))

        while True:
            try:
                items.remove(dummy)
            except ValueError:
                break

    @classmethod
    def trimPrefixes(cls, items):
        for item in items:
            if not item.pyName:
                if (isinstance(item, AUTO_PYNAME_TYPES) and
                    item.name.startswith('wx')):
                    item.pyName = item.name[2:]
                else:
                    item.pyName = item.name
            cls.trimPrefixes(item.items)
            cls.trimPrefixes(getattr(item, 'innerclasses', []))
            cls.trimPrefixes(getattr(item, 'overloads', []))