diff --git a/python/dataload.ipynb b/python/dataload.ipynb index 689b71d..0b7a3a3 100644 --- a/python/dataload.ipynb +++ b/python/dataload.ipynb @@ -201,8 +201,7 @@ " return f'{normalizar(nome_email)}_{gerar_valor_aleatorio(4)}@{choice(provedores_email)}'\n", "\n", "def gerar_telefone() -> str:\n", - " return f'{choice(ddds)}9{randint(1000, 9999)}{randint(1000,9999)}'\n", - "\n" + " return f'{choice(ddds)}9{randint(1000, 9999)}{randint(1000,9999)}'" ] }, { @@ -385,10 +384,13 @@ "gestor_ciclo = pd.DataFrame(dados_colaborador).groupby('id_empresa').first().reset_index()\n", "\n", "for _, gestor in gestor_ciclo.iterrows():\n", - " qtd_ciclos = QTD_CICLOS_FIXOS if gestor['id_empresa'] in [1, 2, 3] else randint(QTD_CICLOS_MIN, QTD_CICLOS_MAX)\n", + " if gestor['id_empresa'] in [1, 2, 3]:\n", + " status_por_ciclo = status_ciclo\n", + " else:\n", + " qtd_ciclos = randint(QTD_CICLOS_MIN, QTD_CICLOS_MAX)\n", + " status_por_ciclo = [choices(status_ciclo, weights=[25, 25, 25, 20, 3, 1, 1], k=1)[0] for _ in range(qtd_ciclos)]\n", "\n", - " for _ in range(qtd_ciclos):\n", - " status = choices(status_ciclo, weights=[25, 25, 25, 20, 3, 1, 1], k=1)[0]\n", + " for status in status_por_ciclo:\n", " dados_ciclo.append({\n", " 'id_empresa': gestor['id_empresa'],\n", " 'id_responsavel': gestor['id_usuario'],\n",