def editCustomItem(request, id, token):
db = dataBaseModel()
if not (request.user.is_authenticated() and request.method == 'POST' and db.checkIfOwnCustomProduct(request.user.id, id) == True):
return HttpResponse(json.dumps({'errCode' : dataBaseModel.ERR_BAD_REQUEST}), content_type='application/json')
overlayfilename = ""
imagefilename = ""
config = db.findPositionalConfig(token) # find before clear temp
if 'overlay' in request.FILES:
temppath = db.removeTempProduct(request.user.id) # try remove temp product
# remove temp image
if temppath != []:
for path in temppath:
ImageRW.removeImage(path, False)
ImageRW.removeImage(path.replace('.png', '.jpg'), False)
try:
filename = token_generator()
filename += str(request.user.id)
filename += '.jpg'
overlayfilename = filename.replace('.jpg', 'ol.jpg')
imagefilename = ImageRW.writeImage(request.FILES['overlay'], True, filename) # write image
ImageRW.writeImage(request.FILES['overlay'], True, overlayfilename) # write image
#check here
overlayfilename = ImageRW.Process(overlayfilename, True, request.POST["category"]) # convert it to transparent, return the new ol file name
#check here
ImageRW.removeImage(overlayfilename.replace('.png', '.jpg'), True)
except:
return HttpResponse(json.dumps({'errCode' : dataBaseModel.ERR_UNABLE_TO_EDIT_CUSTOM_PRODUCT}), content_type='application/json')
form = CustomProductForm(request.POST)
if form.is_valid():
pcategory = form.cleaned_data['category'].lower()
pname = form.cleaned_data['itemname'].lower()
pbrand = form.cleaned_data['brand']
purl = form.cleaned_data['url']
pprice = float(form.cleaned_data['price'])
pdescription = form.cleaned_data['description']
result = ""
if config != None:
result = db.editProduct(request.user.id, imagefilename, overlayfilename, pcategory, pbrand, pname, purl, pprice, pdescription, id, config[0], config[1], config[2], config[3], True)
else:
result = db.editProduct(request.user.id, imagefilename, overlayfilename, pcategory, pbrand, pname, purl, pprice, pdescription, id)
if result!= dataBaseModel.SUCCESS:
return HttpResponse(json.dumps({'errCode' : dataBaseModel.ERR_UNABLE_TO_EDIT_CUSTOM_PRODUCT}), content_type='application/json')
data = {'errCode' : dataBaseModel.SUCCESS}
return HttpResponse(json.dumps(data), content_type='application/json')
else:
return HttpResponse(json.dumps({'errCode' : dataBaseModel.ERR_UNABLE_TO_EDIT_CUSTOM_PRODUCT}), content_type='application/json')
def editCustomItem(request, id, token):
db = dataBaseModel()
if not (request.user.is_authenticated() and request.method == 'POST'
and db.checkIfOwnCustomProduct(request.user.id, id) == True):
return HttpResponse(json.dumps(
{'errCode': dataBaseModel.ERR_BAD_REQUEST}),
content_type='application/json')
overlayfilename = ""
imagefilename = ""
config = db.findPositionalConfig(token) # find before clear temp
if 'overlay' in request.FILES:
temppath = db.removeTempProduct(
request.user.id) # try remove temp product
# remove temp image
if temppath != []:
for path in temppath:
ImageRW.removeImage(path, False)
ImageRW.removeImage(path.replace('.png', '.jpg'), False)
try:
filename = token_generator()
filename += str(request.user.id)
filename += '.jpg'
overlayfilename = filename.replace('.jpg', 'ol.jpg')
imagefilename = ImageRW.writeImage(request.FILES['overlay'], True,
filename) # write image
ImageRW.writeImage(request.FILES['overlay'], True,
overlayfilename) # write image
#check here
overlayfilename = ImageRW.Process(
overlayfilename, True, request.POST["category"]
) # convert it to transparent, return the new ol file name
#check here
ImageRW.removeImage(overlayfilename.replace('.png', '.jpg'), True)
except:
return HttpResponse(json.dumps(
{'errCode': dataBaseModel.ERR_UNABLE_TO_EDIT_CUSTOM_PRODUCT}),
content_type='application/json')
form = CustomProductForm(request.POST)
if form.is_valid():
pcategory = form.cleaned_data['category'].lower()
pname = form.cleaned_data['itemname'].lower()
pbrand = form.cleaned_data['brand']
purl = form.cleaned_data['url']
pprice = float(form.cleaned_data['price'])
pdescription = form.cleaned_data['description']
result = ""
if config != None:
result = db.editProduct(request.user.id, imagefilename,
overlayfilename, pcategory, pbrand, pname,
purl, pprice, pdescription, id, config[0],
config[1], config[2], config[3], True)
else:
result = db.editProduct(request.user.id, imagefilename,
overlayfilename, pcategory, pbrand, pname,
purl, pprice, pdescription, id)
if result != dataBaseModel.SUCCESS:
return HttpResponse(json.dumps(
{'errCode': dataBaseModel.ERR_UNABLE_TO_EDIT_CUSTOM_PRODUCT}),
content_type='application/json')
data = {'errCode': dataBaseModel.SUCCESS}
return HttpResponse(json.dumps(data), content_type='application/json')
else:
return HttpResponse(json.dumps(
{'errCode': dataBaseModel.ERR_UNABLE_TO_EDIT_CUSTOM_PRODUCT}),
content_type='application/json')
def addProduct(request, token):
# authenticate
if (not request.user.is_authenticated()) or request.method != "POST":
return HttpResponse(json.dumps({"errCode": dataBaseModel.ERR_BAD_REQUEST}), content_type="application/json")
# delete the item from temp list
# add the product
db = dataBaseModel()
config = db.findPositionalConfig(token) # find it before clear temp
temppath = db.removeTempProduct(request.user.id) # try remove temp product
# remove temp image
if temppath != []:
for path in temppath:
ImageRW.removeImage(path, False)
ImageRW.removeImage(path.replace(".png", ".jpg"), False)
try:
# helper method to generate random token
import string
import random
def token_generator(size=32, chars=string.ascii_lowercase + string.digits):
return "".join(random.choice(chars) for _ in range(size))
filename = token_generator()
filename += str(request.user.id)
filename += ".jpg"
overlayfilename = filename.replace(".jpg", "ol.jpg")
imagefilename = ImageRW.writeImage(request.FILES["overlay"], True, filename) # write image
ImageRW.writeImage(request.FILES["overlay"], True, overlayfilename) # write image
# check here
overlayfilename = ImageRW.Process(
overlayfilename, True, request.POST["category"]
) # convert it to transparent, return the new ol file name
# check here
ImageRW.removeImage(overlayfilename.replace(".png", ".jpg"), True)
form = CustomProductForm(request.POST)
if form.is_valid():
pcategory = form.cleaned_data["category"].lower()
pname = form.cleaned_data["itemname"].lower()
pbrand = form.cleaned_data["brand"]
purl = form.cleaned_data["url"]
pprice = float(form.cleaned_data["price"])
pdescription = form.cleaned_data["description"]
result = ""
if config != None:
# print "run"
result = db.addProduct(
request.user.id,
imagefilename,
overlayfilename,
pcategory,
pbrand,
pname,
purl,
pprice,
pdescription,
config[0],
config[1],
config[2],
config[3],
False,
)
else:
result = db.addProduct(
request.user.id,
imagefilename,
overlayfilename,
pcategory,
pbrand,
pname,
purl,
pprice,
pdescription,
)
if result != dataBaseModel.SUCCESS:
return HttpResponse(
json.dumps({"errCode": dataBaseModel.ERR_UNABLE_TO_ADD_PRODUCT}), content_type="application/json"
)
data = {"errCode": dataBaseModel.SUCCESS}
return HttpResponse(json.dumps(data), content_type="application/json")
else:
raise Exception("")
except:
return HttpResponse(
json.dumps({"errCode": dataBaseModel.ERR_UNABLE_TO_ADD_PRODUCT}), content_type="application/json"
)
def testRemoveNonExistImage(self):
result = ImageRW.removeImage("wrongoneoops.jpg", True, True)
self.assertEquals(result, True, "Incorrect remove image behavior")
def testRemoveImageInTemp(self):
ImageRW.removeImage("sample.png", False)
self.assertTrue(
os.path.isfile(ImageRW.TEMP_DIR + "sample.png") == False,
'Image not removed')
def testRemoveImageInPermanent(self):
ImageRW.removeImage("sample.png", True)
self.assertTrue(
os.path.isfile(ImageRW.IMAGE_DIR + "sample.png") == False,
'Image not removed')
def testRemoveNonExistImage(self):
result = ImageRW.removeImage("wrongoneoops.jpg", True, True)
self.assertEquals(result, True, "Incorrect remove image behavior")
def testRemoveImageInTemp(self):
ImageRW.removeImage("sample.png", False)
self.assertTrue(os.path.isfile(ImageRW.TEMP_DIR + "sample.png") == False, 'Image not removed')
def testRemoveImageInPermanent(self):
ImageRW.removeImage("sample.png", True)
self.assertTrue(os.path.isfile(ImageRW.IMAGE_DIR + "sample.png") == False, 'Image not removed')