From 5bc4a65eea5c59ecddb71499915934190fca7d5c Mon Sep 17 00:00:00 2001 From: pukkandan Date: Mon, 23 Aug 2021 03:48:26 +0530 Subject: [PATCH] [lazy_extractor] Import actual class if an attribute is accessed Now all core tests pass with lazy extraction enabled --- devscripts/lazy_load_template.py | 20 ++++++++++++++------ 1 file changed, 14 insertions(+), 6 deletions(-) diff --git a/devscripts/lazy_load_template.py b/devscripts/lazy_load_template.py index 20322e04b..a3f3fedf9 100644 --- a/devscripts/lazy_load_template.py +++ b/devscripts/lazy_load_template.py @@ -1,16 +1,24 @@ -#!/usr/bin/env python3 # coding: utf-8 -from __future__ import unicode_literals - import re -class LazyLoadExtractor(object): +class LazyLoadMetaClass(type): + def __getattr__(cls, name): + return getattr(cls._get_real_class(), name) + + +class LazyLoadExtractor(metaclass=LazyLoadMetaClass): _module = None + @classmethod + def _get_real_class(cls): + if '__real_class' not in cls.__dict__: + mod = __import__(cls._module, fromlist=(cls.__name__,)) + cls.__real_class = getattr(mod, cls.__name__) + return cls.__real_class + def __new__(cls, *args, **kwargs): - mod = __import__(cls._module, fromlist=(cls.__name__,)) - real_cls = getattr(mod, cls.__name__) + real_cls = cls._get_real_class() instance = real_cls.__new__(real_cls) instance.__init__(*args, **kwargs) return instance