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
