setup.py

"""
Script de setup inicial do projeto.
Cria estrutura de diretórios e arquivos necessários.
"""

import sys
from pathlib import Path


def setup_project():
    """Configura a estrutura inicial do projeto."""
    print("Configurando Avature Scraper Pro...")

    # Diretórios necessários
    directories = [
        "data/inputs",
        "data/outputs",
        "logs",
        "tests",
        "examples",
    ]

    # Cria diretórios
    for directory in directories:
        Path(directory).mkdir(parents=True, exist_ok=True)
        print(f"[OK] Diretorio criado: {directory}")

    # Cria arquivo .env se não existir
    env_file = Path(".env")
    env_example = Path(".env.example")

    if not env_file.exists() and env_example.exists():
        import shutil

        shutil.copy(env_example, env_file)
        print("[OK] Arquivo .env criado a partir de .env.example")

    # Cria arquivo domains.txt se não existir
    domains_file = Path("data/inputs/domains.txt")
    domains_example = Path("data/inputs/domains.txt.example")

    if not domains_file.exists():
        if domains_example.exists():
            import shutil

            shutil.copy(domains_example, domains_file)
            print("[OK] Arquivo data/inputs/domains.txt criado")
        else:
            domains_file.touch()
            print("[OK] Arquivo data/inputs/domains.txt criado (vazio)")

    print("\n[SUCESSO] Setup concluido!")
    print("\nProximos passos:")
    print("1. Edite data/inputs/domains.txt com os dominios que deseja processar")
    print("2. Execute: python main.py")


if __name__ == "__main__":
    try:
        setup_project()
    except Exception as e:
        print(f"❌ Erro durante setup: {e}", file=sys.stderr)
        sys.exit(1)