Python编程_Lesson014_json和pickle模块讲解

json和pickle模块的使用

poedu_shop.py

import PoeduShop.user_operator

user_info_fp = "all_user_info.pickle"

def main():
username = input("username:")
password = input("password:")
if PoeduShop.user_operator.user_login(username, password, user_info_fp):
print("login success")
else:
print("login failed!")
while True:
print("""
Please input:
y: register a new user
n: do not register
""")
user_input = input()
if user_input == "y":
username = input("username:")
while username == "":
username = input("username:")
password = input("password:")
while password == "":
password = input("password:")
salary = input("salary:")
while salary == "":
salary = input("salary:")
balance = input("balance:")
while balance == "":
balance = input("balance:")
if PoeduShop.user_operator.user_register(username, password, salary, balance, user_info_fp):
print("register success!")
break
else:
print("register failed!")
elif user_input == "n":
break

if __name__ == "__main__":
main()
else:
print("Please Start The Progress from poedu_shop.py module!")

file_operator.py

import os.path

def __get_file_suffix(fp):
"""
__get_file_suffix(fp)

Get file suffix
:param fp: file path
:return:   return file suffix
"""
return os.path.splitext(fp)[1]

def __get_module_info(fp):
"""
__get_module_info(fp)

Get file format
:param fp: file path
:return:   if the suffix either equals ".json" or equals ".pickle" return module info which
includes module name, open and write mode, else return None
"""
module_info = None
suffix = __get_file_suffix(fp)
if suffix == ".json":
module_info = ["json", "r", "w"]
elif suffix == ".pickle":
module_info = ["pickle", "rb", "wb"]

return module_info

def dump_file(list_data, fp):
"""
dump_file(dict_data, fp)

dump the dict_data to file with fp
:param list_data: the data storage in the form of list
:param fp: file path
:return:   if storage success return True, else return False
"""
return_value = True
module_info = __get_module_info(fp)
if module_info is None:
return_value = False
print("An unsupported file format")
else:
current_module = __import__(module_info[0])
with open(fp, module_info[2]) as file:
current_module.dump(list_data, file)

return return_value

def load_file(fp):
user_info_dict = None
module_info = __get_module_info(fp)
if module_info is None:
print("An unsupported file format")
else:
current_module = __import__(module_info[0])
try:
with open(fp, module_info[1]) as file:
user_info_dict = current_module.load(file)
except Exception as e:
print(e)

return user_info_dict

user_operator.py

import PoeduShop.file_operator

all_user_info_list = []

def user_register(username, password, salary, balance, fp):
"""
user_register(username, password, salary, balance)

register a new user with user name, password, monthly salary and user balance.
:param username:  User Name
:param password:  Pass Word
:param salary:    Monthly Salary
:param balance:   User Balance
:param fp:        file path
:return:          return True if register success, else return False!
"""
all_user_info_list.append({"username": username, "password": password, "salary": salary, "balance": balance})
return PoeduShop.file_operator.dump_file(all_user_info_list, fp)

def user_login(username, password, fp):
login_status = False
user_info_list = PoeduShop.file_operator.load_file(fp)
if user_info_list is not None:
all_user_info_list.clear()
for user_info in user_info_list:
all_user_info_list.append(user_info)
if username == user_info["username"] and password == user_info["password"]:
login_status = True
break

return login_status