Accept arguments for stdout, stderr rich consoles

This commit is contained in:
Ceda EI 2024-04-07 20:12:03 +05:30
parent ffe3ae76bb
commit dad71eebfd
1 changed files with 5 additions and 1 deletions

View File

@ -1,3 +1,5 @@
from rich.console import Console
from inni.template import template_to_vars from inni.template import template_to_vars
@ -5,8 +7,10 @@ class BaseModule:
login_template_keys = () login_template_keys = ()
logout_template_keys = () logout_template_keys = ()
def __init__(self, config: dict): def __init__(self, config: dict, out: Console, err: Console):
self.config = config self.config = config
self.out = out
self.err = err
self.setUp() self.setUp()
def setUp(self): def setUp(self):