87 lines
2.2 KiB
Python
87 lines
2.2 KiB
Python
#!/usr/bin/python
|
|
import psycopg2
|
|
|
|
def connect_db(db_settings):
|
|
conn = None
|
|
print("Connecting to the PostgreSQL database...")
|
|
try:
|
|
conn = psycopg2.connect(**db_settings)
|
|
print("Connection success")
|
|
except (Exception, psycopg2.DatabaseError) as error:
|
|
print(error)
|
|
return conn
|
|
|
|
def add_item(db_settings, itemid, skuid, choice, attributes):
|
|
connection = connect_db(db_settings)
|
|
cursor = connection.cursor()
|
|
|
|
cursor.execute("""
|
|
INSERT INTO item (itemid, skuid, choice, attributes)
|
|
VALUES (%s, %s, %s, %s)
|
|
""", (itemid, skuid, choice, attributes))
|
|
connection.commit()
|
|
connection.close()
|
|
|
|
def add_history_entry(db_settings, itemid, skuid, choice, attributes, price, currency, quantity, discount_percentage):
|
|
connection = connect_db(db_settings)
|
|
cursor = connection.cursor()
|
|
|
|
cursor.execute("""
|
|
SELECT *
|
|
FROM item
|
|
WHERE itemid = %s
|
|
AND skuid = %s
|
|
""", (itemid, skuid))
|
|
|
|
if cursor.rowcount == 0:
|
|
add_item(db_settings, itemid, skuid, choice, attributes)
|
|
|
|
cursor.execute("""
|
|
INSERT INTO history (itemid, skuid, price, currency, quantity, discount_percentage, h_timestamp)
|
|
VALUES (%s, %s, %s, %s, %s, %s, (SELECT LOCALTIMESTAMP))
|
|
""", (itemid, skuid, price, currency, quantity, discount_percentage))
|
|
connection.commit()
|
|
connection.close()
|
|
|
|
|
|
|
|
|
|
def initialize(db_settings):
|
|
connection = connect_db(db_settings)
|
|
cursor = connection.cursor()
|
|
|
|
cursor.execute("""
|
|
DROP TABLE IF EXISTS history
|
|
""")
|
|
cursor.execute("""
|
|
DROP TABLE IF EXISTS item
|
|
""")
|
|
cursor.execute("""
|
|
CREATE TABLE item
|
|
(
|
|
itemid bigint,
|
|
skuid bigint,
|
|
choice boolean,
|
|
attributes text[],
|
|
primary key (itemid,skuid)
|
|
)
|
|
""")
|
|
cursor.execute("""
|
|
CREATE TABLE history
|
|
(
|
|
itemid bigint,
|
|
skuid bigint,
|
|
quantity integer,
|
|
discount_percentage numeric(2),
|
|
price money,
|
|
currency varchar(4),
|
|
h_timestamp timestamp,
|
|
foreign key (itemid,skuid) references item(itemid,skuid),
|
|
primary key (itemid,skuid,h_timestamp)
|
|
)
|
|
""")
|
|
|
|
connection.commit()
|
|
connection.close()
|
|
print("Database initialized")
|