codegeex可快速生成postgresql连接与操作脚本：一、用提示词生成psycopg2连接代码；二、辅助编写sqlalchemy orm模型与crud函数；三、将sql语句反向封装为python函数；四、生成带连接池的高并发访问脚本。
☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜
如果您希望使用CodeGeeX快速生成连接并操作PostgreSQL数据库的脚本，则可能是由于手动编写SQL连接逻辑耗时或易出错。以下是实现该目标的具体方法：
一、利用CodeGeeX提示词直接生成Python psycopg2连接脚本
该方法通过向CodeGeeX输入结构清晰的自然语言提示，触发其生成符合PEP 8规范、含异常处理的完整Python连接代码。模型基于对psycopg2常见用法的训练，可准确输出带参数化查询示例的脚本。
1、在支持CodeGeeX插件的IDE（如VS Code）中新建Python文件。
2、输入提示词：“生成一个使用psycopg2连接本地PostgreSQL数据库的Python脚本，数据库名testdb，用户postgres，密码123456，端口5432；包含连接测试、执行SELECT COUNT(*) FROM pg_tables语句、打印结果、关闭连接”。
3、按下快捷键（如Ctrl+Enter）触发CodeGeeX补全，接受生成的代码块。
4、检查生成代码是否包含try-except结构及conn.close()调用，确认无硬编码敏感信息残留。
二、使用CodeGeeX辅助编写SQLAlchemy ORM交互脚本
该方法适用于需面向对象操作PG表的场景，CodeGeeX可依据表结构描述自动生成Model定义与CRUD函数，避免手写映射类和session管理逻辑。
1、在代码编辑区输入提示词：“为PostgreSQL中名为users的表生成SQLAlchemy 2.0风格的ORM模型，字段包括id（Integer, primary_key）、name（String(50)）、email（String(100), unique）；再生成一个add_user函数，接收name和email参数并提交到数据库”。
2、选中生成的Model类代码，再次输入提示词：“补全一个使用上述Model查询所有users并按name升序返回字典列表的函数”。
3、将生成的两个函数合并至同一文件，确保已导入create_engine、Session、select等必要模块。
4、验证生成代码中URL字符串是否使用postgresql+psycopg2://协议前缀。
CodeGeeX 2.21.0是智谱AI推出的AI编程助手版本，对Inline Chat功能进行了重大优化，包括UI设计升级、支持流式输出以提升响应速度，并新增“查看思路”按钮以便理解代码原理。同时，该版本在编辑器中新增CodeLens控件，支持一键“解释”代码或“添加注释”。目前该工具对个人开发者免费开放。
三、借助CodeGeeX从SQL语句反向生成Python执行片段
该方法适用于已有成熟SQL逻辑（如复杂JOIN或窗口函数），需快速封装为可复用Python函数的场景。CodeGeeX能识别SQL语法特征并匹配对应execute()或exec_driver_sql()调用形式。
1、在空行粘贴原始SQL：SELECT u.name, COUNT(o.id) FROM users u LEFT JOIN orders o ON u.id = o.user_id GROUP BY u.name HAVING COUNT(o.id) > 5;。
2、在SQL下方输入提示词：“将以上SQL封装为Python函数get_active_users，使用psycopg2连接，返回结果列表”。
3、接受补全结果，确认函数内含cursor.execute()调用且SQL字符串被正确包裹于三重引号中。
4、检查返回语句是否为return cursor.fetchall()，而非print或未赋值的execute调用。
四、通过CodeGeeX生成带连接池的高并发PG访问脚本
该方法针对需要稳定处理多请求的应用场景，CodeGeeX可基于关键词“连接池”“maxconn”“thread-safe”生成使用psycopg2.pool.SimpleConnectionPool的初始化与获取逻辑。
1、新建文件后输入提示词：“生成使用psycopg2 SimpleConnectionPool连接PostgreSQL的Python模块，最大连接数10，最小连接数2，数据库testdb，用户postgres，密码123456，主机localhost”。
2、在生成的pool初始化代码下方，追加提示词：“添加get_connection()函数，从池中获取连接；添加return_connection(conn)函数，将连接归还池中”。
3、确认生成代码中import语句包含from psycopg2 import pool，且SimpleConnectionPool构造参数顺序与文档一致。
4、检查get_connection()是否捕获psycopg2.OperationalError并触发重新初始化池的操作。
