def iniciar_atendimento(self):
"""
Inicia o atendimento
Returns:
None
"""
print("Sua senha é {}".format(next(self.gerador_de_senhas)))
print("Aguarde para ser atendido...")
time.sleep(3)
interface = Interface(title="BANCO DA FAMÍLIA FELIZ",
message="Aquilo que é seu, é nosso!",
box_weight=2,
bottom_weight=1,
bottom_padding=1)
interface.add_menu_option("Abrir conta", self.atendimento_abrir_conta)
interface.add_menu_option("Encerrar conta",
self.atendimento_encerrar_conta)
interface.add_menu_option("Logar como administrador",
self.atendimento_logar_admin)
interface.run()
return
def __init__(self, tests_directory_path, local_modules_path, modules_path=None, debug_level=0):
"""
Constructor
"""
self.interface = Interface(debug_level=debug_level)
self.interface.debug('Starting MakeTests', 1)
if not os.path.isdir(local_modules_path):
self.interface.error('No such dir: ' + local_modules_path, 1)
if not os.path.isdir(tests_directory_path):
self.interface.error('No such dir: ' + tests_directory_path, 1)
self.__local_modules_path = local_modules_path
self.__modules_path = local_modules_path
if modules_path:
self.__modules_path = modules_path
self.__tests_directory_path = tests_directory_path
self.__default_template_file = 'puppet_module_test.py'
self.__test_file_name_prefix = 'TestPuppetModule'
self.__modules = []
self.__module_templates = {}
self.__make_tests_dir = os.path.dirname(os.path.abspath(__file__))
self.set_templates_dir('puppet_tests/templates')
self.set_internal_modules_path('/etc/puppet/modules')
self.set_internal_manifests_path('/etc/puppet/manifests')
self.set_image_path(None)
self.find_modules()
def iniciar_atendimento(self):
"""
Realiza o atendimento de um usuário
Returns:
None
"""
interface = Interface(
title="BANCO DA FAMÍLIA FELIZ",
message="Onde você, cliente, é o nosso maior patrimômio!",
box_weight=2,
bottom_weight=1,
bottom_padding=1)
interface.add_menu_option("Entrar na conta", self.atender_entrada)
interface.add_menu_option("Realizar depósito", self.atender_deposito)
interface.run()
def main():
banco = Banco("BBF", "12312")
banco.abrir_conta(ContaCorrente,
"Jorge",
"12345",
"123",
juros=0.1,
limite=2000)
banco.abrir_conta(ContaPoupanca, "José", "12345", "321")
banco.abrir_conta(ContaAdministrativa, "Maercelo", "12345", "777")
atm = ATM(banco)
interface = Interface(title="BANCO DA FAMÍLIA FELIZ",
message="Seja bem-vinda, potencial fonte de renda!",
box_weight=2,
bottom_weight=1,
bottom_padding=1)
interface.add_menu_option("Atendimento Mesas", banco.iniciar_atendimento)
interface.add_menu_option("Autoatendimento", atm.iniciar_atendimento)
interface.run()
return
def get_tipo_de_conta():
"""
Apresenta opções para selecionar um tipo de conta para ser aberta
Returns:
Subclasse de Conta
"""
interface = Interface(
title="BANCO DA FAMÍLIA FELIZ",
message="Escolha o tipo de conta",
box_weight=2,
bottom_weight=1,
bottom_padding=1,
return_selected_option=True,
)
interface.add_menu_option("Conta Corrente", ContaCorrente)
interface.add_menu_option("Conta Poupanca", ContaPoupanca)
interface.add_menu_option("Conta Administrativa", ContaAdministrativa)
tipo_selecionado = interface.run()
return tipo_selecionado
class MakeTests:
"""
This is main class. It finds all modules in the given directory and creates tests for them.
You should give constructor following arguments:
- *local_modules_path* Path to puppet modules which will be scanned for test files
- *tests_directory_path* Output directory where files will be written
- *modules_path* (Optional) Use this path to modules on test host system instead of local_modules_path.
Useful when path to puppet modules differ on machine where tests are made and where they are executed.
"""
def __init__(self, tests_directory_path, local_modules_path, modules_path=None, debug_level=0):
"""
Constructor
"""
self.interface = Interface(debug_level=debug_level)
self.interface.debug('Starting MakeTests', 1)
if not os.path.isdir(local_modules_path):
self.interface.error('No such dir: ' + local_modules_path, 1)
if not os.path.isdir(tests_directory_path):
self.interface.error('No such dir: ' + tests_directory_path, 1)
self.__local_modules_path = local_modules_path
self.__modules_path = local_modules_path
if modules_path:
self.__modules_path = modules_path
self.__tests_directory_path = tests_directory_path
self.__default_template_file = 'puppet_module_test.py'
self.__test_file_name_prefix = 'TestPuppetModule'
self.__modules = []
self.__module_templates = {}
self.__make_tests_dir = os.path.dirname(os.path.abspath(__file__))
self.set_templates_dir('puppet_tests/templates')
self.set_internal_modules_path('/etc/puppet/modules')
self.set_internal_manifests_path('/etc/puppet/manifests')
self.set_image_path(None)
self.find_modules()
def set_templates_dir(self, template_dir):
"""
Set directory to take templates from
"""
if not os.path.isdir(template_dir):
self.interface.error("No such dir: " + template_dir, 1)
self.__template_directory_path = template_dir
self.__template_loader = jinja2.FileSystemLoader(searchpath=template_dir)
self.__template_environment = templateEnv = jinja2.Environment(
loader=self.__template_loader,
)
def set_image_path(self, image_path):
"""
"""
self.__image_path = image_path
def set_module_template_overrides(self, module_templates_dictionary):
"""
Set module template file override dictionary
"""
if type(module_templates_dictionary) is dict:
self.__module_templates = module_templates_dictionary
else:
self.interface.error("Argument is not Dictionary", 1)
def set_default_template_file(self, template_file):
"""
Set default script template file
"""
self.__default_template_file = template_file
def set_internal_modules_path(self, internal_modules_path):
"""
Set path to modules inside virtual machine
"""
self.__internal_modules_path = internal_modules_path
def set_internal_manifests_path(self, internal_manifests_path):
"""
Set path to manifests directory inside virtual machine
"""
self.__internal_manifests_path = internal_manifests_path
def get_modules_list(self):
"""
Get list of PuppetModule objects
"""
return self.__modules
def find_modules(self):
"""
Find all Puppet modules in module_library_path
and create array of PuppetModule objects
"""
self.interface.debug('Starting findModules in "%s"' % self.__local_modules_path, 2)
for module_dir in os.listdir(self.__local_modules_path):
full_local_module_path = os.path.join(self.__local_modules_path, module_dir)
full_local_tests_path = os.path.join(full_local_module_path, 'tests')
if not os.path.isdir(full_local_tests_path):
continue
self.interface.debug('Found Puppet module: "%s"' % full_local_module_path, 3)
puppet_module = PuppetModule(full_local_module_path, self.interface)
self.__modules.append(puppet_module)
def compile_script(self, module):
"""
Compile script template for given module and return it
"""
template_file = self.__module_templates.get(module.get_name(), self.__default_template_file)
template = self.__template_environment.get_template(template_file)
general = {
'modules_path': self.__modules_path,
'local_modules_path': self.__local_modules_path,
'internal_modules_path': self.__internal_modules_path,
'internal_manifests_path': self.__internal_manifests_path,
'tests_directory_path': self.__tests_directory_path,
'image_path': self.__image_path,
}
compiled_template = template.render(module=module, **general)
return compiled_template
def save_script(self, module):
"""
Saves compiled script to a file
"""
file_name = self.__test_file_name_prefix + module.get_name().title() + '.py'
full_file_path = os.path.join(self.__tests_directory_path, file_name)
script_content = self.compile_script(module)
script_file = open(full_file_path, 'w+')
script_file.write(script_content)
script_file.close()
def make_all_scripts(self):
"""
Compile and save to tests_directory_path all the test scripts. Main procedure.
"""
self.interface.debug('Starting makeAllScripts', 2)
for module in self.get_modules_list():
self.interface.debug('Processing module: "%s"' % module.get_name(), 3)
self.save_script(module)
def remove_all_tests(self):
"""
Remove all tests from tests_directory_path
"""
self.interface.debug('Starting removeAllTests in "%s"' % self.__tests_directory_path, 2)
file_list = os.listdir(self.__tests_directory_path)
for test_file in file_list:
if not test_file.endswith('.py'):
continue
if not test_file.startswith('TestPuppetModule'):
continue
full_file_path = os.path.join(self.__tests_directory_path, test_file)
self.interface.debug('Removing test file: "%s"' % full_file_path, 3)
os.remove(full_file_path)
def atender_cliente(self):
"""
Atende a logins de contas comuns
Returns:
None
"""
interface = Interface(
title="BANCO DA FAMÍLIA FELIZ",
message="Sua felicidade é nossa maior prioridade!",
box_weight=2,
bottom_weight=1,
bottom_padding=1)
interface.add_menu_option("Saque", self.atender_saque)
interface.add_menu_option("Depósito", self.atender_deposito)
interface.add_menu_option("Transferência", self.atender_transferencia)
interface.add_menu_option("Extrato", self.fornecer_extrato)
interface.add_menu_option("Saldo", self.fornecer_saldo)
interface.run()
self.desconectar_conta()
def atender_gerente(self):
"""
Atende a logins de contas administrativas
Returns:
None
"""
interface = Interface(title="BANCO DA FAMÍLIA FELIZ",
message="Trabalhe, escravo, trabalhe!",
box_weight=2,
bottom_weight=1,
bottom_padding=1)
interface.add_menu_option("Saldar ATM", self.listar_notas)
interface.add_menu_option("Resetar ATM", self.reset)
interface.add_menu_option("Listar operações recentes",
self.mostrar_historico)
interface.add_menu_option("Depositar na ATM", self.adicionar_dinheiro)
interface.run()
self.desconectar_conta()
def startGame(self): id = Interface.classPick(4) self.createHero(id) Interface.greetPick(self.hero.name)
