问题描述
我对python还是很陌生,但是我创建了一个小脚本来在postgres中插入行,以及一个用于存储股价的表格。 该表对日期和行情自动收录器有唯一的约束,以确保每个行情自动售票器每天只插入一个价格。我希望脚本跳过插入并继续执行其余的操作,但是我无法弄清楚在触发异常块时如何使循环继续。
Python脚本如下:
def createConnection(db="db",user="john",password='doe',host='host',port=5432):
conn = psycopg2.connect(
database=db,user=user,password=password,host=host,port=port)
return conn
def insertNewPrices(df):
conn = createConnection()
cur = conn.cursor()
for row in df.head().itertuples(index=False):
try:
print(row)
cur.execute(
"INSERT INTO daily_price (price_date,ticker,close_price) VALUES (%s,%s,%s)",row)
except psycopg2.IntegrityError as e:
print("something went wrong")
print(e)
continue
conn.commit()
conn.close()
引发错误:
psycopg2.errors.UniqueViolation: duplicate key value violates unique constraint "daily_price_ticker_price_date_key"
DETAIL: Key (ticker,price_date)=(EQUInor,1990-02-28) already exists.
解决方法
在try语句之外有insert语句。您可以从外面取下插件吗?应该可以。
for row in df.head().itertuples(index=False):
#cur.execute(
# "INSERT INTO daily_price (price_date,ticker,close_price) VALUES (%s,%s,%s)",row)
try:
print(row)
cur.execute(
"INSERT INTO daily_price (price_date,row)
except psycopg2.IntegrityError,psycopg2.errors.UniqueViolation) as e:
print("something went wrong")
print(e)
此外,您无需在except语句的末尾作为最后一行继续。
除了检查特定错误外,您还可以使用以下方法捕获所有错误和警告:
except (psycopg2.Error,psycopg2.Warning) as e: