一、使用python连接postgreSQL
import pprint
import random
from random import Random
from faker import Faker
import radar
import psycopg2 # 导入pgSQL驱动
conn = psycopg2.connect(database="数据库名",
user="数据库账号",
password="数据库密码",
host="xx.xx.xx.xx",
port="端口号")
cursor = conn.cursor()
value = ''
for i in range(1,20):
fake = Faker("zh_CN") # zhel使用了一个fake库来创建测试数据
value+=f"({i-1},'{fake.name()}','{random.randint(123456,987654)}'),"
value = value[:-1]
insert_into_sql = f'insert into test values {value}'
print(insert_into_sql)
cursor.execute("select * from test;")
result = cursor.fetchall()
pprint.pprint(result)
try:
cursor.execute(insert_into_sql)
conn.commit()
except Exception as e:
conn.rollback()
cursor.close()
conn.close()
二、使用python连接MySQL(与pgsql类似,需要修改下驱动)
import pprint
import pymysql
connect = pymysql.connect(host='xx.xx.xx.xx',
port=xxx,
user='xxx',
password='xxxx',
database='xxx',
charset='xxx'
)
insert_into_sql = """
insert into apps(column_name,url,country)values('商店','http://a.com','CN'),('水果','http://shui.com','CN');
"""
cursor = connect.cursor()
cursor.execute("select * from apps")
result = cursor.fetchall()
pprint.pprint(result)
try:
cursor.execute(insert_into_sql)
connect.commit()
except Exception as e:
connect.rollback()
cursor.close()
connect.close()