diff --git a/src/actions/posicao.py b/src/actions/posicao.py index edf56c5..0def305 100644 --- a/src/actions/posicao.py +++ b/src/actions/posicao.py @@ -13,13 +13,12 @@ def __init__(self, args, web, excel): self.args = args self.web = web self.excel = excel - # TODO access like args.cods_cliente - self.cods_clientes = pasted_to_list(self.args['cods_cliente'] or '') - self.nomes_clientes = pasted_to_list(self.args['nomes_cliente'] or '') + self.cods_clientes = pasted_to_list(self.args.cods_cliente or '') + self.nomes_clientes = pasted_to_list(self.args.nomes_cliente or '') if not self.cods_clientes or not self.nomes_clientes: for status, rede, nome_cliente, cod_cliente in self.excel.file_vertical_search( - self.args['nome_rede'].upper(), self.args['dinamica'], 9, 5, 9, 10, 11): + self.args.nome_rede.upper(), self.args.dinamica, 9, 5, 9, 10, 11): if cod_cliente is None or cod_cliente == '': msg(f'CLIENTE SEM CÓDIGO: "{rede}" --> "{nome_cliente}"') continue @@ -30,7 +29,7 @@ def __init__(self, args, web, excel): self.cods_clientes.append(cod_cliente) self.nomes_clientes.append(nome_cliente) - self.prevs_emb = pasted_to_list(self.args['prevs_emb'] or '') + self.prevs_emb = pasted_to_list(self.args.prevs_emb or '') if not self.prevs_emb: from datetime import datetime from utils import datetime_to_simple @@ -52,7 +51,7 @@ def __init__(self, args, web, excel): self.web.open() self.web.totvs_access() if not self.web.totvs_logged: - password = self.args['senha_totvs'] + password = self.args.senha_totvs self.web.totvs_login(password) self.web.totvs_fav_pedidos() @@ -125,10 +124,10 @@ def make_sheet(self, cod_cliente, nome_cliente): if __name__ == '__main__': from automators import web, excel - from utils import example_args + from utils import ExampleArgs Posicao( - example_args, + ExampleArgs(), web.Web(), excel.Excel() ) diff --git a/src/main.py b/src/main.py index ef4ac0f..ac60b34 100644 --- a/src/main.py +++ b/src/main.py @@ -70,9 +70,9 @@ def group(parser_, name, **kwargs): argument(posicao_advanced_group, '--nomes_cliente', widget='Textarea', gooey_options={'height': 100}, help="Os respectivos nomes para cada cliente, separados por ENTER") - args = parser.parse_args().__dict__ - cache['default'] = {key: str(value) for key, value in args.items()} - args = {key: None if value == 'X' else value.strip() for key, value in args.items()} + args = parser.parse_args() + cache['default'] = {key: str(value) for key, value in args.__dict__.items()} + args.__dict__ = {key: None if value == 'X' else value.strip() for key, value in args.__dict__.items()} save_cache() def run(action): @@ -95,7 +95,7 @@ def run(action): log_file.write(traceback.format_exc() + '\n') raise e - if args['action'] == 'Posição': + if args.action == 'Posição': from actions.posicao import Posicao run(Posicao) diff --git a/src/utils.py b/src/utils.py index c48d88e..37b7adb 100644 --- a/src/utils.py +++ b/src/utils.py @@ -159,22 +159,22 @@ def new_function(*args, **kwargs): return new_function -example_args = { - 'cods_cliente': """969611 +class ExampleArgs: + cods_cliente = """969611 1000560 611379 980420 -""", - 'nomes_cliente': """""", - 'prevs_emb': """03122021 +""" + nomes_cliente = """""" + prevs_emb = """03122021 03012022 03022022 -""", - 'senha_totvs': "Comunidade15", - 'dinamica': '/Users/macbookpro/Desktop/NOVA DINÂMICA.xlsx', - # 'dinamica': 'C:\\Users\\Nathan\\Downloads\\NOVA DINÂMICA.xlsx', - 'nome_rede': 'diversa' -} +""" + senha_totvs = "Comunidade15" + dinamica = '/Users/macbookpro/Desktop/NOVA DINÂMICA.xlsx' + # 'dinamica = 'C:\\Users\\Nathan\\Downloads\\NOVA DINÂMICA.xlsx' + nome_rede = 'diversa' + if __name__ == '__main__': print(msg("nova dinâmica"))