34 lines
890 B
Python
34 lines
890 B
Python
import psycopg2
|
|
import config
|
|
|
|
|
|
|
|
|
|
|
|
def conn():
|
|
try:
|
|
connection = psycopg2.connect(**config.POSTGRES_CONFIG)
|
|
|
|
return connection
|
|
# cursor = connection.cursor()
|
|
|
|
# postgres_insert_query = """ INSERT INTO mobile (ID, MODEL, PRICE) VALUES (%s,%s,%s)"""
|
|
# record_to_insert = (5, 'One Plus 6', 950)
|
|
# cursor.execute(postgres_insert_query, record_to_insert)
|
|
|
|
# connection.commit()
|
|
# count = cursor.rowcount
|
|
# print(count, "Record inserted successfully into mobile table")
|
|
|
|
except (Exception, psycopg2.Error) as error:
|
|
print("Failed to insert record into mobile table", error)
|
|
return False
|
|
|
|
# finally:
|
|
# # closing database connection.
|
|
# if connection:
|
|
# connection.cursor().close()
|
|
# connection.close()
|
|
# print("PostgreSQL connection is closed")
|
|
|