features.py

### lib modul

FEATURES = {}

def feature(name, source):
    
    def wrapper(f):
        FEATURES[name] = {
            "name": name,
            "source": source,
            "fn": f,
        }
    
    return wrapper
    

def build_train_data(features):
    context = {}
    data = {f: FEATURES[f]["fn"](context) for f in features}
    return data
    
### moduli s featurima
    
@feature(name="sales_sum", source="sales")
def get_sales_sum(context):
    # context.sales_source.select("...")
    return 10

@feature(name="trips_sum", source="sales")
def get_trips_sum(context):
    # context.sales_source.select("...")
    return 12

### pipeline

train_data = build_train_data(["sales_sum", "trips_sum"])
print(train_data)