def update_lottery_client(id):
found_lottery = lottery.objects(id=ObjectId(id)).first()
if found_lottery != None:
form = request.form
print(form['gender'])
if form['gender'] == "0":
gender = "male"
else:
gender = "female"
found_lottery.gender = gender
found_lottery.name = form['name']
found_lottery.phone = form['phone']
found_lottery.update(name=form['name'],
phone=form['phone'],
gender=gender)
all_lottery = lottery.objects()
lottery_list = []
for i in range(all_lottery.count()):
lottery_list.append(all_lottery[i])
res = result(True, lottery_list, None)
return res.convert_to_json()
else:
res = result(False, None, "ID Error or People Not Found")
return res.convert_to_json(), 400
def delete_member(id):
found_member = member.objects(id=ObjectId(id)).first()
if found_member != None:
found_member.update(is_active=False)
res = result(True, "success", None)
return res.convert_to_json()
else:
res = result(False, None, "ID Error or People Not Found")
return res.convert_to_json(), 400
def delete_client_data(id):
found_client = local_client.objects(id=ObjectId(id)).first()
if found_client != None:
found_client.update(is_active=False)
res = result(True, "success", None)
return res.convert_to_json()
else:
res = result(False, None, "ID Error or People Not Found")
return res.convert_to_json(), 400
def delete_lottery_client(id):
found_lottery = lottery.objects(id=ObjectId(id)).first()
if found_lottery != None:
found_lottery.delete()
all_lottery = lottery.objects()
lottery_list = []
for i in range(all_lottery.count()):
lottery_list.append(all_lottery[i])
res = result(True, lottery_list, None)
return res.convert_to_json()
else:
res = result(False, None, "ID Error or People Not Found")
return res.convert_to_json(), 400
def delete_one_staff(id):
found_staff = staff.objects(id=ObjectId(id)).first()
if found_staff != None:
found_staff.delete()
staffs = staff.objects()
staff_list = []
for i in range(staffs.count()):
staff_list.append(staffs[i])
res = result(True, staff_list, None)
return res.convert_to_json()
else:
res = result(False, None, "ID Error or People Not Found")
return res.convert_to_json(), 400
def update_client_data(id):
found_client = local_client.objects(id=ObjectId(id)).first()
if found_client != None:
form = request.form
found_client.update(name=form['name'],
phone=form['phone'],
address=form['address'],
email=form['email'])
res = result(True, "success", None)
return res.convert_to_json()
else:
res = result(False, None, "ID Error or People Not Found")
return res.convert_to_json(), 400
def delete_brand_data(id):
found_brand = brand.objects(id=ObjectId(id)).first()
if found_brand != None:
found_brand.delete()
all_brand = brand.objects()
brand_list = []
for i in range(all_brand.count()):
brand_list.append(all_brand[i])
res = result(True, brand_list, None)
return res.convert_to_json()
else:
res = result(False, None, "ID Error or People Not Found")
return res.convert_to_json(), 400
def delete_contact_form(id):
found_client = web_client.objects(id=ObjectId(id)).first()
if found_client != None:
found_client.delete()
all_client = web_client.objects()
client_list = []
for i in range(all_client.count()):
client_list.append(all_client[i])
res = result(True, client_list, None)
return res.convert_to_json()
else:
res = result(False, None, "ID Error or People Not Found")
return res.convert_to_json(), 400
def update_staff_data(id):
found_staff = staff.objects(id=ObjectId(id)).first()
if found_staff != None:
form = request.form
found_staff.update(role=form['role'], role_cn=form['role_cn'],facebook=form['facebook'],linkedin=form['linkedin'],wechat=form['wechat'],image=form['image'])
staffs = staff.objects()
staff_list = []
for i in range(staffs.count()):
staff_list.append(staffs[i])
res = result(True, staff_list, None)
return res.convert_to_json()
else:
res = result(False, None, "ID Error or People Not Found")
return res.convert_to_json(), 400
def get_all_products():
lotterys = lottery.objects()
lottery_list = []
for i in range(lotterys.count()):
lottery_list.append(lotterys[i])
res = result(True, lottery_list, None)
return res.convert_to_json()
def get_staff_data():
staffs = staff.objects()
staff_list = []
for i in range(staffs.count()):
staff_list.append(staffs[i])
res = result(True, staff_list, None)
return res.convert_to_json()
def get_brand_data():
brands = brand.objects()
brand_list = []
for i in range(brands.count()):
brand_list.append(brands[i])
res = result(True, brand_list, None)
return res.convert_to_json()
def get_contact_form():
clients = web_client.objects()
client_list = []
for i in range(clients.count()):
client_list.append(clients[i])
res = result(True, client_list, None)
return res.convert_to_json()
def get_all_products():
products = product.objects()
products_list = []
for i in range(products.count()):
products_list.append(products[i])
res = result(True, products_list, None)
return res.convert_to_json()
def get_product_by_searched_word(text):
products = product.objects(code__contains=text)
products_list = []
for i in range(products.count()):
products_list.append(products[i])
res = result(True, products_list, None)
return res.convert_to_json()
def change_password(user_id):
form = request.form
old_password = form['old_password']
new_password = form['new_password']
found_user = user.objects(id=ObjectId(user_id)).first()
old_password_hash = md5(config['password']['private_key'].encode("utf-8"))
old_password_hash.update(old_password.encode('utf-8'))
if old_password_hash.hexdigest() != found_user['password']:
return result(False, None,
"The old password is incorrect.").convert_to_json(), 400
new_password_hash = md5(config['password']['private_key'].encode("utf-8"))
new_password_hash.update(new_password.encode('utf-8'))
found_user.update(password=new_password_hash.hexdigest())
return result(True, "Password changed", None).convert_to_json()
def get_product_category():
product_categorys = product_category.objects()
product_category_list = []
for i in range(product_categorys.count()):
product_category_list.append(product_categorys[i])
res = result(True, product_category_list, None)
return res.convert_to_json()
def get_client_data():
clients = local_client.objects(is_active=True)
client_list = []
for i in range(clients.count()):
client_list.append(clients[i])
res = result(True, client_list, None)
return res.convert_to_json()
def get_client_data_by_page_num(page):
page = int(page)
items_per_page = 10
offset = (page - 1) * items_per_page
clients = local_client.objects(
is_active=True).skip(offset).limit(items_per_page).all()
client_list = []
i = 0
try:
while i < 10:
client_list.append(clients[i])
i += 1
except:
None
pag = {
"data": client_list,
"total": local_client.objects(is_active=True).count()
}
res = result(True, pag, None)
return res.convert_to_json()
def get_product_by_id(product_category_id):
products = product.objects(
product_category_id=ObjectId(product_category_id))
products_list = []
for i in range(products.count()):
products_list.append(products[i])
res = result(True, products_list, None)
return res.convert_to_json()
def add_member():
try:
form = request.form
end_date = datetime.datetime.utcnow() + datetime.timedelta(days=365)
new_member = member(point=0,member_level=0,name=form['name'],email=form['email'],address=form['address'],
phone=form['phone'],company=form['company'],is_active=True,end_date=end_date,total_consumption=0)
new_member.save()
res = result(True,"success",None)
return res.convert_to_json()
except NotUniqueError:
res = result(False,None,'Phone Number exists.')
return res.convert_to_json(), 400
except:
res = result(False, None, 'Server Error')
return res.convert_to_json(), 500
def index(id):
found_transaction = transaction.objects(member_id=ObjectId(id)).order_by('-date_time')
found_transaction_list = []
for i in range(found_transaction.count()):
found_transaction_list.append(found_transaction[i])
res = result(True,found_transaction_list,None)
return res.convert_to_json()
def delete_one_product(id):
found_product = product.objects(id=ObjectId(id)).first()
if found_product != None:
product_category_id = ObjectId(found_product['product_category_id'])
found_product.delete()
products = product.objects(product_category_id=product_category_id)
products_list = []
for i in range(products.count()):
products_list.append(products[i])
res = result(True, products_list, None)
return res.convert_to_json()
else:
res = result(False, None, "ID Error or People Not Found")
return res.convert_to_json(), 400
def login():
form = request.form
pipeline = [{
'$match': {
"username": form['username']
}
}, {
'$lookup': {
"from": "staff",
"localField": "staff_id",
"foreignField": "_id",
"as": "staff_detail"
}
}, {
"$unwind": "$staff_detail"
}]
found_user_list = list(user.objects.aggregate(*pipeline))
found_user = found_user_list[0] if len(found_user_list) != 0 else None
if not found_user:
res = result(False, None, "User Not Found")
return res.convert_to_json(), 404
else:
hash = md5(config['password']['private_key'].encode("utf-8"))
hash.update(form['password'].encode('utf-8'))
password = hash.hexdigest()
if found_user['password'] == password:
token = user.objects().first().generate_auth_token()
data = {
'token': token.decode('ascii'),
'user_name': found_user['staff_detail']['name'],
'user_id': str(found_user['_id'])
}
login_log(user_name=data['user_name'],
user_id=data['user_id'],
date_time=datetime.datetime.utcnow() +
datetime.timedelta(hours=13)).save()
res = result(True, data, None)
return res.convert_to_json()
else:
res = result(False, None, "Password Incorrect")
return res.convert_to_json(), 403
def upload_image():
files = request.files
file = files['image']
storage_client = storage.Client.from_service_account_json(private_key)
bucket = storage_client.get_bucket("midyear-pattern-267020.appspot.com")
blob = bucket.blob("images/"+str(int(time()))+".jpg") #文件名
blob.upload_from_string(file.read()) #文件
blob.make_public()
return result(True,blob.public_url,None).convert_to_json()
def edit_member(id):
form = request.form
found_member = member.objects(id=ObjectId(id)).first()
if found_member != None:
try:
found_member.update(name=form['name'],email=form['email'],address=form['address'],phone=form['phone'],company=form['company'])
res = result(True, "success", None)
return res.convert_to_json()
except NotUniqueError:
res = result(False, None, 'Phone Number exists.')
return res.convert_to_json(), 400
except:
res = result(False, None, 'Server Error')
return res.convert_to_json(), 500
else:
res = result(False, None, "ID Error or People Not Found")
return res.convert_to_json(), 400
def update_brand_data(id):
found_brand = brand.objects(id=ObjectId(id)).first()
if found_brand != None:
form = request.form
found_brand.update(name=form['name'],
name_cn=form['name_cn'],
description=form['description'],
description_cn=form['description_cn'],
image=form['image'])
brands = brand.objects()
brand_list = []
for i in range(brands.count()):
brand_list.append(brands[i])
res = result(True, brand_list, None)
return res.convert_to_json()
else:
res = result(False, None, "ID Error or People Not Found")
return res.convert_to_json(), 400
def spend(id,amount,used_point):
found_member = member.objects(id=ObjectId(id)).first()
if found_member != None:
used_point = int(float(used_point))
amount = Decimal(amount)
update_total_consumption = found_member['total_consumption']+ amount - used_point
if update_total_consumption < 18000:
update_member_level = 0
elif update_total_consumption < 38000:
update_member_level = 1
elif update_total_consumption < 68000:
update_member_level = 2
else:
update_member_level = 3
if found_member['member_level']==0:
update_point = int(int(amount) / 100)
elif found_member['member_level']==1:
update_point = int((int(amount) / 100)* 1.2)
elif found_member['member_level']==2:
update_point = int((int(amount) / 100)* 1.3)
else:
update_point = int((int(amount) / 100)* 1.5)
final_point = update_point + int(found_member['point']) - used_point
found_member.update(point=final_point,total_consumption=update_total_consumption,member_level=update_member_level) #更新积分
transaction(member_id=ObjectId(id),date_time=datetime.datetime.utcnow(),
amount=amount,pay_amount= amount - used_point,used_point=used_point,before_point=found_member['point'],after_point=final_point
).save()
res = result(True, "success", None)
return res.convert_to_json()
else:
res = result(False, None, "ID Error or People Not Found")
return res.convert_to_json(), 400
def update_product_category(id):
found_product_category = product_category.objects(id=ObjectId(id)).first()
if found_product_category != None:
form = request.form
found_product_category.update(name=form['name'],
name_cn=form['name_cn'],
description=form['description'],
description_cn=form['description_cn'],
image=form['image'])
product_categories = product_category.objects()
product_category_list = []
for i in range(product_categories.count()):
product_category_list.append(product_categories[i])
res = result(True, product_category_list, None)
return res.convert_to_json()
else:
res = result(False, None, "ID Error or People Not Found")
return res.convert_to_json(), 400
def add_client():
form = request.form
new_client = local_client(name=form['name'],
phone=form['phone'],
address=form['address'],
email=form['email'],
is_active=True)
new_client.save()
res = result(True, "success", None)
return res.convert_to_json()
