def getEntity(self, paramDict): res_plane = Plane() query = 'SELECT PlaneId, Make, Capacity, Tonnage, TypeId, TTX, DateRelease, DateLastFix, DateLastFlight, Hours, WorkerId ' \ 'FROM Plane ' \ '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_plane.planeId = result[0] res_plane.make = str(result[1]) res_plane.capacity = result[2] res_plane.tonnage = result[3] res_plane.typeId = result[4] res_plane.ttx = str(result[5]) res_plane.dateRelease = result[6] res_plane.dateLastFix = result[7] res_plane.dateLastFlight = result[8] res_plane.hours = result[9] res_plane.workerId = result[10] return res_plane
def getAll(self): planes = [] query = 'SELECT PlaneId, Make, Capacity, Tonnage, TypeId, TTX, DateRelease, DateLastFix, DateLastFlight, Hours, WorkerId ' \ 'FROM Plane' connection = sqlite3.connect(self._dbname) result = connection.execute(query).fetchall() for plane in result: plane = Plane() plane.planeId = user_data[0] plane.make = str(user_data[1]) plane.capacity = user_data[2] plane.tonnage = user_data[3] plane.typeId = user_data[4] plane.ttx = str(user_data[5]) plane.dateRelease = user_data[6] plane.dateLastFix = user_data[7] plane.dateLastFlight = user_data[8] plane.hours = user_data[9] plane.workerId = user_data[10] planes.append(plane) connection.close() return planes