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
