[coopengo_modules/debug] Allow to autoprofile wizards

nicoe · Sep 13, 2023 · 374f1c3 · 374f1c3
1 parent ff82d69
commit 374f1c3
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/coopengo_modules/debug/__init__.py b/coopengo_modules/debug/__init__.py
@@ -212,11 +212,12 @@ def wrapped(cls, *args, **kwargs):
                 wrapped.__origin_function = f.__origin_function
             return wrapped
 
-        for key, data in config.items('auto_profile'):
+        for _, data in config.items('auto_profile'):
             model, methods = data.split(':')
-            model = model.strip()
+            *pool_type, model = model.strip().split('/')
+            pool_type = (pool_type or ['model'])[0]
 
-            Model = pool._pool[pool.database_name].get('model').get(model)
+            Model = pool._pool[pool.database_name].get(pool_type).get(model)
             for method in methods.split(','):
                 method = method.strip()
                 logger.warning('Enabling auto-profile for %s -> %s' % (