Source code for enamlnative.core.import_hooks

"""
Copyright (c) 2017, CodeLV.

Distributed under the terms of the MIT License.

The full license is in the file LICENSE, distributed with this software.

Created on July 10, 2017

@author: jrm
"""
import os
import sys
import imp
from glob import glob


[docs]class ExtensionImporter(object):
    """ Loads renamed extensions files from the app's lib folder"""
    extension_modules = {}

[docs]    def __init__(self):
        #: On iOS extensions are built as dylibs
        ext_type = "dylib" if sys.platform == 'darwin' else 'so'

        #: Android only copies libraries with a lib prefix
        prefix = "" if sys.platform == 'darwin' else 'lib.'
        start = 0 if sys.platform == 'darwin' else 1

        #: Find all included extension modules
        lib_dir = os.environ.get('PY_LIB_DIR','.')
        #print("Loading {} extensions from {}".format(ext_type, lib_dir))

        for lib in glob('{}/{}*.{}'.format(lib_dir, prefix, ext_type)):
            name = lib.split("/")[-1] # Lib filename
            mod = ".".join(name.split(".")[start:-1])  # Strip lib and so
            self.extension_modules[mod] = lib

        #print("Libraries found: {}".format(self.extension_modules))

[docs]    def load_module(self, mod):
        """ Load the extension using the load_dynamic method. """
        try:
            return sys.modules[mod]
        except KeyError:
            pass

        lib = ExtensionImporter.extension_modules[mod]
        m = imp.load_dynamic(mod, lib)
        sys.modules[mod] = m
        return m

[docs]    def find_module(self, mod, path=None):
        """ Use this as the loader if the desired module is an extension
            within the given library folder.
        """
        if mod in self.extension_modules:
            return self
        return None