def getEntity(self, paramDict): res_worker = Worker() query = 'SELECT WorkerId, PIB, Age, Gender, Adress, Phone, Passport, PositionId ' \ 'FROM Workers ' \ 'WHERE ' equal_substr = '{attr_name} = ?' counter = len(paramDict) args = [] for param in paramDict: query += equal_substr.format(attr_name=param) args.append(paramDict[param]) if counter == 1: query += ';' else: query += ' AND ' counter -= 1 connection = sqlite3.connect(self._dbname) result = connection.execute(query, args).fetchone() connection.close() res_worker.workerId = result[0] res_worker.pib = result[1] res_worker.age = result[2] res_worker.gender = result[3] res_worker.address = result[4] res_worker.phone = result[5] res_worker.passport = result[6] res_worker.positionId = result[7] return res_worker
def getAll(self): workers = [] query = 'SELECT WorkerId, PIB, Age, Gender, Adress, Phone, Passport, PositionId ' \ 'FROM Workers' connection = sqlite3.connect(self._dbname) result = connection.execute(query).fetchall() for worker in result: worker = Worker() worker.workerId = user_data[0] worker.pib = user_data[1] worker.age = user_data[2] worker.gender = user_data[3] worker.address = user_data[4] worker.phone = user_data[5] worker.passport = user_data[6] worker.positionId = user_data[7] workers.append(worker) connection.close() return workers