Python办公自动化之PostgreSQL篇3——利用Python读取一张表写入PostgreSQL

import psycopg2
import pandas as pd
from sqlalchemy import create_engine
 
# 定义连接PostgreSQL的参数
def connect_to_postgresql(user, password, host, database):
    return psycopg2.connect(
        dbname=database,
        user=user,
        password=password,
        host=host
    )
 
# 将DataFrame写入PostgreSQL
def write_to_postgresql(df, conn, table_name):
    # 将DataFrame转换为SQLlite数据库
    engine = create_engine('postgresql://{0}:{1}@{2}/{3}'.format(user, password, host, database))
    df.to_sql(table_name, engine, if_exists='replace', index=False)
 
# 定义用户输入的参数
user = 'your_username'
password = 'your_password'
host = 'your_host'
database = 'your_database'
table_name = 'your_table_name'
 
# 读取Excel文件
df = pd.read_excel('your_excel_file.xlsx')
 
# 连接PostgreSQL
conn = connect_to_postgresql(user, password, host, database)
 
# 将DataFrame写入PostgreSQL
write_to_postgresql(df, conn, table_name)
 
# 关闭连接
conn.close()

这段代码首先定义了连接到PostgreSQL数据库的函数，然后定义了将DataFrame写入PostgreSQL的函数。在主程序中，用户需要提供连接所需的参数，并将Excel文件读取为DataFrame。接下来，程序连接到PostgreSQL，并使用定义好的函数将DataFrame写入指定的表中。最后，关闭数据库连接。这个过程展示了如何将数据从一个地方移动到另一个地方，这是办公自动化的一个基本步骤。