PT--work 02 日志文件+购物系统(注册 登录 购物)

# 1. 打印图案
# 1)
for i in range(1,6):
    for j in range(1,i+1):
        print('*',end='')
    print('')

# 2)
for i in range(5,0,-1):
    for j in range(i,0,-1):
        print('*',end='')
    print('')

# 3)
for i in range(1,10,2):
    s='*'*i
    print(s.center(9))


# 2.  将日志文件读取到不同的小文件中

#  想复杂了,很多没有用到
import os
def func(a1):
    with open(a1,mode='r',encoding='utf-8') as f1:
        time=[]
        for line in f1:
            li = line.split()
            time.append(li[0])
            for m in time:
                    if li[0] == m:
                            with open(li[0],mode='a',encoding='utf-8')as f3:
                                    f3.write(line)
                                    print('')
                            break
            else:
                with open(a1,mode='a',encoding='utf-8') as f2:
                    f2.write(line)
                    print('')
                os.rename(a1+'副本',li[0])
func('a1')


# 方法二  简单化
import os
def func(a1):
    with open(a1,mode='r',encoding='utf-8')as f:
        for line in f:
            li = line.split()
            with open(li[0], mode='a', encoding='utf-8')as f3:   #直接写入文件中即可
                f3.write(line)
                print('')
func('a1')


# 3.

def regist():  #注册
    my_username = input('请输入用户名:')
    my_password = input('请输入密码:')
    if my_username == ' ' or  my_password == ' ':
        print('输入用户名不合法,请重新输入!')
        return
    else:
        with open('a2',mode='a+',encoding='utf-8')as f:
            f.seek(0)
            li = f.readlines()
            for el in li:
                el=el.split(':')
                if my_username == el[0]:
                    print('用户名已存在,请重新输入!')
                    f.close()
                    return
            else:
                f.write(my_username+':'+my_password+'\n')
                f.flush()
                f.close()
                print('注册成功!')
                return
m = 0  #用于记录登录是否成功  1(成功)  -1(不成功)
def login():  #登录
    global m
    m = 0
    for i in range(3,0,-1):
        my_username = input('请输入用户名:')
        my_password = input('请输入密码:')
        with open('a2',mode='r',encoding='utf-8')as f:
            li = f.readlines()
            for el in li:
                lis = el.strip().split(':')  #strip() 去掉'\n'符号
                if my_username == lis[0] and my_password == lis[1]:
                    print('登陆成功!')
                    m = m+1
                    return
            else:
                print('登录失败,剩余次登录次数为%s'%(i-1))
    else:
        m = m-1
        return

def shopping(zichan):  #购物
    goods = [
        {"name": "电脑", "price": 1999},
        {"name": "鼠标", "price": 10},
        {"name": "游艇", "price": 20},
        {"name": "美女", "price": 998},
    ]
    sum=0   #计算商品总额
    shop={}  # 索引  数量
    if zichan.isdigit():  # 判断输入的是数字
        while 1:
            for i in goods:
                i_index = goods.index(i)
                print(i_index+1,i['name'],i['price'])  # 打印商品菜单
            xu_hao = input('请用户选择商品的序号(N(结算)  Q(退出结算))):')
            if xu_hao.upper() == 'Q':  #暂不结算
                break
            elif xu_hao.isdigit() and 0 < int(xu_hao) < len(goods)+1:  #选择商品
                id=int(xu_hao)-1
                print(goods[id]['name'],goods[id]['price'])  #打印选择的商品
                if shop.get(id) ==None :
                    shop[id] =1
                else:
                    shop[id] = shop[id]+1
            elif xu_hao.upper() == 'N':  #结算
                for i in shop:  #  i为idex
                    sum = sum + shop[i]*goods[i]['price']
                if int(zichan) >= sum:
                    for i in shop:
                        print('购买的商品为%s,单价为%s,数量为%s'%(goods[i]['name'],goods[i]['price'],shop[i]))
                    print('商品总额为%s,剩余额度为%s.'%(sum,int(zichan)-sum,))
                    return
                else:
                    print('账户余额不足!')
                    break
            else:
                print('输入有误,请重新输入!')
    else:
        print('输入有误,请重新输入!')


while 1:
    print('''欢迎进入老男孩系统:
1. 登录
2. 注册
3. 购物
4. 退出
    ''')
    content=input('请选择菜单:')
    if content == '1':
        print('1. 登录')
        login()
        if  m == -1:
            break
    elif content == '2':
        print('2. 注册')
        regist()
    elif content == '3':
        print('3. 购物')
        if m == 1:    #购物成功时
            money = input('请输入你的资产:')
            shopping(money)
        else:
            print('您未登录系统,请先登录!')
    elif content == '4':
        print('4. 退出')
        break
    else:
        print('输入错误,请重新输入!')