From dad71eebfd72530babecd0fa525f95ee13ba2360 Mon Sep 17 00:00:00 2001 From: Ceda EI Date: Sun, 7 Apr 2024 20:12:03 +0530 Subject: [PATCH] Accept arguments for stdout, stderr rich consoles --- inni/modules/base.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/inni/modules/base.py b/inni/modules/base.py index adb8521..17d3f9d 100644 --- a/inni/modules/base.py +++ b/inni/modules/base.py @@ -1,3 +1,5 @@ +from rich.console import Console + from inni.template import template_to_vars @@ -5,8 +7,10 @@ class BaseModule: login_template_keys = () logout_template_keys = () - def __init__(self, config: dict): + def __init__(self, config: dict, out: Console, err: Console): self.config = config + self.out = out + self.err = err self.setUp() def setUp(self):