如何用DataStax Python Cassandra驱动执行CQL文件初始化KeySpace表

我之前在使用Python Cassandra驱动时也碰到过一模一样的问题——官方文档里确实没提供直接支持SOURCE命令的API，execute()又只能处理单条CQL语句，当时折腾了好一会儿才搞定，给你分享几个实用的解决方案：

方案一：手动解析并逐条执行CQL语句 #

这是最可控的方法，核心就是读取CQL文件后，把内容拆分成单个语句，过滤掉注释和空行，再逐个执行。其实代码量并不大，而且能灵活处理各种情况：

from cassandra.cluster import Cluster

def execute_cql_file(session, file_path):
    with open(file_path, 'r') as f:
        cql_content = f.read()
    
    # 按分号分割语句，同时清理掉多余的空格和换行
    cql_statements = [stmt.strip() for stmt in cql_content.split(';') if stmt.strip()]
    
    for stmt in cql_statements:
        # 跳过单行注释和空语句
        if not stmt.startswith('--'):
            try:
                session.execute(stmt)
                print(f"✅ 成功执行: {stmt[:50]}..." if len(stmt) > 50 else f"✅ 成功执行: {stmt}")
            except Exception as e:
                print(f"❌ 执行失败: {stmt[:50]}... 错误: {str(e)}")

# 初始化Cassandra连接
cluster = Cluster(['127.0.0.1'])  # 替换成你的Cassandra节点地址
session = cluster.connect()

# 切换到目标KeySpace（如果CQL文件里没有USE语句的话）
session.set_keyspace('your_target_keyspace')

# 执行CQL文件
execute_cql_file(session, 'path/to/your/schema.cql')

# 关闭连接
cluster.shutdown()

小提醒： #

• 如果你的CQL语句里包含带分号的字符串（比如INSERT语句中的文本内容），简单的分号分割会出错，这种情况可以加个简单的状态机判断是否在字符串引号内，避免误分割。
• 像DROP TABLE、CREATE TABLE这类DDL语句，Cassandra本身是原子执行的，不用担心部分执行的问题。

方案二：通过subprocess调用cqlsh执行SOURCE命令 #

如果不想自己写解析逻辑，也可以直接在Python里调用cqlsh工具，用它原生的SOURCE命令来执行整个文件，相当于模拟手动操作cqlsh：

import subprocess

def run_cqlsh_source(cql_file, cassandra_host='127.0.0.1', keyspace='your_target_keyspace'):
    # 构造cqlsh执行命令
    cmd = [
        'cqlsh', cassandra_host,
        '-k', keyspace,
        '-e', f"SOURCE '{cql_file}';"
    ]
    
    try:
        result = subprocess.run(cmd, check=True, capture_output=True, text=True)
        print("🎉 CQL文件执行成功！输出:", result.stdout)
    except subprocess.CalledProcessError as e:
        print("😢 执行失败，错误信息:", e.stderr)

# 调用函数执行CQL文件
run_cqlsh_source('path/to/your/schema.cql')

优缺点分析： #

• 优点：完全复用cqlsh的语法支持，不用自己处理复杂的CQL解析，省心省力。
• 缺点：依赖系统环境中已安装的cqlsh，而且要确保Python进程能访问到cqlsh的可执行文件，跨平台兼容性稍弱。

内容的提问来源于stack exchange，提问作者dbrew5