def getCosechas(self, productor): db = Database() cursor = db.main() cosechas = [] try: data = cursor.cosecha.find({"productor": productor}) for i in data: ventas = [] id = i["_id"] ventasJSON = cursor.ventas.find({"cosecha": ObjectId(id)}) for ven in ventasJSON: v = Venta() v.fecha = ven["fecha"] v.cantidad = ven["cantidad"] v.monto = ven["monto"] ventas.append(v) nc = Cosecha(id, i["cereal"], i["cantidadProduccion"], i["cantidadParcial"], i["inicio"], i["fin"], i["productor"], ventas) nc.id = id productorJSON = cursor.usuario.find_one({"user": productor}) productorObject = User() productorObject.parse(productorJSON) nc.productor = productorObject cosechas.append(nc) return cosechas except NameError: print(NameError) return None
def getUser(self, user, password): #Hacer try catch, que pasa si no encuentra un usuario? db = Database() cursor = db.main() logUserJSON = cursor.usuario.find_one({"user": user, "password": password}) if logUserJSON != None: logUserModel = User() logUserModel.parse(logUserJSON) return logUserModel else: return None