def delete_items(tablename, ids, idname, idtype):
if ids:
cur = Connector.conn.cursor()
ids = Connector.delete_formatter(ids, idname, idtype)
try:
cur.execute("DELETE FROM " + tablename + " WHERE " + ids + ";")
except psycopg2.Error as err:
global error_window
error_window = ErrorPopUp(ErrorFormatter.get_error(err.pgcode))
error_window.show()
Connector.conn.commit()
def get_cur_date():
cur = Connector.conn.cursor()
try:
cur.execute("SELECT current_date;")
except psycopg2.Error as err:
global error_window
error_window = ErrorPopUp(ErrorFormatter.get_error(err.pgcode))
error_window.show()
res = cur.fetchone()
cur.close()
return res[0]
def get_table_data(tablename, columns):
cur = Connector.conn.cursor()
columns = Connector.column_formatter(columns)
try:
cur.execute("SELECT " + columns + " FROM " + tablename + ";")
except psycopg2.Error as err:
global error_window
error_window = ErrorPopUp(ErrorFormatter.get_error(err.pgcode))
error_window.show()
res = cur.fetchall()
cur.close()
return res
def insert_row(tablename, columns, values):
correct = True
cur = Connector.conn.cursor()
columns = Connector.column_formatter(columns)
values = Connector.value_formatter(values)
try:
cur.execute("INSERT INTO " + tablename + " (" + columns + ") " +
"VALUES (" + values + ");")
except psycopg2.Error as err:
correct = False
global error_window
error_window = ErrorPopUp(ErrorFormatter.get_error(err.pgcode))
error_window.show()
cur.close()
Connector.conn.commit()
return correct
def get_record(tablename, columns, id, idname, idtype):
cur = Connector.conn.cursor()
columns = Connector.column_formatter(columns)
try:
if idtype is int:
cur.execute("SELECT " + columns + " FROM " + tablename +
" WHERE " + idname + " = " + str(id) + ";")
else:
cur.execute("SELECT " + columns + " FROM " + tablename +
" WHERE " + idname + " LIKE '" + id + "';")
except psycopg2.Error as err:
global error_window
error_window = ErrorPopUp(ErrorFormatter.get_error(err.pgcode))
error_window.show()
res = cur.fetchone()
cur.close()
return res
def create_vehicle(input):
correct = True
for i, x in enumerate(input):
if x == "":
input[i] = None
cur = Connector.conn.cursor()
try:
cur.execute(
"CALL public.create_vehicle(%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s);",
input)
except psycopg2.Error as err:
correct = False
global error_window
error_window = ErrorPopUp(ErrorFormatter.get_error(err.pgcode))
error_window.show()
cur.close
Connector.conn.commit()
return correct
def update_row(tablename, columns, values, id, idname, idtype):
correct = True
cur = Connector.conn.cursor()
data = Connector.update_formatter(columns, values)
try:
if idtype is int:
cur.execute("UPDATE " + tablename + " SET " + data +
" WHERE " + idname + " = " + str(id) + ";")
else:
cur.execute("UPDATE " + tablename + " SET " + data +
" WHERE " + idname + " LIKE '" + id + "';")
except psycopg2.Error as err:
correct = False
global error_window
error_window = ErrorPopUp(ErrorFormatter.get_error(err.pgcode))
error_window.show()
cur.close()
Connector.conn.commit()
return correct
def create_zamowienie_pojazd(input):
res = None
for i, x in enumerate(input):
if x == "":
input[i] = None
cur = Connector.conn.cursor()
if input[0] is not None:
input[0] = input[0].replace(",", ".")
input[0] = float(input[0])
try:
cur.execute("CALL public.create_zamowienie_pojazd(%s, %s, %s);",
input)
cur.execute("SELECT currval('zamowienie_poj_id_sequence')")
res = cur.fetchone()
except psycopg2.Error as err:
global error_window
error_window = ErrorPopUp(ErrorFormatter.get_error(err.pgcode))
error_window.show()
cur.close
Connector.conn.commit()
if res is not None:
return res[0]
return None