python基础day2作业:购物车

#功能:
1.可注册账号
2.登录买家账号
3.可查询编辑购物车里商品
4.可以余额充值
5.可提示余额不足
6.购物车结算

#使用:
1.第一次使用先注册账号填写账号金额
2.账号金额信息保存在buyer_account.txt (默认为空)
3.只针对帐号文件里的用户进行判断并锁定，针对用户和密码各有三次错误重试机会。

流程图:



代码:

#!/usr/bin/env python3

#antuor:Alan

from prettytable import PrettyTable
import sys,os
import getpass,time

#用户注册函数#
def register():
new_username = input('请输入注册用户名:')
new_password = input('请输入新用户密码:')
bank_money = 0

while True:
new_password_again = input('请再次确认新用户密码:')    ###再次确认密码
if new_password_again != new_password:                    ###密码错误则循环
continue
else:
break                                         ###正确则跳出循环执行下一个循环
while True:
bank_money = input('请输入你要充值的金额:')          ###输入金额数字
if bank_money.isdigit():                         ###判断金额是否为数字
bank_money = int(bank_money)                  ###金额数字整数化
break
else:                                             ###如果金额非数字则不断循环
print('请输入正确的金额')

user_account = "%s %s %d" % (new_username,new_password,bank_money)
# file = open('login_account.txt','a+')
# save_file = file.write(user_account)
# file.close()

# print (("++++++++++恭喜你注册成功:%s++++++++++++") % (user_account))

with open('login_account.txt','a+') as file:          ###把用户信息存入文档,a+为读写追加模式
file.write(user_account+'\n')                     ###写入并换行
print (("==============恭喜你注册成功:%s==============") % (user_account))
print("================请重新登录你的账号======================")
############################################################################

def digit_input(s):
if str.isdigit(s):                                                     #对输入是否是数字进行判断
s = int(s)                                                         #如果输出的是个数字，则转化为整数类型
return s                                                               #返回输入字符

#用户信息函数#
def shopping_func(user='',init_money='',now_money='',recharge_money='',value=''):
os.system('clear')                                                     #清屏###
init_money = int(init_money)                                           #初始总金额数字化
now_money= int(now_money)                                              #当前余额数字化
recharge_money = int(recharge_money)                                   #充值金额数字化

###########################通过prettytable模块将输出内容如表格方式整齐#########
product_page = PrettyTable(['用户名','金额','当前余额','充值余额','购物车'])
product_page.align['用户名'] = '1'                                      #对齐
product_page.add_row([user,init_money,now_money,recharge_money,value])
print(product_page)

#商品页面展示函数#
def products_page(my_dict):

items_dict = {}                                                        #定义函数内部字典
############对商品列表进行遍历并加上数字编号###########################
i = 1
print('商品列表 : ')
print(30*'===')
print('%-5s \t %-20s     \t%s  \t%-10s    \t %s' % ('编号','商品名称','商品价格(元)','商品总数量(个)','商品剩余数量(个)'))
for k in my_dict:
v = my_dict[k]
if type(v) is dict:
print('%-5d  %-30s  %-10d  %-10d  %-40d' % (i,k,v['price'],v['sum'],v['num']))
items_dict[i] = [k,v['price'],v['num'],v['sum']]               #将商品列表赋值到items_dict
i += 1
print(30*'===')

return items_dict                                                      #返回格式化后的字典

########################显示购物车商品列表函数#############################
def shopping_cart(show_dict):
show_all_sum = 0                                                       #初始化购物车商品总价
show_all_num = 0                                                       #初始化购物车商品数量
############对商品列表进行遍历并加上数字编号###########################
print('%s ,%s ,%-10s,%s,%s,%s' % ('编号','商品名称','商品价格(元)','商品总数量(个)','购买数量(个)','购买金额（元）'))
for k in show_dict:
v = show_dict[k]
if type(v) is list:
print('%-5s \t %-20s \t %-10d \t %-10d \t %-10d \t %-10d' % (k,v[0],v[1],v[2],v[3],v[4]))
show_all_sum += v[4]                                           #商品总价累计
show_all_num += 1                                              #商品个数累计
print('请确认您购买商品，总金额：%d（元）'.center(84,'#') % show_all_sum)
return (show_all_sum,show_all_num)                                     #返回商品总价和商品个数

#编辑购物车商品函数#
def shopping_cart_edit(modify_dict,modify_goods_dict):

a_flag = 1                                                             #初始化第一层编号循环标记
while a_flag:
index = input('请输入商品编号 | 完成修改(q) : ')
if len(index) != 0:                                                #商品编号不为空进行字符转化
index = digit_input(index)                                    #调用函数(自建)
if index == 'q':                                                   #输入为q,即完成修改退出
break
elif index in modify_dict:                                         #输入为正确编号###
b_flag = 1                                                     #初始化第二层数量循环标记
name = modify_dict[index][0]                                   #对name赋值商品名称
while b_flag:
num = input('请输入新的商品数量(最大值为%d) |  完成修改(q) : ' % modify_dict[index][2] )
if len(num) != 0:                                          #商品数量不为空进行字符转化###
num = digit_input(num)
if num == 'q':                                             #输入为q,即完成修改退出###
break
elif num == 0:                                             #输入为0,即删除字典中的这个条目
modify_goods_dict[name]['num'] = modify_dict[index][2] #商品列表中的商品数量变为初始值
del modify_dict[index]                                 #购物车删除这个商品
b_flag = 0
elif num > 0 and num <= modify_dict[index][2]:             #输入为要求数字,则进行数量和金额修改
modify_dict[index][3] = num
modify_dict[index][4] = modify_dict[index][1] * num
modify_goods_dict[name]['num'] = modify_dict[index][2] - num    #更新商品列表中的商品数量
b_flag = 0
else:
pass
else:
pass
return (modify_dict,modify_goods_dict)                                 ###返回修改后的购物车商品列表###

#购物车展示函数#
def shopping_cart_show(my_cart,my_goods_dict):

print('欢迎来到您的购物车'.center(80,'#'))
goods_all_sum = 0                                                      ###初始化购物车商品总金额###
goods_all_num = 0                                                      ###初始化购物车商品总数量###

if my_cart:                                                            ###购物车参数不为空###
###########调用购物车商品列表函数,并返回商品总金额和总数量#########
(goods_all_sum,goods_all_num)  = shopping_cart(my_cart)

choice = input('请进行如下操作 : 修改记录(c) | 继续购物(!c)')
if choice == 'c':                                                  ###对购物车商品列表进行修改###
(my_shop_cart,my_goods_dict) = shopping_cart_edit(my_cart,my_goods_dict)  ###调用商品修改函数，并返回新的商品列表###
(goods_all_sum,goods_all_num) = shopping_cart(my_shop_cart)  ###购物车商品展示，并返回最新的商品总价###
else:
pass
else:
print('当前您的购物车为空'.center(80,' '))

time.sleep(2)
return (goods_all_sum,goods_all_num,my_goods_dict)                      #购物车中商品总价、数量和用户商品列表#

#余额充值函#
def balance_recharge(recharge_init_balance,recharge_now_balance,recharge_money):
recharge_flag = 1                                                       #充值循环参数初始化###
while recharge_flag:
recharge_num = input('请输入充值金额 | 返回(b) | 退出(q):')
if len(recharge_num) != 0:                                          #如果输入非空，对输入进行判断并转化类型
recharge_num = digit_input(recharge_num)
if recharge_num == 'q':                                             #如果输入为q,则退出程序
sys.exit(0)
elif recharge_num == 'b':                                           #如果输入为b,则返回第一层循环，重新选择商品编号
break
elif type(recharge_num) is int and recharge_num > 0 :               #输入要求充值金额
recharge_init_balance += recharge_num                           #初始金额增加
recharge_now_balance += recharge_num                            #当前余额增加
recharge_money += recharge_num                                  #充值金额增加
recharge_flag = 0                                               #改变充值循环参数
print('充值成功，请查收'.center(80,' '))                          #提示充值成功
else:
pass
return (recharge_init_balance,recharge_now_balance,recharge_money)      #返回初始、当前、充值金额

#用户结帐函数#
def pending_order(billing_list,my_cart,billing_balance):

print('欢迎来到结算菜单'.center(80,'#'))
if my_cart:                                                             #购物车参数不为空
#############调用购物车商品列表函数################################
shopping_cart(my_cart)
billing_flag = input('请确认是否商品结算（y | n）：')
if billing_flag == 'y':                                             #商品结算确认
billing_file = open('buyer_account','w')                             #打开读写帐号文件
for user_info in billing_list:
billing_file.writelines(user_info)                          #回写用户列表信息
billing_file.close()                                            #关闭帐号文件
sys.exit('结帐成功，你当前余额 ：%d'.center(80,' ') % billing_balance)
else:
print('退出结算菜单，继续购物'.center(80,' '))
time.sleep(2)
else:
print('当前您的购物车为空，无需结算'.center(80,' '))
time.sleep(2)

################################主程序开始##################################

################################商品列表####################################

goods_list = {
'iphone6s': {'price':6000,'num':10,'sum':10},     #num为当前库存，sum为商品总数量
'ipad pro': {'price':3000,'num':20,'sum':20},
'Bike': {'price':2000,'num':15,'sum':15},
'DLS Camera': {'price':1999,'num':8,'sum':8},
'macbook pro': {'price':14000,'num':5,'sum':5},
'macbook air': {'price':7000,'num':10,'sum':10},
'Fountain Pen':{'price':15,'num':50,'sum':50},

}

######用户注册,写入login_account.txt文件里######################

ask_question = input('是否已经有账号? y|n:')                                 ###是否有账号,是则登录,不是则注册

while True:

if ask_question == 'y':
break
else:
register()                                                         ###调用登录函数
break

##############################################################
i = 0
while i < 3:                                                                ###只要用户登录异常不超过3次就不断循环###
username = input('请输入用户名:')                                      ###输入用户名###
password = input('请输入密码:')                                        ###输入隐藏密码###
user_file = open('login_account.txt','r')                              ###打开帐号文件###
#next(user_file)
user_list = user_file.readlines()                                     ###用户列表信息###
user_file.close()                                                     ###关闭帐号文件###

for user_line in user_list:                                        #对帐号文件进行遍历,因为只有三个用户,所以只读前3行
######################分别获取帐号、密码信息和当前余额##############
(user,passwd,init_balance) = user_line.strip('\n').split()
init_balance = int(init_balance)                                    ###对总金额进行数字化###
now_balance = init_balance                                          ###对当前余额进行数字化###
my_goods_sum = 0                                                    ###初始化购买商品总金额###

if username == user and password == passwd:                         ###如用户名和密码正常匹配###
user_shopping_cart = {}                                         ###初始化用户购物车字典###
user_shopping_cart_count = 0                                    ###初始化用户购物车内商品的数量###
recharge_value = 0                                              ###初始化充值金额###
line_num = user_list.index(user_line)                           ###赋值匹配用户的下标###
first_flag = 1                                                  ###定义第一层循环变量参数###

while first_flag:

########################调用框架函数输出用户信息############
shopping_func(username,init_balance,now_balance,recharge_value,user_shopping_cart_count)

goods_output_dict = products_page(goods_list)             ###调用商品列表展示函数输出商品信息###
goods_index = input('请选择菜单 ：输入商品编号 | 购物车(c) | 余额充值(r) | 结帐(b) | 退出(q) : ')

if len(goods_index) != 0:                                   ###如果输入非空，对输入进行判断并转化类型###
goods_index = digit_input(goods_index)

if goods_index == 'q':                                      ###如果输入为q,则退出程序###
sys.exit(0)

elif goods_index == 'c':                                    ###如果输入为c，则展示购物车###
###调用购物车显示函数，并返回购物车商品总金额###########
(my_goods_sum,user_shopping_cart_count,goods_list) = shopping_cart_show(user_shopping_cart,goods_list)
now_balance = init_balance - my_goods_sum               ###计算出当前余额###
if now_balance < 0:
print('您的余额不足，请及时充值，谢谢')
time.sleep(2)

elif goods_index == 'r':                                    ###如果输入为r，则进行余额充值###
(init_balance,now_balance,recharge_value) = balance_recharge(init_balance,now_balance,recharge_value)

elif goods_index == 'b':                                    ###如果输入为b,则进入结账菜单###
##############更新用户文件中用户的余额##################
user_list[line_num] = user + ' ' + passwd + ' ' + repr(now_balance) + '\n'
pending_order(user_list,user_shopping_cart,now_balance)  ###调用结算函数###

elif goods_index in goods_output_dict:                      ###输入编号为正确的商品编号###

###############################取出goods_index商品列表信息并进行赋值和展示############
(goods_name,goods_price,goods_num) = (goods_output_dict[goods_index][0], goods_output_dict[goods_index][1], goods_output_dict[goods_index][2])
print('【 编号：%-5d \t 名称：%-15s \t 价格：%-5d（元） \t 数量：%-5d（个）】' % (goods_index,goods_name,goods_price,goods_num))

second_flag = 1                                         ###定义第二层循环变量参数###
while second_flag:
buy_num = input('请输入购买商品个数(最大值为%d) | 返回(b) | 退出(q): ' % goods_num)
if len(buy_num) != 0:                               ###如果输入非空，对输入进行判断并转化类型###
buy_num = digit_input(buy_num)
if buy_num == 'q':                                  ###如果输入为q,则退出程序###
sys.exit(0)
elif buy_num == 'b':                                ###如果输入为b,则返回第一层循环，重新选择商品编号###
break
elif type(buy_num) is int and buy_num > 0 and buy_num <= goods_num:          ###输入要求商品数量###
my_goods_sum = goods_price * buy_num            ###计算购买商品的总金额###
if my_goods_sum <= now_balance:
print('购买商品 %s 总价格为 : %d' % (goods_name,my_goods_sum))
add_flag = input('请确认是否加入购物车（y | n）：')
if add_flag == 'y':                         ###购买商品确认加入购物车###

###判断购物车不存在该商品###############
if goods_index not in user_shopping_cart:
user_shopping_cart_count += 1       ###购物车里商品数量加1###
####将该商品加入用户购物车字典中####
user_shopping_cart[goods_index] = [goods_name,goods_price,goods_num,buy_num,my_goods_sum]
####购物车已经存在该商品，则进行数量和金额累计计算############
else:
user_shopping_cart[goods_index][3] += buy_num
user_shopping_cart[goods_index][4] += my_goods_sum

now_balance -= my_goods_sum             ###计算出当前余额###

goods_list[goods_name]['num'] -= buy_num   ###商品列表的商品数量更新###
second_flag = 0                         ###设置第二层循环值为0，结束第二层循环####

else:
break
else:
print('您的余额不足，请充值或重新选择，谢谢')
time.sleep(2)
else:
pass
else:
pass
else:
if i != 2:                                                         ###i=2时，最后一次机会，不在提示还剩余0次机会了##
print('用户或密码错误，请重新输入，还有 %d 次机会' % (2 - i))
i += 1                                                                 ###当用户输入错误时，循环值增加1###
else:
sys.exit('用户或密码输入错误超过三次，退出系统,欢迎下次光临')          #三次错误后，异常退出###