Python设计超市收银描述流程图 模拟商场收银程序python

思路：ATM是一个单独程序，提供给消费的是一个接口core下的settlement.py，只做了个人的，没写管理的模块

　　　Shopping也是一个单独的，只做了一个购物的消费模块，没写商家模块，偷懒用了银行的数据库，用户名和密码都是用的一套的

具体目录如下：

atm.py:

import os
import sys
base_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
# print(base_dir)
sys.path.append(base_dir)

from core import main

if __name__ == '__main__':
 main.run()

settings.py:
import os
import sys
import logging
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

DATABASE = {
'engine': 'file_storage',
'name':'accounts',
'path':"%s\\db" % BASE_DIR
}

LOG_LEVEL = logging.INFO
LOG_TYPES = {
'transaction':'transactions.log',
'access':'access.log',
}

TRANSACTION_TYPE = {
'repay':{'action':'plus','interest':0},
'withdraw':{'action':'minus','interest':0.05},
'transfer':{'action':'minus','interest':0.05},
'consume':{'action':'minus','interest':0},
}

 accounts.py:

import json,os,sys,time

BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(BASE_DIR)
from core import db_handler
from conf import settings

def load_current_balance(account_id):
"""

 :param account_id:
 :return:返回账号的资金和其它信息
 """
 db_path = db_handler.db_handler(settings.DATABASE)
 account_file = "%s\%s.json" %(db_path,account_id)
with open(account_file) as f:
 acc_data = json.load(f)
return acc_data
def dump_account(account_data):
 db_path = db_handler.db_handler(settings.DATABASE)
 account_file = "%s\%s.json" %(db_path,account_data['id'])
with open(account_file,'w') as f:
acc_data = json.dump(account_data,f)

return True

 auth.py:

import os
import sys
import json
import time
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(BASE_DIR)
from core import db_handler
from conf import settings
from core import logger

def acc_auth(account,password):
"""

 :param account: 信用卡账号
 :param password: 信用卡密码
 :return:
 """
 db_path = db_handler.db_handler(settings.DATABASE) #数据库存储的目录
 account_file = "%s\%s.json" %(db_path,account) #个人的json数据
 # print(account_file)
 if os.path.isfile(account_file): #还必须判断这个文件是否存在
 with open(account_file) as f:
 account_data = json.load(f) #还原成字典
 if account_data['password'] == password:
#mktime() 接收struct_time对象作为参数，返回用秒数来表示时间的浮点数，
 # time.strptime(string[, format])根据指定的格式把一个时间字符串解析为时间元组
 exp_time_stamp = time.mktime(time.strptime(account_data['expire_date'],"%Y-%m-%d"))
if time.time() > exp_time_stamp:
print("\033[31;1m你的账号[%s]已经过期，请联系银行服务中心\033[0m"%account)
else: #如果通过了，就返回账号的所有信息
 return account_data
else:
print("\033[31;1m你的账号或者密码错误\033[0m")

else:
print("\033[31;1m账号[%s]不存在！\033[0m"% account)

def acc_login(user_data,log_obj):
"""
 登录接口
 :param user_data:user_data在主函数定义是临时账户信息
 :param log_obj:log_obj执行完logger()之后返回的logger，可以调用相关方法写入日志信息
 :return:
 """
 retry_count = 0
 while user_data['is_authenticated'] is not True and retry_count < 3:
 account = input("请输入账号：").strip()
 password = input("请输入密码：").strip()
 auth = acc_auth(account,password)
if auth: #不是空代表验证成功了，因为验证成功之后会返回用户所有信息
 user_data['is_authenticated'] = True
 user_data['account_id'] = account
return auth #又返回了用户的全部数据
 retry_count +=1
 else:
 log_obj.error("account[%s] too many login attempts" %account)
exit()

db_handler.py:
"""
根据数据库的类型，对数据进行处理
"""
import os
def file_db_handle(conn_params):
"""
 数据库的存放的路径
 :param conn_params: 传入的数据是settings中的DATABASE
 :return:
 """
 l = os.sep
# print('file db:',conn_params)
 db_path = '%s\%s' %(conn_params['path'],conn_params['name'])
return db_path

def db_handler(conn_parms):
"""
 根据不同的类型，用不同的函数进行处理
 :param conn_parms:
 :return:
 """
 if conn_parms['engine'] == 'file_storage':
return file_db_handle(conn_parms)
elif conn_parms['engine'] == 'mysql':
pass

logger.py:

import logging
import os,sys
#想要从父类导入，需要先加入环境变量
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(BASE_DIR)
from conf import settings

def logger(log_type):

#create logger
 logger = logging.getLogger(log_type)
 logger.setLevel(settings.LOG_LEVEL)
#create file handler and set level to warning,这是会输出到屏幕上的
 ch = logging.StreamHandler()
 ch.setLevel(settings.LOG_LEVEL)
#创建输入到文件夹中的日志
 log_file = "%s/log/%s" %(settings.BASE_DIR,settings.LOG_TYPES[log_type])
 fh = logging.FileHandler(log_file)
 fh.setLevel(settings.LOG_LEVEL)
 formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
#在日志中加入格式
 # ch.setFormatter(formatter)
 fh.setFormatter(formatter)
#将内容输入到创建的日志文件中
 # logger.addHandler(ch)
 logger.addHandler(fh)

return logger #这可以后边可以调用，直接写入日志

# #对logger进行测试
# a = logger("access")
# a.warn("test")

main.py:

import os
import sys,time

BARE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(BARE_DIR)
from core import auth
from core import logger
from core import transaction
from core import accounts
from conf import settings

#logger()函数执行，并且返回了logger，可以进行调用warn方法进行日志写入
trans_logger = logger.logger('transaction')
access_logger = logger.logger('access')

user_data = {
'account_id':None,
'is_authenticated':False,
'account_data':None
}
def account_info(acc_data):
print(user_data['account_data'])
def repay(acc_data):
 account_data = user_data['account_data']
# for k,v in account_data.items():
 # print(k,v)
 current_balance = """
 ----------- BALANCE INFO --------
 Credit:%s
 Balance:%s
 """ %(account_data['credit'],account_data['balance'])
print(current_balance)
 back_flag = False #设置标志，输入b跳出while循环
 while not back_flag:
 repay_amount = input("\033[33;1mInput repay amount:\033[0m").strip()
if len(repay_amount) > 0 and repay_amount.isdigit():
# print('ddd 00')
 new_balance = transaction.make_transaction(trans_logger,account_data,'repay',repay_amount)
# print(new_balance)
 if new_balance:
print("""\033[42;1mNew Balance:%s\033[0m"""%(new_balance['balance']))
else:
print('\033[31;1m[%s] is not a valid amount,only accept integer!\033[0m' % repay_amount)
if repay_amount == 'b':
 back_flag = True

def withdraw(acc_data):
"""
 取款
 :param acc_data: 变成了数据库数据,而且是字典的格式
 :return:
 """
 accout_data = accounts.load_current_balance(acc_data['account_id']) #再去数据库拿一遍数据，有点多余
 current_balance = '''
 ------- BALANCE INFO -------
 Credit: %s
 Balance: %s
 '''%(accout_data['credit'],accout_data['balance'])
print(current_balance)
 back_flag = False
 while not back_flag:
 withdraw_amount = input("\033[33;1mInput withdraw amount:\033[0m").strip()
if len(withdraw_amount) > 0 and withdraw_amount.isdigit():
 new_balance = transaction.make_transaction(trans_logger,accout_data,'withdraw',withdraw_amount)
if new_balance:
print('''\033[42;1mNew Balance:%s\033[0m'''%(new_balance['balance']))
else:
print('\033[31;1m[%s] is not a valid amount,only accept integer!')
if withdraw_amount == 'b':
 back_flag = True

def transfer(acc_data):
"""

 :param acc_data: 变成了数据库数据,而且是字典的格式
 :return:
 """
 account_data = accounts.load_current_balance(acc_data['account_id'])
 current_balance = '''
 ------- BALANCE INFO -------
 Credit: %s
 Balance: %s'''%(account_data['credit'],account_data['balance'])
print(current_balance)

#钱数必须是数字
 back_flag = False
 while not back_flag:
 tranfer_amount = input("\033[33;1mInput tranfer amount:\033[0m").strip()
#只支持整数
 if len(tranfer_amount) > 0 and tranfer_amount.isdigit():

#调用transaction，传入账号信息，交易类型，交易金额，交易日志
 new_account_data = transaction.make_transaction(trans_logger,account_data,'transfer',tranfer_amount)
if new_account_data:
print("\033[42;1mNew Balance:%s\033[0m"%new_account_data['balance'])
else:
print('\033[31;1m[%s] is not a valid amount,only accept integer!')
if tranfer_amount == 'b':
 back_flag = True

def pay_check(acc_data): #账单功能
 back_flag = False
 while not back_flag:
 Time_input = input("\033[33;1mplease input time(Y-M-D),输入b退出:\033[0m").strip()
if Time_input == 'b': back_flag = True continue else: try: select_time = time.mktime(time.strptime(Time_input,'%Y-%m-%d')) # print(select_time) except Exception as ValueError: print("清输入正确的日期格式") continue # log_file = "atm_test/log/transactions.log" log_file = "%s/log/%s" % (settings.BASE_DIR, settings.LOG_TYPES["transaction"]) # print(log_file) with open(log_file,"r",encoding="utf-8") as f: for i in f: # for i in f : 是和大文件读写，不用写f,readlines() i_times = i[0:10] account_id = i[55:59] #字符串相同还判断不出来，必须转换成数字才行,估计数字类型的得这样做 account_id = int(account_id) user_data['account_data']['id'] = int(user_data['account_data']['id']) # print("account_id:%s" %account_id) i_time = time.mktime(time.strptime(i_times,'%Y-%m-%d')) # print(i_time) if i_time >= select_time and account_id == user_data['account_data']['id']: print(i)def logout(acc_data): """ 退出登录 :param acc_data: :return: """ q = input("\033[33;1mIf you want to quit,please input q:\033[0m").strip() if q == 'q': exit() else: print("请输入q")def interactive(acc_data): """ :param acc_data: :return: """ # 加u是为了防止中文乱码 menu = u""" --------- Oldboy Bank --------- \033[32;1m1.账号信息 2.还款 3.取款 4.转账 5.账单 6.退出 \033[0m"""#这里比较牛的是写个字典，所以数字对应的是函数的内存地址 menu_dic = { '1':account_info, '2':repay, '3':withdraw, '4':transfer, '5':pay_check, '6':logout } exit_flag = False #这个标记为有很重要，是循环的前提 while not exit_flag: print(menu) user_option = input(">>:").strip() if user_option in menu_dic: # for i in menu_dic 显示出的是字典的key menu_dic[user_option](acc_data)def run(): """ 认证字段为真，用户数据为数据库的数据，信息补全了 :return: """ #经过执行acc_login之后，如果认证成功会改变user_data中的数据，认证成功字段置为真，赋予相应账号, acc_data = auth.acc_login(user_data,access_logger) #acc_data变成了数据库数据,而且是字典的格式 if user_data['is_authenticated']: user_data['account_data'] = acc_data interactive(user_data) else: print("\033[31;1mOption does not exist!\033[0m")# run()

settlement.py:

import os
import sys,time

BARE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(BARE_DIR)
from core import auth
from core import logger
from core import transaction
from core import accounts
from conf import settings

#logger()函数执行，并且返回了logger，可以进行调用warn方法进行日志写入
trans_logger = logger.logger('transaction')
access_logger = logger.logger('access')

user_data = {
'account_id':None,
'is_authenticated':False,
'account_data':None
}
def account_info(acc_data):
print(user_data['account_data'])
def repay(acc_data):
 account_data = user_data['account_data']
# for k,v in account_data.items():
 # print(k,v)
 current_balance = """
 ----------- BALANCE INFO --------
 Credit:%s
 Balance:%s
 """ %(account_data['credit'],account_data['balance'])
print(current_balance)
 back_flag = False #设置标志，输入b跳出while循环
 while not back_flag:
 repay_amount = input("\033[33;1mInput repay amount:\033[0m").strip()
if len(repay_amount) > 0 and repay_amount.isdigit():
# print('ddd 00')
 new_balance = transaction.make_transaction(trans_logger,account_data,'repay',repay_amount)
# print(new_balance)
 if new_balance:
print("""\033[42;1mNew Balance:%s\033[0m"""%(new_balance['balance']))
else:
print('\033[31;1m[%s] is not a valid amount,only accept integer!\033[0m' % repay_amount)
if repay_amount == 'b':
 back_flag = True

def withdraw(acc_data):
"""
 取款
 :param acc_data: 变成了数据库数据,而且是字典的格式
 :return:
 """
 accout_data = accounts.load_current_balance(acc_data['account_id']) #再去数据库拿一遍数据，有点多余
 current_balance = '''
 ------- BALANCE INFO -------
 Credit: %s
 Balance: %s
 '''%(accout_data['credit'],accout_data['balance'])
print(current_balance)
 back_flag = False
 while not back_flag:
 withdraw_amount = input("\033[33;1mInput withdraw amount:\033[0m").strip()
if len(withdraw_amount) > 0 and withdraw_amount.isdigit():
 new_balance = transaction.make_transaction(trans_logger,accout_data,'withdraw',withdraw_amount)
if new_balance:
print('''\033[42;1mNew Balance:%s\033[0m'''%(new_balance['balance']))
else:
print('\033[31;1m[%s] is not a valid amount,only accept integer!')
if withdraw_amount == 'b':
 back_flag = True

def transfer(acc_data):
"""

 :param acc_data: 变成了数据库数据,而且是字典的格式
 :return:
 """
 account_data = accounts.load_current_balance(acc_data['account_id'])
 current_balance = '''
 ------- BALANCE INFO -------
 Credit: %s
 Balance: %s'''%(account_data['credit'],account_data['balance'])
print(current_balance)

#钱数必须是数字
 back_flag = False
 while not back_flag:
 tranfer_amount = input("\033[33;1mInput tranfer amount:\033[0m").strip()
#只支持整数
 if len(tranfer_amount) > 0 and tranfer_amount.isdigit():

#调用transaction，传入账号信息，交易类型，交易金额，交易日志
 new_account_data = transaction.make_transaction(trans_logger,account_data,'transfer',tranfer_amount)
if new_account_data:
print("\033[42;1mNew Balance:%s\033[0m"%new_account_data['balance'])
else:
print('\033[31;1m[%s] is not a valid amount,only accept integer!')
if tranfer_amount == 'b':
 back_flag = True

def pay_check(acc_data): #账单功能
 back_flag = False
 while not back_flag:
 Time_input = input("\033[33;1mplease input time(Y-M-D),输入b退出:\033[0m").strip()
if Time_input == 'b': back_flag = True continue else: try: select_time = time.mktime(time.strptime(Time_input,'%Y-%m-%d')) # print(select_time) except Exception as ValueError: print("清输入正确的日期格式") continue # log_file = "atm_test/log/transactions.log" log_file = "%s/log/%s" % (settings.BASE_DIR, settings.LOG_TYPES["transaction"]) # print(log_file) with open(log_file,"r",encoding="utf-8") as f: for i in f: # for i in f : 是和大文件读写，不用写f,readlines() i_times = i[0:10] account_id = i[55:59] #字符串相同还判断不出来，必须转换成数字才行,估计数字类型的得这样做 account_id = int(account_id) user_data['account_data']['id'] = int(user_data['account_data']['id']) # print("account_id:%s" %account_id) i_time = time.mktime(time.strptime(i_times,'%Y-%m-%d')) # print(i_time) if i_time >= select_time and account_id == user_data['account_data']['id']: print(i)def logout(acc_data): """ 退出登录 :param acc_data: :return: """ q = input("\033[33;1mIf you want to quit,please input q:\033[0m").strip() if q == 'q': exit() else: print("请输入q")def interactive(acc_data): """ :param acc_data: :return: """ # 加u是为了防止中文乱码 menu = u""" --------- Oldboy Bank --------- \033[32;1m1.账号信息 2.还款 3.取款 4.转账 5.账单 6.退出 \033[0m"""#这里比较牛的是写个字典，所以数字对应的是函数的内存地址 menu_dic = { '1':account_info, '2':repay, '3':withdraw, '4':transfer, '5':pay_check, '6':logout } exit_flag = False #这个标记为有很重要，是循环的前提 while not exit_flag: print(menu) user_option = input(">>:").strip() if user_option in menu_dic: # for i in menu_dic 显示出的是字典的key menu_dic[user_option](acc_data)def run(): """ 认证字段为真，用户数据为数据库的数据，信息补全了 :return: """ #经过执行acc_login之后，如果认证成功会改变user_data中的数据，认证成功字段置为真，赋予相应账号, acc_data = auth.acc_login(user_data,access_logger) #acc_data变成了数据库数据,而且是字典的格式 if user_data['is_authenticated']: user_data['account_data'] = acc_data interactive(user_data) else: print("\033[31;1mOption does not exist!\033[0m")# run()

transaction.py：

import os,sys,json
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(BASE_DIR)
from conf import settings
from core import logger
from core import accounts

def make_transaction(log_obj,account_data,tran_type,amount,**others):
"""
 处理所有的交易信息
 :param log_obj:
 :param account_data: 用户账号
 :param tran_type: 交易类型
 :param amount: 交易额度
 :param others:日志等
 :return:
 """
 amount = float(amount)
if tran_type in settings.TRANSACTION_TYPE:
 interest = amount * settings.TRANSACTION_TYPE[tran_type]['interest']
 old_balance = account_data['balance']
if settings.TRANSACTION_TYPE[tran_type]['action'] == 'plus':
 new_balance = old_balance + amount + interest
elif settings.TRANSACTION_TYPE[tran_type]['action'] == 'minus':
 new_balance = old_balance - amount - interest
#check credit
 if new_balance < 0:
print("""\033[31;1mYour credit [%s] is not enough for this transaction
 \[-%s],your current balance is [%s]""" %(account_data['credit'],(amount + interest),old_balance))
return
 account_data['balance'] = new_balance
#将更改完的数据写入到json文件中
 account_data['balance'] = new_balance
 accounts.dump_account(account_data) #保存新的账目到文件中
 log_obj.info("account:%s action:%s amount:%s interest:%s" %
 (account_data['id'], tran_type, amount, interest))

return account_data
else:
print("\033[31;1mTransaction type [%s] is not exist!\033[0m" % tran_type)


shopping.py:

import os,sys,json,time
BARE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(BARE_DIR)
from core import accounts
from conf import settings
from core import db_handler
from core import logger
from core import settlement
"""
购物车功能，只能用来进行购物
"""

product_list = []
#此处是对txt文档进行处理，只能处理成列表格式，如果是对JSON格式，就是load成字典格式
product = "%s/shopping/product.txt" % (settings.BASE_DIR)
with open (product,'r',encoding='utf-8') as f:
for line in f:
 line = line.strip() #去掉最后一个换行符
 index,item = line.split(":")
 product_list.append((index,item)) #以括号的格式变成一个元素，其实里边是一个个的元组
 # print(product_list)
 # print(len(product_list)) #元素的个数，以 1 开头
#enumerate() 函数用于将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列，同时列出数据和数据下标

#现在需要在ATM上写一个接口，提供用户输入用户名和账号，然后返回 用户的账号资金

def acc_auth(func): #函数里套函数就是高阶函数
 def deco(*args,**kwargs):
 log_obj = logger.logger('access')
 retry_count = 0
 while retry_count < 3:
 account = input("\033[32;1m登录账号:\033[0m").strip()
 password = input("\033[32;1m密码:\033[0m").strip()
 db_path = db_handler.db_handler(settings.DATABASE)
 account_file = "%s/%s.json" %(db_path,account)
if os.path.isfile(account_file):
with open(account_file,'r') as f:
 account_data = json.load(f)
if account_data['password'] == password:
 func(*args,**kwargs) #需要将认证完之后的数据传入到shopping中
 else:
print("\033[31;1mAccount ID or password is incorrect!\033[0m")
else:
print("\033[31;1mAccount [%s] does not exist!\033[0m" % account)
 retry_count +=1
 else:
 log_obj.error("shopping account [%s] too many login attempts" % account)
exit()
return deco

def print_product_list(): #可以显示小标和具体的数据
 for index,item in enumerate(product_list):
print(index,item)
@acc_auth #进行装饰器的认证
def user_shopping():
print("--------------------------"
 "--------------------------"
 "\n"
 " 欢迎进入购物菜单 "
 "\n")
 print_product_list() # 打印商品
 salary_account =settlement.settlement() #需要调用settlement模块的settlement()方法
 salary = salary_account['balance']
if salary > 0:
 shopping_list = []
while True:
 option = input("请输入你要购买的商品的编号：").strip()
if option.isdigit():
 option = int(option)
if option >= 0 and option <= len(product_list):
 p_item = product_list[option] #用户选择的商品，对应的是一个元组
 c_num = int(p_item[1]) #对应的是商品价格
 if salary >= c_num:
 shopping_list.append(p_item)
 salary -= c_num
print("添加购物车成功，你的余额还有%s" %salary)
else:
print("你的余额不足，只剩%s元" % (salary))
else:
print("输入错误，请重新输入！")
elif option == "q":
print("----------------购物清单---------------")
for s_list in shopping_list:
print(s_list)
print("你的余额为%s" % (salary))
 salary_account['balance'] = salary
 accounts.dump_account(salary_account) #将修改完的数据写到数据库中
 print("..........exit.........")
exit() #退出程序
 else:
exit('余额不足！')

user_shopping()