#!/usr/bin/env python3
# Author: Robert
# --*-- coding: utf-8 --*--

set = False  #設置set 當輸入為q就可以退出
file = open("購物車用戶信息檔案.txt","r+",encoding="utf-8")  #讀取購物車用戶信息文件
f = str(file.read())  #將文件內容轉化成字符串
for line in f:
  file_str = str(f)
data = eval(file_str)  #將字符串轉換為字典data

name = input("輸入姓名：")
password = input("輸入密碼：")
while True:
  if name in data:      #用戶在檔案中
    if password in data[name]:     #密碼和用戶名對應，校驗正確，登錄。
      salary = float(data[name][password])
      print('''\033[32;1m歡迎登錄，當前余額為%s\033[0m'''%salary)
      break
    else:      #否則密碼錯誤，請重新輸入
      password = input("密碼錯誤，請重新輸入：")
      continue
  else:      #否則判斷為首次登錄，將用戶名，密碼，工資存到用戶信息文件中
    password_salary = {}
    salary_str = input("歡迎首次登錄，請輸入你的工資：")
    salary = float(salary_str)
    password_salary[password] = salary     #工資對應到密碼
    data[name] = password_salary      #將密碼-工資對應到用戶名
    file.seek(0)
    file.write(str(data))
    file.tell()
    break

list = [#購物清單
  ["iphone",5800],
  ["sifei",800],
  ["macbook",17500],
  ["book",75],
  ["apple",5]
]

file_list_r = open("歷史購物信息.txt","r+",encoding="utf-8")
file_list_r = str(file_list_r.read())
shoppinglist_dict = eval(file_list_r)
if name not in shoppinglist_dict:
  shoppinglist_dict[name] = []
shoppinglist = shoppinglist_dict[name]
shoppinglist_dict_now = []
choose = input("\n是否需要查詢歷史購物記錄(y/n)：")
if choose == 'y':
  print("\n\n---------->歷史購物記錄<----------")
  print(shoppinglist)
  print("---------->結束<----------")

while not set:   #購物車開始
  print("---------->商品清單<----------")
  for index,item in enumerate(list,1):
    print(index,item)
  print("---------->結束<----------")
  number = input("請輸入想購買商品編號：")
  if number == "q":
    set = True
    data[name][password] = str(salary)
    file.seek(0)
    file.write(str(data))
    file.tell()
    print("---------->購物清單<----------")
    print(shoppinglist)
    print("您的余額：",salary)
    print("---------->結束<----------")
    shoppinglist.extend(shoppinglist)
    shoppinglist_dict[name] = shoppinglist
  elif number.isdigit() == False:
    print("\033[31;1m輸入不是編號內容，請重新輸入\033[0m")
  elif int(number)>int(len(list)) or int(number)<= 0:     #輸入值不在清單中，報錯
    print("\033[31;1m您所購買的商品不在清單中\033[0m")
  else:
    number_buy = int(number)-1
    if list[number_buy][1]<(salary):      #如果余額足夠，提示購買成功并顯示余額。
      salary = salary - int(list[number_buy][1])
      msg = '\033[32;1m您已經將%s加入購物車中，余額為%d\033[0m'%(list[number_buy][0],salary)
      print(msg)
      shoppinglist.append(list[number_buy])      #將本次購物信息加到購買記錄中
    else:
      print("\033[31;1m余額不足，無法購買!\033[0m")      #提示余額不足