def sku_bins_get(id):
resp = Response()
existing = Sku.from_mongodb_doc(db.sku.find_one({"_id": id}))
if not existing:
resp.status_code = 404
resp.mimetype = "application/problem+json"
resp.data = json.dumps({
"type":
"missing-resource",
"title":
"Can not get locations of sku that does not exist.",
"invalid-params": [{
"name": "id",
"reason": "must be an exisiting sku id"
}]
})
return resp
contained_by_bins = [
Bin.from_mongodb_doc(bson)
for bson in db.bin.find({f"contents.{id}": {
"$exists": True
}})
]
locations = {bin.id: {id: bin.contents[id]} for bin in contained_by_bins}
resp.status_code = 200
resp.mimetype = "application/json"
resp.data = json.dumps({"state": locations})
return resp
def bin_delete(id):
existing = Bin.from_mongodb_doc(db.bin.find_one({"_id": id}))
if existing is None:
return problem.missing_bin_response(id)
if request.args.get('force', 'false') == 'true' or len(
existing.contents.keys()) == 0:
db.bin.delete_one({"_id": id})
return success.bin_deleted_response(id)
else:
return problem.dangerous_operation_unforced_response(
"id", "bin must be empty")
def bin_patch(id):
try:
json = bin_patch_schema(request.json)
except MultipleInvalid as e:
return problem.invalid_params_response(e)
existing = Bin.from_mongodb_doc(db.bin.find_one({"_id": id}))
if existing is None:
problem.missing_bin_response(id)
if "props" in json.keys():
db.bin.update_one({"_id": id}, {"$set": {"props": json['props']}})
return BinEndpoint.from_bin(existing).updated_success_response()
def from_id(cls, batch_id, retrieve=False):
if not retrieve:
raise NotImplementedError()
contained_by_bins = [
Bin.from_mongodb_doc(bson)
for bson in db.bin.find({
f"contents.{batch_id}": {"$exists": True}
})]
locations = {bin.id: {batch_id: bin.contents[batch_id]}
for bin in contained_by_bins}
endpoint = BatchBinsEndpoint(
resource_uri=url_for("batch.batch_bins_get", id=batch_id),
state=locations
)
return endpoint
def bin_get(id):
existing = Bin.from_mongodb_doc(db.bin.find_one({"_id": id}))
if existing is None:
return problem.missing_bin_response(id)
else:
return BinEndpoint.from_bin(existing).get_response()
def search():
query = request.args['query']
limit = getIntArgs(request.args, "limit", 20)
startingFrom = getIntArgs(request.args, "startingFrom", 0)
resp = Response()
results = []
# debug flags
if query == '!ALL':
results.extend([Sku.from_mongodb_doc(e) for e in db.sku.find()])
results.extend([Batch.from_mongodb_doc(e) for e in db.batch.find()])
results.extend([Bin.from_mongodb_doc(e) for e in db.bin.find()])
if query == '!BINS':
results.extend([Bin.from_mongodb_doc(e) for e in db.bin.find()])
if query == '!SKUS':
results.extend([Sku.from_mongodb_doc(e) for e in db.sku.find()])
if query == '!BATCHES':
results.extend([Batch.from_mongodb_doc(e) for e in db.batch.find()])
# search by label
if query.startswith('SKU'):
results.append(Sku.from_mongodb_doc(db.sku.find_one({'_id': query})))
if query.startswith('BIN'):
results.append(Bin.from_mongodb_doc(db.bin.find_one({'_id': query})))
if query.startswith('BAT'):
results.append(
Batch.from_mongodb_doc(db.batch.find_one({'_id': query})))
results = [result for result in results if result != None]
# search for skus with owned_codes
cursor = db.sku.find({"owned_codes": query})
for sku_doc in cursor:
results.append(Sku.from_mongodb_doc(sku_doc))
# search for skus with associated codes
cursor = db.sku.find({"associated_codes": query})
for sku_doc in cursor:
results.append(Sku.from_mongodb_doc(sku_doc))
# search for skus with owned_codes
cursor = db.batch.find({"owned_codes": query})
for batch_doc in cursor:
results.append(Batch.from_mongodb_doc(batch_doc))
# search for batchs with associated codes
cursor = db.batch.find({"associated_codes": query})
for batch_doc in cursor:
results.append(Batch.from_mongodb_doc(batch_doc))
# if not DEV_ENV: # maybe use global flag + env variable instead. Shouldn't need to check this every time in production/
if "name_text" in db.sku.index_information().keys():
cursor = db.sku.find({"$text": {"$search": query}})
for sku_doc in cursor:
results.append(Sku.from_mongodb_doc(sku_doc))
if "name_text" in db.batch.index_information().keys():
cursor = db.batch.find({"$text": {"$search": query}})
for batch_doc in cursor:
results.append(Batch.from_mongodb_doc(batch_doc))
if results != []:
paged = results[startingFrom:(startingFrom + limit)]
resp.status_code = 200
resp.mimetype = "application/json"
# TODO: Add next page / prev page operations
resp.data = json.dumps(
{
'state': {
"total_num_results": len(results),
"starting_from": startingFrom,
"limit": limit,
"returned_num_results": len(paged),
"results": paged
},
"operations": []
},
cls=Encoder)
return resp
resp.status_code = 200
resp.mimetype = "application/json"
resp.data = json.dumps(
{
'state': {
"total_num_results": len(results),
"starting_from": startingFrom,
"limit": limit,
"returned_num_results": 0,
"results": []
},
"operations": []
},
cls=Encoder)
return resp
def bin_contents_post(id):
into_bin = Bin.from_mongodb_doc(db.bin.find_one({"_id": id}))
item_id = request.json['id']
quantity = request.json['quantity']
resp = Response()
resp.headers.add("Cache-Control", "no-cache")
if not into_bin:
resp.status_code = 404
resp.mimetype = "application/problem+json"
resp.data = json.dumps({
"type":
"missing-resource",
"title":
"Can not receive items into a bin that does not exist.",
"invalid-params": [{
"name": "id",
"reason": "must be an exisiting bin id"
}]
})
return resp
if item_id.startswith("SKU"):
exisiting_sku = Sku.from_mongodb_doc(db.sku.find_one({"_id": item_id}))
if not exisiting_sku:
resp.status_code = 409
resp.mimetype = "application/problem+json"
resp.data = json.dumps({
"type":
"missing-resource",
"title":
"Can not receive sku that does not exist.",
"invalid-params": [{
"name":
"item_id",
"reason":
"must be an exisiting batch or sku id"
}]
})
return resp
else:
db.bin.update_one({"_id": into_bin.id},
{"$inc": {
f"contents.{item_id}": quantity
}})
resp.status_code = 201
return resp
elif item_id.startswith("BAT"):
existing_batch = Batch.from_mongodb_doc(
db.batch.find_one({"_id": item_id}))
if not existing_batch:
resp.status_code = 409
resp.mimetype = "application/problem+json"
resp.data = json.dumps({
"type":
"missing-resource",
"title":
"Can not receive batch that does not exist.",
"invalid-params": [{
"name":
"item_id",
"reason":
"must be an exisiting batch or sku id"
}]
})
return resp
else:
db.bin.update_one({"_id": into_bin.id},
{"$inc": {
f"contents.{item_id}": quantity
}})
resp.status_code = 201
return resp
else:
resp.status_code = 409
resp.mimetype = "application/problem+json"
resp.data = json.dumps({
"type":
"bad-id-format",
"title":
"Received item id must be a batch or sku.",
"invalid-params": [{
"name": "item_id",
"reason": "must be an exisiting batch or sku id"
}]
})
return resp