Я новичок в PostgreSQL и psycopg2, и у меня возникла проблема.
import psycopg2
def create_tables(whichone):
in_str_station = "CREATE TABLE station (id SMALLINT NOT NULL PRIMARY KEY, name VARCHAR(40) NOT NULL," \
" country VARCHAR(3) NOT NULL, latitude VARCHAR(10), " \
"longitude VARCHAR(10),height SMALLINT);"
in_str_dailyData = "CREATE TABLE daily_data (id BIGSERIAL NOT NULL PRIMARY KEY," \
"s_id SMALLINT NOT NULL REFERENCES station(id)," \
"d_date DATE NOT NULL, d_mean NUMERIC(6, 1), quality SMALLINT);"
int_str_monthlyMean = "CREATE TABLE monthly_mean (id BIGSERIAL NOT NULL PRIMARY KEY,"\
"s_id SMALLINT NOT NULL REFERENCES station(id),"\
"m_date DATE NOT NULL, m_mean NUMERIC(9, 3),"\
"var NUMERIC(9, 3), std NUMERIC(9, 3));"
in_str_yearlymean = "CREATE TABLE yearly_mean (id BIGSERIAL NOT NULL PRIMARY KEY, " \
"s_id SMALLINT NOT NULL REFERENCES station(id)," \
"y_date DATE NOT NULL, y_mean NUMERIC(9, 3),var NUMERIC(9, 3)," \
"std NUMERIC(9, 3), var_m NUMERIC(9, 3), std_m NUMERIC(9, 3));"
database_list = {'station': in_str_station, 'monthly_mean': int_str_monthlyMean,
'daily_data': in_str_dailyData, 'yearly_mean': in_str_yearlymean}
try:
conn = psycopg2.connect(
host = "localhost",
database = "climate",
user = "postgres",
password = "1")
cur = conn.cursor()
in_str = database_list.get(whichone)
cur.execute(in_str)
output_ = cur.fetchall()
print(output_)
cur.close()
except (Exception, psycopg2.DatabaseError) as error:
print(error)
finally:
if conn is not None:
conn.close()
После того, как я запускаю скрипт, независимо от того, какой из in_str_ я выбираю, таблица не создается. Я проверил, и когда я копирую содержимое in_str, которое я выполнил в cur.execute, и использую его в оболочке PostgreSQL, все работает.
Где я сделал ошибку?
Я не знаю, сделали ли вы это здесь, но это неправильный отступ. Вам нужно сделать отступ кода после функции.
def foo(a):
pass
о, извините, ошибка, когда я вставлял код, везде есть отступы
Звоните conn.commit()
после cur.execute()
, но до conn.close()
. Транзакции не совершаются неявно с помощью psycopg2:
Если соединение закрыто (используя метод close()) или уничтожено (используя del или позволив ему выйти из области действия) во время выполнения транзакции, сервер отменит транзакцию.
Хорошо, спасибо, это работает после того, как я добавил это после cur.execute()
Этот код определяет функцию create_tables(), но не вызывает ее. Если у вас есть другой код, который на самом деле вызывает его, покажите этот код.