worker_id=int(
random.choice(
convert_workers_to_pd_dataframe(
self.query(GlobalWorker)).id)),
location_id=int(
random.choice(
convert_locations_to_pd_dataframe(
self.query(Location)).id))))
self.commit()
def delete_preference(self, worker_id):
preference = self.query(GlobalWorkerPreference).\
filter(GlobalWorkerPreference.worker_id == worker_id)
preference.delete()
self.commit()
if __name__ == '__main__':
testInstance = GlobalWorkerPreferenceController(
DbConnectionAdrianSarajevo())
# prefs = testInstance.get_preferences()
# print(convert_to_pd_dataframe(prefs))
# pref = testInstance.get_preference(1)
# print(pref.worker.name)
# testInstance.add_preference(2, 2)
# Cuts all duplicates
#prefs = testInstance.get_preference(41)
#print(prefs.worker.name)
testInstance.delete_preference(138)
#print(convert_to_pd_dataframe(testInstance.get_preference(6)))
self.session.delete(row)
self.session.commit()
def get_worker(self, worker_id):
return self.session.query(GlobalWorker).get(worker_id)
def add_worker(self, name, surname, birth_date, position_id, location_id, insurance_id):
print("started")
self.session.add(GlobalWorker(id=1, name=name, surname=surname, birth_date=birth_date,
position_id=position_id, location_id=location_id, insurance_id=insurance_id))
self.session.commit()
def delete_worker(self, worker_id):
worker = self.session.query(GlobalWorker).get(worker_id)
self.session.delete(worker)
self.session.commit()
if __name__ == '__main__':
testInstance = GlobalWorkerTest(DbConnectionAdrianSarajevo())
workers = testInstance.get_workers()
print(convert_to_pd_dataframe(workers))
testInstance.add_worker("A","C",date(2000,1,1),3,1,7)
# testInstance.add_worker('Brianek', 'Patusinski', date(2000, 1, 1), 1, 2, 1)
# workers = testInstance.get_workers()
# print(convert_to_pd_dataframe(workers))
# testInstance.delete_worker(2)
# print(convert_to_pd_dataframe(workers))
# def getAllGlobalInsurances(self):
# print(self.session.query(GlobalInsurance.id, GlobalInsurance.expiration_date, Insurer.name).
# filter(GlobalInsurance.insurer_id == Insurer.id).all())
#
# def getAllWorkers(self):
# print(self.session.query(Worker.name, Worker.surname, Worker.birth_date, Location.name,
# GlobalPosition.position, GlobalInsurance.expiration_date, Insurer.name)
# .filter(Worker.location_id == Location.id)
# .filter(Worker.position_id == GlobalPosition.id)
# .filter(Worker.insurance_id == GlobalInsurance.id)
# .filter(GlobalInsurance.insurer_id == Insurer.id)
# .all())
#
# def getAllPrefs(self):
# print(self.session.query(GlobalWorkerPreference.worker.name,
# GlobalWorkerPreference.worker.surname,
# GlobalWorkerPreference.location.name))
#
if __name__ == '__main__':
testInstance = TablesController(DbConnectionAdrianSarajevo())
print(
testInstance.getPropsFor(
GlobalWorker(id=126,
name='Cindy',
surname='Bennet',
position_id=4,
location_id=2)))
# print(testInstance.get_possible_exchanges())
convert_insurances_to_pd_dataframe(
self.query(GlobalInsurance)).id)))
def add_random(self):
for i in range(1, 40):
self.add(self.__generate_random())
self.commit()
def add_worker(self, name, surname, birth_date, position_id, location_id,
insurance_id):
print("started")
self.session.add(
GlobalWorker(id=1,
name=name,
surname=surname,
birth_date=birth_date,
position_id=position_id,
location_id=location_id,
insurance_id=insurance_id))
self.session.commit()
def get_worker(self, worker_id):
return self.query(GlobalWorker).get(worker_id)
if __name__ == '__main__':
testInstance = WorkerController(DbConnectionAdrianSarajevo())
workers = testInstance.get_all()
print(convert_workers_to_pd_dataframe(workers))
worker = testInstance.get_worker(1)
print(worker.name)
from PyQt5 import QtWidgets from orm_controllers.DbConnection import DbConnectionAlbertSarajevo, DbConnectionAdrianSarajevo, DbConnectionAdrianTokyo from ui.main_win import MainWindow import sys app = QtWidgets.QApplication(sys.argv) main_window = QtWidgets.QMainWindow() ui = MainWindow(DbConnectionAdrianSarajevo()) ui.setupUi(main_window) main_window.show() sys.exit(app.exec_())
super().__init__(controller)
def get_all(self):
return self.query(GlobalPosition).order_by(GlobalPosition.position)
#return pd.read_sql_query(positions.statement, con=self.conn)
# def deleteAllPositions(self):
# positions = self.session.query(Position).order_by(Position.id)
# for row in positions:
# self.session.delete(row)
# self.session.commit()
def add_position(self, name):
self.add(GlobalPosition(position=name))
self.commit()
def delete_position(self, name):
position = self.query(GlobalPosition).get(name)
self.delete(position)
self.commit()
if __name__ == '__main__':
testInstance = GlobalPositionController(DbConnectionAdrianSarajevo())
# testInstance.add_position('Test')
positions = convert_positions_to_pd_dataframe(testInstance.get_all())
print(positions.to_string())
testInstance.delete("Waiter")
# testInstance.delete_position('Test')
# testInstance.getPositions()
def init_workers(controller):
if not controller.workers.get_all():
print("workers table is empty: add one manually in one location")
return False
elif controller.workers.get_all().count() == 1:
controller.workers.add_random()
print("workers initialized")
else:
print("workers already initialized")
return True
def init_preferences(controller):
if controller.preferences.get_all().count() < 40:
controller.preferences.add_random()
print("preferences initialized")
else:
print("preferences already initialized")
return True
if __name__ == "__main__":
controller = TablesController(DbConnectionAdrianSarajevo())
# assert init_insurers(controller)
# assert init_locations(controller)
# assert init_insurances(controller)
# assert init_positions(controller)
# assert init_workers(controller)
assert init_preferences(controller)