Python之路05-网上商城之商品上下架

需求：
1、实现对网上商城的商品上架

2、实现对网上商城的商品下架

3、实现对网上商城的商品更新价格功能

4、实现对网上商城的商品信息查询

要求：充分使用函数，请尽你的最大限度来减少重复代码！

# -*- coding: utf-8 -*-

import sys

def readGoogs():
    '''读取商品文件'''
    try:
        with open('goodsList.txt','r') as f:
            goods=f.readlines()
    except:
        print("文件未存在！")
        sys.exit(-1)
    return goods

def getGoods():
    '''获取商品信息'''
    good_list = []
    for good in readGoogs():
        good_list.append(good.split())

    return dict(good_list)

def printGoods():
    '''打印商品'''
    print("商品列表如下：")
    for index,item in enumerate(readGoogs()):
        print "%s	%s" % (index+1,item)


def goodsShelves():
    '''商品上架'''
    name=raw_input("请输入要上架的商品名称")
    if getGoods().has_key(name):
        print('上架商品已存在！')
        sys.exit(-1)
    price=raw_input("请输入该商品的价格")
    f = open('goodsList.txt','a')
    f.write(name + "	" + price + "
")
    print('已经将商品33[31;1m%s33[0m上架完成') % name
    f.flush()
    f.close()
    printGoods()

def goodDelete():
    '''商品下架'''
    name=raw_input("请输入要下架的商品名称")
    if getGoods().has_key(name):
        # 要先对文件读，再覆盖写入
        goods = readGoogs()
        with open('goodsList.txt','w') as f_w:
            for line in goods:
                # 写入时跳过要下架的商品即可
                if name in line:
                    continue
                f_w.write(line)
        print('已经将商品33[31;1m%s33[0m下架完成') % name
        printGoods()

    else:
        print('下架的商品不存在！')
        sys.exit(-1)

def goodModify():
    '''修改商品价格'''
    printGoods()
    name=raw_input("请输入要修改商品价格的名称")
    if getGoods().has_key(name):
        price=raw_input("请输入要修改商品的价格")
        goods = readGoogs()
        with open('goodsList.txt','w') as f_w:
            for line in goods:
                # 写入时跳过要下架的商品即可
                if name in line:
                    f_w.write(name+'	'+price+'
')
                    continue
                f_w.write(line)
        printGoods()
    else:
        print('修改的商品不存在！')
        sys.exit(-1)

def goodSearch():
    '''商品搜索'''
    name=raw_input("请输入搜索商品名称")
    if getGoods().has_key(name):
        for line in readGoogs():
            if name in line:
                print("搜索的商品价格为：%s" % line)
                break
    else:
        print('搜索的商品不存在！')
        sys.exit(-1)

def userChoice():
    '''用户选择'''
    print("欢迎登录XXX系统后台，请选择操作")
    print("1、商品上架")
    print("2、商品下架")
    print("3、修改商品价格")
    print("4、商品查询")
    print("5、退出")
    choice = raw_input('请选择：')
    if choice.isdigit():
        choice = int(choice)
        if 1 == choice:
            goodsShelves()
        elif 2 == choice:
            goodDelete()
        elif 3 == choice:
            goodModify()
        elif 4 == choice:
            goodSearch()
        elif 5 == choice:
            print("谢谢使用！")
            sys.exit(-1)
        else:
            print("输入有误，请重新输入")
            userChoice()
    else:
        print("请输入有效数字！")
        userChoice()

if __name__ == '__main__':
    userChoice()