65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
数据库创建脚本
|
||
|
|
用于在 PostgreSQL 服务器上创建 ai_vision 数据库
|
||
|
|
"""
|
||
|
|
|
||
|
|
import psycopg2
|
||
|
|
from psycopg2.extensions import ISOLATION_LEVEL_AUTOCOMMIT
|
||
|
|
|
||
|
|
# 数据库连接信息
|
||
|
|
DB_HOST = "331002.xyz"
|
||
|
|
DB_PORT = 2022
|
||
|
|
DB_USER = "user_xREpkJ"
|
||
|
|
DB_PASSWORD = "password_DZz8DQ"
|
||
|
|
DB_NAME = "ai_vision"
|
||
|
|
|
||
|
|
def create_database():
|
||
|
|
"""创建数据库"""
|
||
|
|
try:
|
||
|
|
# 连接到默认的 postgres 数据库
|
||
|
|
print(f"🔗 正在连接到 PostgreSQL 服务器 {DB_HOST}:{DB_PORT}...")
|
||
|
|
conn = psycopg2.connect(
|
||
|
|
host=DB_HOST,
|
||
|
|
port=DB_PORT,
|
||
|
|
user=DB_USER,
|
||
|
|
password=DB_PASSWORD,
|
||
|
|
database="postgres" # 先连接到默认数据库
|
||
|
|
)
|
||
|
|
conn.set_isolation_level(ISOLATION_LEVEL_AUTOCOMMIT)
|
||
|
|
cursor = conn.cursor()
|
||
|
|
|
||
|
|
# 检查数据库是否存在
|
||
|
|
cursor.execute(f"SELECT 1 FROM pg_database WHERE datname = '{DB_NAME}'")
|
||
|
|
exists = cursor.fetchone()
|
||
|
|
|
||
|
|
if exists:
|
||
|
|
print(f"✅ 数据库 {DB_NAME} 已经存在")
|
||
|
|
else:
|
||
|
|
# 创建数据库
|
||
|
|
print(f"🔧 正在创建数据库 {DB_NAME}...")
|
||
|
|
cursor.execute(f'CREATE DATABASE {DB_NAME}')
|
||
|
|
print(f"✅ 数据库 {DB_NAME} 创建成功！")
|
||
|
|
|
||
|
|
cursor.close()
|
||
|
|
conn.close()
|
||
|
|
|
||
|
|
print(f"\n📊 数据库信息:")
|
||
|
|
print(f" 主机: {DB_HOST}:{DB_PORT}")
|
||
|
|
print(f" 数据库名: {DB_NAME}")
|
||
|
|
print(f" 用户: {DB_USER}")
|
||
|
|
print(f"\n💡 下一步：运行 python init_db.py 创建数据表")
|
||
|
|
|
||
|
|
except psycopg2.Error as e:
|
||
|
|
print(f"❌ 数据库操作失败: {e}")
|
||
|
|
print(f"\n可能的原因:")
|
||
|
|
print(f" 1. 用户 {DB_USER} 没有创建数据库的权限")
|
||
|
|
print(f" 2. 网络连接问题")
|
||
|
|
print(f" 3. 数据库服务器配置限制")
|
||
|
|
except Exception as e:
|
||
|
|
print(f"❌ 发生错误: {e}")
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
create_database()
|