pankajp/caching_mod.py

## 12 changes: 12 additions & 0 deletions caching_mod.py
@@ -0,0 +1,12 @@

    from mod_prop import mod_property, cached_mod_property
from mod_prop import mod_property, cached_mod_property


    @mod_property
@mod_property

    def my_prop():
def my_prop():

        print('my_prop called')
    print('my_prop called')

        return 42
    return 42


    @cached_mod_property
@cached_mod_property

    def my_cached_prop():
def my_cached_prop():

        print('my_cached_property called')
    print('my_cached_property called')

        return '24'
    return '24'

## 13 changes: 13 additions & 0 deletions caching_mod_check.py
@@ -0,0 +1,13 @@

    import caching_mod
import caching_mod

    print(caching_mod)
print(caching_mod)

    print('p', caching_mod.my_prop)
print('p', caching_mod.my_prop)

    print('q', caching_mod.my_prop)
print('q', caching_mod.my_prop)


    from caching_mod import my_prop
from caching_mod import my_prop

    print('r', caching_mod.my_prop)
print('r', caching_mod.my_prop)

    print('s', caching_mod.my_prop)
print('s', caching_mod.my_prop)


    from caching_mod import my_cached_prop
from caching_mod import my_cached_prop

    print('t')
print('t')

    print('u', my_cached_prop)
print('u', my_cached_prop)

    print('v', caching_mod.my_cached_prop)
print('v', caching_mod.my_cached_prop)

## 33 changes: 33 additions & 0 deletions mod_prop.py
@@ -0,0 +1,33 @@

    """ Module level property and cached property decorator """
""" Module level property and cached property decorator """

    import sys
import sys

    import types
import types

    import functools
import functools


    def update_module_class(mod):
def update_module_class(mod):

        class CachingModule(types.ModuleType):
    class CachingModule(types.ModuleType):

            pass
        pass

        mod.__class__ = CachingModule
    mod.__class__ = CachingModule


    def mod_property(func, cached=False):
def mod_property(func, cached=False):

        func_name = func.__name__
    func_name = func.__name__

        if '.' in func_name:
    if '.' in func_name:

            raise ValueError('mod_property only applicable to top-level module functions')
        raise ValueError('mod_property only applicable to top-level module functions')

        func_mod = sys.modules[func.__module__]
    func_mod = sys.modules[func.__module__]

        if func_mod.__class__ == types.ModuleType:
    if func_mod.__class__ == types.ModuleType:

            update_module_class(func_mod)
        update_module_class(func_mod)

        elif func_mod.__class__.__name__ != 'CachingModule':
    elif func_mod.__class__.__name__ != 'CachingModule':

            raise RuntimeError(f'mod_property incompatible with module type: {func_mod.__name__}({func_mod.__class__.__qualname__})')
        raise RuntimeError(f'mod_property incompatible with module type: {func_mod.__name__}({func_mod.__class__.__qualname__})')

        @functools.wraps(func)
    @functools.wraps(func)

        def wrapper(mod):
    def wrapper(mod):

            value = func()
        value = func()

            if cached:
        if cached:

                setattr(func_mod.__class__, func_name, value)
            setattr(func_mod.__class__, func_name, value)

                delattr(func_mod, func_name)
            delattr(func_mod, func_name)

            return value
        return value

        wrapper.__name__ = func_name
    wrapper.__name__ = func_name

        setattr(func_mod.__class__, func_name, property(wrapper))
    setattr(func_mod.__class__, func_name, property(wrapper))

        return wrapper
    return wrapper


    def cached_mod_property(func):
def cached_mod_property(func):

        return mod_property(func, cached=True)
    return mod_property(func, cached=True)