31 lines
844 B
Python
31 lines
844 B
Python
import psycopg
|
|
from datetime import datetime, timedelta
|
|
|
|
from read_dat import read_binary_file
|
|
|
|
file_path = "lz.out/lz.rivqdown.dat"
|
|
df = read_binary_file(file_path)
|
|
|
|
# PostgreSQL连接配置
|
|
postgres_config = {
|
|
"host": "111.231.71.81",
|
|
"user": "lzflood",
|
|
"password": "lzfloodpassword",
|
|
"dbname": "lzflooddb",
|
|
}
|
|
|
|
postgres_conn = psycopg.connect(**postgres_config)
|
|
postgres_cursor = postgres_conn.cursor()
|
|
|
|
for index, row in df.iterrows():
|
|
current_date = datetime(2010, 7, 20) + timedelta(hours=index)
|
|
for i in range(len(row)):
|
|
postgres_cursor.execute(
|
|
"INSERT INTO river_runoff (river_id, created_at, value) VALUES (%s, %s, %s)",
|
|
(i + 1, current_date, row[i] / 86400),
|
|
)
|
|
print("index, i", index, i)
|
|
postgres_conn.commit()
|
|
postgres_cursor.close()
|
|
postgres_conn.close()
|