import sys

from .magic import SelfWrapper


def register_module(name):
    self = sys.modules['zipa']
    sys.modules[name] = SelfWrapper(self)


class ModuleImporter(object):
    def __init__(self, module_name):
        self.name = module_name

    def find_module(self, name, path=None):
        if name.startswith('{}.'.format(self.name)):
            return self
        return None

    def load_module(self, name):
        register_module(name)

import sys

from .magic import SelfWrapper


def register_module(name):
    self = sys.modules['zipa']
    sys.modules[name] = SelfWrapper(self)


class ModuleImporter(object):
    def __init__(self, module_name):
        self.module = module_name + '.'

    def find_module(self, name, path=None):
        if name.startswith(self.module):
            return self
        return None

    def load_module(self, name):
        register_module(name)

Make the logic more readable
Make the logic more readable
