def find(self, product_id):
c = self._conn.cursor()
c.execute(
"""
SELECT id, name FROM Products WHERE id = ?
""", [product_id])
return DTO.Product(*c.fetchone())
def create_tables(self, path):
self._conn.executescript("""
CREATE TABLE Employees(
id INTEGER PRIMARY KEY,
name TEXT NOT NULL,
salary REAL NOT NULL,
coffee_stand INTEGER REFERENCES Coffee_stands(id)
);
CREATE TABLE Suppliers(
id INTEGER PRIMARY KEY,
name TEXT NOT NULL,
contact_information TEXT
);
CREATE TABLE Products(
id INTEGER PRIMARY KEY,
description TEXT NOT NULL,
price REAL NOT NULL,
quantity INTEGER NOT NULL
);
CREATE TABLE Coffee_stands(
id INTEGER PRIMARY KEY,
location TEXT NOT NULL,
number_of_employees INTEGER
);
CREATE TABLE Activities(
product_id INTEGER REFERENCES Products(id),
quantity INTEGER NOT NULL,
activator_id INTEGER NOT NULL,
date DATE NOT NULL
);
""")
text = open(path).readlines()
for line in text:
line = line.split(', ')
for element in range(0, line.__len__()):
line[element] = str(line[element]).replace("\n", '')
if (line[0] == "C"):
coffee_stand = DTO.Coffee_stand(int(line[1]), line[2],
int(line[3]))
self.coffee_stands.insert(coffee_stand)
elif (line[0] == "S"):
supplier = DTO.Supplier(int(line[1]), line[2], line[3])
self.suppliers.insert(supplier)
elif (line[0] == "E"):
employee = DTO.Employee(int(line[1]), line[2], float(line[3]),
int(line[4]))
self.employees.insert(employee)
elif (line[0] == "P"):
product = DTO.Product(int(line[1]), line[2], float(line[3]), 0)
self.products.insert(product)
def enter(entry, repo):
if entry[0] == 'C':
repo.coffeeStands.insert(DTO.CoffeeStand(entry[1], entry[2], entry[3]))
elif entry[0] == 'E':
repo.employees.insert(
DTO.Employee(entry[1], entry[2], entry[3], entry[4]))
elif entry[0] == 'P':
repo.products.insert(DTO.Product(entry[1], entry[2], entry[3]))
elif entry[0] == 'S':
repo.suppliers.insert(DTO.Supplier(entry[1], entry[2], entry[3]))