1 # Author:Byron Li
  2 #-*-coding:utf-8-*-
  3 
  4 '''----------------------------------------------使用文件说明----------------------------------------------------------
  5 使用文件说明
  6 userlist.txt         存放用户账户信息文件，包括用户名、密码、登陆次数和余额
  7 ***_cost_record.txt  存放某用户***消费记录的文件，用户首次购买商品后创建，没有购买过商品的用户不会产生该文件
  8 ---------------------------------------------------------------------------------------------------------------------'''
  9 import os
 10 import datetime
 11 
 12 def login(name,password):   #用户登陆,用户名和密码验证，登陆成功则返回登陆次数
 13     with open('userlist.txt', 'r+',encoding='UTF-8') as f:
 14         line = f.readline()
 15         while(line):
 16             pos=f.tell()
 17             line=f.readline()
 18             if [name,password] == line.split()[0:2]:
 19                 times=int(line.split()[2])
 20                 line=line.replace(str(times).center(5,' '),str(times+1).center(5,' '))
 21                 f.seek(pos)
 22                 f.write(line)
 23                 return times+1
 24     return None
 25 
 26 def get_balance(name):   #获取用户余额数据
 27     with open('userlist.txt', 'r',encoding='UTF-8') as f:
 28         line = f.readline()
 29         for line in f:
 30             if name == line.split()[0]:
 31                 return line.split()[3]
 32     print("用户%s不存在，无法获取其余额信息！"%name)
 33     return False
 34 
 35 def update_balance(name,balance):    #更新用户余额数据
 36     with open('userlist.txt', 'r+',encoding='UTF-8') as f:
 37         line = f.readline()
 38         while(line):
 39             pos1=f.tell()
 40             line=f.readline()
 41             if name == line.split()[0]:
 42                 pos1=pos1+line.find(line.split()[2].center(5,' '))+5
 43                 pos2=f.tell()
 44                 f.seek(pos1)
 45                 f.write(str(balance).rjust(pos2-pos1-2,' '))
 46                 return True
 47     print("用户%s不存在，无法更新其余额信息！" % name)
 48     return False
 49 
 50 def inquire_cost_record(name):     #查询用户历史消费记录
 51     if os.path.isfile(''.join([name,'_cost_record.txt'])):
 52         with open(''.join([name,'_cost_record.txt']), 'r',encoding='UTF-8') as f:
 53             print("历史消费记录".center(40, '='))
 54             print(f.read())
 55             print("".center(46, '='))
 56             return True
 57     else:
 58         print("您还没有任何历史消费记录！")
 59         return False
 60 
 61 def update_cost_record(name,shopping_list):   #更新用户消费记录
 62     if len(shopping_list)>0:
 63         if not os.path.isfile(''.join([name, '_cost_record.txt'])):     #第一次创建时第一行标上“商品  价格”
 64             with open(''.join([name, '_cost_record.txt']), 'a',encoding='UTF-8') as f:
 65                 f.write("%-5s%+20s\n" % ('商品', '价格'))
 66                 f.write(''.join([datetime.datetime.now().strftime('%c'), ' 消费记录']).center(40,'-'))   #写入消费时间信息方便后续查询
 67                 f.write('\n')
 68                 for product in shopping_list:
 69                     f.write("%-5s%+20s\n"%(product[0],str(product[1])))
 70         else:
 71             with open(''.join([name, '_cost_record.txt']), 'a',encoding='UTF-8') as f:
 72                 f.write(''.join([datetime.datetime.now().strftime('%c'), ' 消费记录']).center(40, '-'))
 73                 f.write('\n')
 74                 for product in shopping_list:
 75                     f.write("%-5s%+20s\n"%(product[0],str(product[1])))
 76         return True
 77     else:
 78         print("您本次没有购买商品，不更新消费记录！")
 79         return False
 80 
 81 def shopping_chart():    #主函数，用户交互，函数调用，结果输出
 82     product_list=[
 83         ('Iphone',5000),
 84         ('自行车',600),
 85         ('联想电脑',7800),
 86         ('衬衫',350),
 87         ('洗衣机',1000),
 88         ('矿泉水',3),
 89         ('手表',12000)
 90     ]   #商店商品列表
 91     shopping_list=[]   #用户本次购买商品列表
 92     while(True):
 93         username = input("请输入用户名：")
 94         password = input("请输入密码：")
 95         login_times=login(username,password)   #查询输入用户名和密码是否正确,正确则返回登陆次数
 96         if login_times:
 97             print('欢迎%s第%d次登陆!'.center(50,'*')%(username,login_times))
 98             if login_times==1:
 99                 balance = input("请输入工资：")   #第一次登陆输入账户资金
100                 while(True):
101                     if balance.isdigit():
102                         balance=int(balance)
103                         break
104                     else:
105                         balance = input("输入工资有误，请重新输入：")
106             else:
107                 balance=int(get_balance(username))  #非第一次登陆从文件获取账户余额
108             while(True):
109                 print("请选择您要查询消费记录还是购买商品：")
110                 print("[0] 查询消费记录")
111                 print("[1] 购买商品")
112                 choice=input(">>>")
113                 if choice.isdigit():
114                     if int(choice)==0:                 #查询历史消费记录
115                         inquire_cost_record(username)
116                     elif int(choice)==1:               #购买商品
117                         while (True):
118                             for index,item in enumerate(product_list):
119                                 print(index,item)
120                             choice=input("请输入商品编号购买商品:")
121                             if choice.isdigit():
122                                 if int(choice)>=0 and int(choice)<len(product_list):
123                                     if int(product_list[int(choice)][1])<balance:   #检查余额是否充足，充足则商品购买成功
124                                         shopping_list.append(product_list[int(choice)])
125                                         balance = balance - int(product_list[int(choice)][1])
126                                         print("\033[31;1m%s\033[0m已加入购物车中，您的当前余额是\033[31;1m%s元\033[0m" %(product_list[int(choice)][0],balance))
127                                     else:
128                                         print("\033[41;1m您的余额只剩%s元，无法购买%s！\033[0m" %(balance,product_list[int(choice)][0]))
129                                 else:
130                                     print("输入编号错误，请重新输入!")
131                             elif choice=='q':      #退出账号登陆，退出前打印本次购买清单和余额信息，并更新到文件
132                                 if len(shopping_list)>0:
133                                     print("本次购买商品清单".center(50,'-'))
134                                     for product in shopping_list:
135                                         print("%-5s%+20s"%(product[0],str(product[1])))
136                                     print("".center(50, '-'))
137                                     print("您的余额：\033[31;1m%s元\033[0m"%balance)
138                                     update_cost_record(username,shopping_list)
139                                     update_balance(username, balance)
140                                     print("退出登陆！".center(50, '*'))
141                                     exit()
142                                 else:
143                                     print("您本次没有消费记录，欢迎下次购买！")
144                                     print("退出登陆！".center(50, '*'))
145                                     exit()
146                             else:
147                                 print("选项输入错误，请重新输入！")
148                     else:
149                         print("选项输入错误，请重新输入！")
150                 elif choice=='q':   #退出账号登陆
151                     print("退出登陆！".center(50, '*'))
152                     exit()
153                 else:
154                     print("选项输入错误，请重新输入！")
155             break
156         else:
157             print('用户名或密码错误,请重新输入!')
158 
159 shopping_chart() #主程序运行