예제 #1
0
def get_user_catalogs():
    try:
        req_data = request.get_json()
        user_id = ObjectId(str(req_data['user_id']))
        user = validate_user(user_id)
        if user is None:
            response = {
                "status": "error",
                "message": "Invalid user_id '{}'".format(user_id)
            }
            return json.dumps(response)
        user_catalogs_docs = list(
            USER_CATALOGS_COL.find({"user": user_id}, {"user": 0}))
        response = {"status": "success", "data": user_catalogs_docs}
    except Exception as err:
        response = {"status": "error", "message": str(err)}
    return json.dumps(response, default=str)
예제 #2
0
def get_catalog_details():
    try:
        req_data = request.get_json()
        user_id = ObjectId(str(req_data['user_id']))
        user = validate_user(user_id)
        if user is None:
            response = {
                "status": "error",
                "message": "Invalid user_id '{}'".format(user_id)
            }
            return json.dumps(response)
        user_catalogs = list(
            USER_CATALOGS_COL.find({'user': user['_id']}, {"user": 0}))
        for catalog in user_catalogs:
            catalog["total_files"] = len(
                list(CATALOG_FILES_COL.find({'catalog': catalog['_id']})))
            if 'created_date' in catalog.keys():
                catalog["created_date"] = catalog["created_date"].strftime(
                    "%d-%m-%Y")
        response = {"status": "success", "catalogs": list(user_catalogs)}
    except Exception as err:
        response = {"status": "error", "message": str(err)}
    return json.dumps(response, default=str)