def get_lasts_registers(userid, top):
db = Connection.connect()
return db.registros.find({
"userid": userid
},
sort=[("fecha", pymongo.DESCENDING)
]).limit(top)
def get_sueldo(userid):
db = Connection.connect()
return db.registros.find_one(
{
"userid": userid,
"categoria": ["sueldo"]
},
sort=[("fecha", pymongo.DESCENDING)])
def __generate_id():
db = Connection.connect()
try:
max_id = db.usuarios.find_one({}, {"_id": 1},
sort=[("_id", pymongo.DESCENDING)
])["_id"]
except:
max_id = 0
finally:
return max_id + 1
def find_by_id(userid):
db = Connection.connect()
return db.usuarios.find_one({"_id": userid})
def check_password(pwhash, username):
db = Connection.connect()
return True if db.usuarios.find_one({
"username": username,
"password": pwhash
}) else False
def find_by_prop(key, value):
db = Connection.connect()
return db.usuarios.find({key: value})
def find_by_username(username):
db = Connection.connect()
return db.usuarios.find_one({"username": username})
def crearAlumno(alumno):
db = Connection.connect()
nuevo = db.alumnos
nuevo.insert_one(alumno)
def traerAlumnos():
db = Connection.connect()
lista = db.alumnos.find({})
return lista
def update_registro(registro_id, registro):
db = Connection.connect()
db.registros.update_one({"_id": ObjectId(registro_id)},
{"$set": registro},
upsert=False)
def get_registros(userid):
db = Connection.connect()
return db.registros.find({"userid": userid})
def get_categorias(userid, tipo):
db = Connection.connect()
return db.registros.find({"userid": userid, "tipo": tipo})
def get_montos(userid):
db = Connection.connect()
return db.registros.find({"userid": userid}, {"valor": 1, "tipo": 1})
def find_by_prop(key, value):
db = Connection.connect()
return db.registros.find({key: value})
def find_by_categoria(categoria):
db = Connection.connect()
return db.registros.find_one({"categoria": categoria})
def create_registro(registro):
db = Connection.connect()
db.registros.insert_one(registro)
def create_user(user):
db = Connection.connect()
user._id = UserData.__generate_id()
db.usuarios.insert_one(user.__dict__)
def get_by_id(registro_id):
db = Connection.connect()
try:
return db.registros.find_one({"_id": ObjectId(registro_id)})
except InvalidId:
return None