from openpyxl import Workbook

import mysql.connector


def export_cadastros_to_excel():
    # Configurações de conexão com o MySQL
    conn = mysql.connector.connect(
        host="localhost",         # altere para o host do seu banco de dados
        user="root",       # altere para o usuário do seu banco
        password="",     # altere para a senha do seu banco
        database="xcmg"      # altere para o nome do seu banco de dados
    )

    cursor = conn.cursor()
    
    # Consulta da tabela cadastros
    cursor.execute("""
        SELECT id, nome, empresa, cargo, telefone, cliente_xcmg, usa_banco_xcmg, codigo, ativado, gerado, status
        FROM cadastros
    """)
    rows = cursor.fetchall()
    
    cursor.close()
    conn.close()

    # Cria um arquivo Excel
    wb = Workbook()
    ws = wb.active
    ws.title = "Cadastros"

    # Cabeçalho da planilha
    headers = ['id', 'nome', 'empresa', 'cargo', 'telefone', 'cliente_xcmg', 'usa_banco_xcmg', 'codigo', 'ativado', 'gerado', 'status']
    ws.append(headers)

    # Insere os dados na planilha
    for row in rows:
        ws.append(row)

    # Salva a planilha em um arquivo Excel
    wb.save("cadastros.xlsx")

if __name__ == '__main__':
    export_cadastros_to_excel()