一个支持MYSQL语句的PYTHON脚本

起因是，mysql无法连接远程的机器，但是用python却可以。

写了这个脚本

import pymysql
import argparse

# MySQL 连接信息
host = '192.168.1.1'  # 替换为你的 MySQL 服务器 IP
user = 'ecology'         # 替换为你的 MySQL 用户名
password = 'ecology1234'  # 替换为你的 MySQL 用户密码
database = 'ecology'   # 替换为数据库名




def execute_query(query):
    try:
        # 建立连接
        connection = pymysql.connect(
            host=host,
            user=user,
            password=password,
            database=database,
            port=3306,   # MySQL 默认端口
            connect_timeout=5  # 设置超时时间
        )
        print("Connection successful!")

        # 执行查询
        with connection.cursor() as cursor:
            cursor.execute(query)
            results = cursor.fetchall()  # 获取所有结果
            columns = [desc[0] for desc in cursor.description] if cursor.description else []  # 获取列名

            print("Query executed successfully.")
            return columns, results  # 返回列名和查询结果

    except pymysql.MySQLError as e:
        print("Connection failed.")
        print(f"Error: {e}")
        return None, None

    finally:
        # 关闭连接
        if 'connection' in locals() and connection.open:
            connection.close()
            print("Connection closed.")

if __name__ == "__main__":
    # 设置命令行参数解析
    parser = argparse.ArgumentParser(description="Execute a MySQL query from command line.")
    parser.add_argument("query", type=str, help="The SQL query to execute.")
    args = parser.parse_args()

    # 从命令行参数接收 SQL 查询
    query = args.query
    columns, results = execute_query(query)

    # 打印查询结果
    if results is not None:
        # 打印列名
        print("Columns:", columns)
        # 打印每一行结果
        for row in results:
            print(row)

用法：
python testmy.py "SHOW COLUMNS FROM ws_datasource;"
python testmy.py "select * FROM weaver_sso LIMIT 100;"