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
def createUser(self, name, surname, user, password, wallet):
db = Database()
cursor = db.main()
newUser = {
'name': name,
'surname': surname,
'user': user,
'password': password,
'wallet': wallet
}
try:
cursor.usuario.insert_one(newUser)
return True
except:
return False
def getVentas(self, cosecha):
db = Database()
cursor = db.main()
try:
ventasJSON = cursor.ventas.find({"cosecha": cosecha})
ventas = []
for v in ventasJSON:
ven = Venta()
ven.cantidad = v['cantidad']
ven.monto = v['monto']
ven.fecha = v['fecha']
ventas.append(ven)
return ventas
except NameError:
print(NameError)
return None
def addVenta(self, cosecha, precioUnitario, cantidad):
db = Database()
cursor = db.main()
monto = precioUnitario*cantidad
try:
newVenta = {
'cosecha': cosecha,
'fecha': date.today().strftime("%d/%m/%y"),
'cantidad': cantidad,
'monto': monto
}
cursor.ventas.insert_one(newVenta)
#print('Venta Agregada')
return True
except NameError:
print(NameError)
return False
def addCosecha(self, cosecha):
db = Database()
cursor = db.main()
newCosecha = {
'id': cosecha.id,
'cereal': cosecha.cereal,
'cantidadProduccion': cosecha.cantidadProduccion,
'cantidadParcial': cosecha.cantidadProduccion,
'inicio': cosecha.inicio,
'fin': cosecha.fin,
'productor': cosecha.productor
}
print(newCosecha)
try:
cursor.cosecha.insert_one(newCosecha)
print("OK")
return True
except:
print("wrong")
return False
def getLast(self):
db = Database()
cursor = db.main()
r = cursor.cosecha.find()
return r