python连接MySQL和postgreSQL,操作数据库,批量向数据库插入数据;

一、使用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()

你可能感兴趣的:(python)