Czy możliwe jest wykonanie modułu w języku Python?

Question

Tworzę moduł, który chcę traktować jako statyczny kontener obiektów. Te obiekty są typu klasy, które zdefiniowałem. Chcę móc zaimportować ten moduł, a następnie zapętlić obiekty wewnątrz. Oto kod wyjaśniając o co mi chodzi:

example.py

class MyExampleClass(object): 
    def __init__(self, var1, var2, var3): 
     self.var1 = var1 
     self.var2 = var2 
     self.var3 = var3 
     self.var4 = var4 

instanceA = MyExampleClass(1, 2, 3, 4) 
instanceB = MyExampleClass(4, 3, 6, 7) 
instanceC = MyExampleClass(5, 3, 4, 5) 

# something like this 
def __iter__(): 
    return (instanceA, instanceB, instanceC) 

Następnie chciałbym, aby móc importować ten i używać go jak ENUM:

import example 

for e in example: 
    # do stuff with e 

Is to można zrobić w Pythonie? Czy będę musiał zaimportować listę z pakietu example?

example.py

objects = (instanceA, instanceB, instanceC) 

a następnie

import example 

for e in example.objects: 
    # do stuff with e 

Answer 1

Można to osiągnąć poprzez zdefiniowanie klasy głównego w module i wymianie modułu z instancji tej klasy. Oto przykład:

class ModuleClass(object): 
    __init__(self): 
     self.instanceA = MyExampleClass(1, 2, 3, 4) 
     ... 

    __iter__(self): 
     # Your iterator logic here 

# and then in the same module code 
sys.modules[__name__] = ModuleClass() 

Tak więc możesz robić, co chcesz, ponieważ podczas importowania tego modułu, to rzeczywiście być instancją swoim zwyczajem iterowalny ModuleClass:

import example 

for e in example: 
    # do stuff with e 

Answer 2

Choć robię nie polecam, możesz zbudować własny moduł dziedzicząc po types.ModuleType. Możesz następnie zastąpić oryginalny moduł niestandardową klasą przez sys.modules[__name__] = NewModule()

import types 
import sys 

class MyExampleClass(object): 
    def __init__(self, var1, var2, var3, var4): 
     self.var1 = var1 
     self.var2 = var2 
     self.var3 = var3 
     self.var4 = var4 

class MyModule(types.ModuleType): 

    def __init__(self, *args, **kwargs): 
     super().__init__(*args, **kwargs) 
     self.instanceA = MyExampleClass(1, 2, 3, 4) 
     self.instanceB = MyExampleClass(4, 3, 6, 7) 
     self.instanceC = MyExampleClass(5, 3, 4, 5) 

    def __iter__(self): 
     return iter([self.instanceA, self.instanceB, self.instanceC]) 

sys.modules[__name__] = MyModule("example") # Name of the module