python基础训练——DVD系统

为某音像店开发一个迷你DVD管理器，实现DVD碟片的管理，

包括如下功能：

新增DVD

查看DVD

删除DVD

借出DVD

归还DVD

'''
1.自定义数据类型，为dvd类型
2.用list列表来存储数据。
3.通过访问列表来实现增删改查
'''
import datetime
class Dvd:#自定义数据类型为Dvd类型

    def __init__(self,state,name,time):
        self.state=state#可借，不可借
        self.name=name#名字
        self.time=time#借出时间

def startMenu():
    print('***欢迎来到MINI_DVD***')
    print('---------------------')
    print('1.添加DVD')
    print('2.查看DVD')
    print('3.删除DVD')
    print('4.借出DVD')
    print('5.归还DVD')
    print('6.退   出')
    print('---------------------')
    print('请选择：',end='')

def add_Dvd(list):
    name=input('请输入DVD名称：')
    dvds=Dvd('可借',name,'')#转换为Dvd类型，新添加的都为可借
    list.append(dvds)#在列表尾加入
    print('添加成功...')
def see_Dvd(list):
    if list:
        print('序号\t\t状态\t\t名称\t\t  借出日期')
        for num in range(len(list)):
              print('%s\t\t%s\t\t%s\t\t%s'%(num+1,list[num].state,list[num].name,list[num].time))
    else:
        print('不存在任何DVD')
def del_Dvd(list):
    name=input('请输入要删除DVD的名称')
    for x in range(len(list)):
        if list[x].name==name:
            list.pop(x)
            print('删除成功')
            break
    else:
        print('不存在该DVD')
#借出dvd
def lend_Dvd(list):
    name=input('请输入你要借DVD的名称')
    for x in  range(len(list)):
        if list[x].name==name and list[x].state=='可借':
            print('借出成功')
            list[x].state='不可借'
            times=datetime.datetime.now().strftime('%Y-%m-%d')
            print(times)
            list[x].time=times
            break
        elif list[x].name==name and list[x].state=='不可借':
            print('当前DVD已经被借出')
            break
    else:
        print('不存在该dvd')
#归还dvd
def return_Dvd(list):
    name=input('请输入你要归还DVD的名称')
    for x in list:
        if x.name==name and x.state=='不可借':
            #得到借的时间
            return_time=datetime.datetime.now().strftime("%Y-%m-%d")
            lend_time=(datetime.datetime.strptime(return_time,'%Y-%m-%d')-datetime.datetime.strptime(list[x].time,'%Y-%m-%d')).days+1
            lend_time*=5
            print("归还成功，租金为:%d"%(lend_time))
            break
        elif list[x].name==name and list[x].state=='可借':
            print('当前DVD未借出，归还错误')
            break
    else:
        print('不存在该DVD')
'''DVD_list=[['不可借','罗马假日','2017-7-1'],['可借','风声鹤唳',''],['可借','浪漫满屋','']]
for x in DVD_list:
    print(x)
'''
DVD_list=[]
DVD_list.append(Dvd('不可借','罗马假日','2017-7-1'))
DVD_list.append(Dvd('可借','风声鹤唳',''))
DVD_list.append(Dvd('可借','浪漫满屋',''))
while True:
    startMenu()
    choice=input()
    if choice=='1':
        add_Dvd(DVD_list)
    elif choice=='2':
        see_Dvd(DVD_list)
        print(DVD_list)
    elif choice=='3':
        del_Dvd(DVD_list)
    elif choice=='4':
        lend_Dvd(DVD_list)
    elif choice=='5':
        return_Dvd(DVD_list)
    elif choice=='6':
        break
    else:
        print('输入错误')