python大作業——列車管理系統

      這學期學了python,結果學的還是很淺,然後就寫了個簡單的大作業,學習一下。這是個簡單的操作系統,然後設計了一個簡單的界面,還是有很多粗糙的地方,後面還是需要好好學習,好好幹,繼續學。

import os
import re
import numpy as np
from tkinter import *


root=Tk()
root.title('列車查詢系統')
root.geometry('500x240')
tralist =[]
s=''

class Train: #定義一個列車類
    def __init__(self):
        self.forname = ''
        self.ID =''
        self.lastname = ''
        self.time = 0.0
        self.price = 0
        self.seating = 0

def Init(tralist):  #初始化函數
    print ("初始化......")
    file_object = open('train.txt', 'r')
    for line in file_object:
        tra = Train()
        line = line.strip("\n")
        s = line.split(" ")
        tra.ID = s[0]
        tra.forname = s[1]
        tra.lastname = s[2]
        tra.time = s[3]
        tra.price = s[4]
        tra.seating = s[5]
        tralist.append(tra)
    file_object.close()
    print ("初始化成功!")





class SEA():  
    def Search_f(self):
        roob=Toplevel(root)
        roob.title('列車信息查詢')
        roob.geometry('650x150')

        lb4 = Label(roob, text='請輸入列車車次')
        lb4.place(relx=0.2, rely=0.1, relwidth=0.4, relheight=0.2)
        d_text=StringVar()
        d_text.set("")
        self.inp4 = Entry(roob,textvariable =d_text)
        self.inp4.place(relx=0.6, rely=0.1, relwidth=0.2, relheight=0.2)

        self.bt=Button(roob,text="查詢",command=self.Search)
        self.bt.place(relx=0.1, rely=0.3, relwidth=0.2, relheight=0.2)
        
        self.btClose=Button(roob,text='關閉',command=roob.destroy)
        self.btClose.place(relx=0.5, rely=0.3, relwidth=0.2, relheight=0.2)
        
        self.txt = Text(roob)
        self.txt.place(relx=0.1, rely=0.6, relwidth=0.8, relheight=0.2)

    def Search(self):
        ID=self.inp4.get()
        if self.searchByID(tralist, ID) == False:
            self.txt.delete(1.0,END)
            self.txt.insert(END,'沒有該列車信息!' )
            self.inp4.delete(0, END)
            return
        
        self.txt.delete(1.0,END)
        self.txt.insert(END,"{0:^6}\t{1:{6}^6}\t{2:{6}^6}\t{3:{6}^6}\t{4:{6}^6}\t{5:{6}^6}".format("車次","起點","終點","時間","票價","座位數",chr(12288)))
        for item in tralist:
            if item.ID == ID:
                self.txt.insert(END,"{0:^8}\t{1:{6}^6}\t{2:{6}^6}\t{3:^12}\t{4:^12}\t{5:^8}".format(item.ID,item.forname,item.lastname,item.time,item.price,item.seating,chr(12288)))
                self.inp4.delete(0, END)
                break

    def searchByID(self,tralist,ID):
        for item in tralist:
            if item.ID == ID:
                return True
        return False

class CHA():
    def Change_f(self):
        roob=Toplevel(root)
        roob.title('列車信息修改')
        roob.geometry('600x240')

        lb = Label(roob, text='請輸入列車車次')
        lb.place(relx=0.1, rely=0, relwidth=0.4, relheight=0.1)
        d_text=StringVar()
        d_text.set("")
        self.inp = Entry(roob,textvariable =d_text)
        self.inp.place(relx=0.5, rely=0, relwidth=0.2, relheight=0.1)

        self.bt=Button(roob,text="查詢",command=self.sal)
        self.bt.place(relx=0.8, rely=0, relwidth=0.2, relheight=0.1)
        
        self.txt = Text(roob)
        self.txt.place(relx=0.1, rely=0.1, relwidth=0.8, relheight=0.1)

    
        
        lb1 = Label(roob, text='請輸入列車的車次')
        lb1.place(relx=0.1, rely=0.2, relwidth=0.5, relheight=0.1)
        a_text=StringVar()
        a_text.set("")
        self.inp1 = Entry(roob,textvariable =a_text)
        self.inp1.place(relx=0.6, rely=0.2, relwidth=0.3, relheight=0.1)
            
        lb2 = Label(roob, text='請輸入列車的起點')
        lb2.place(relx=0.1, rely=0.3, relwidth=0.5, relheight=0.1)
        b_text=StringVar()
        b_text.set("")
        self.inp2 = Entry(roob,textvariable =b_text)
        self.inp2.place(relx=0.6, rely=0.3, relwidth=0.3, relheight=0.1)
        
        lb3 = Label(roob, text='請輸入列車的終點')
        lb3.place(relx=0.1, rely=0.4, relwidth=0.5, relheight=0.1)
        c_text=StringVar()
        c_text.set("")
        self.inp3 = Entry(roob,textvariable =c_text)
        self.inp3.place(relx=0.6, rely=0.4, relwidth=0.3, relheight=0.1)
          
        lb4 = Label(roob, text='請輸入列車的時間')
        lb4.place(relx=0.1, rely=0.5, relwidth=0.5, relheight=0.1)
        d_text=StringVar()
        d_text.set("")
        self.inp4 = Entry(roob,textvariable =d_text)
        self.inp4.place(relx=0.6, rely=0.5, relwidth=0.3, relheight=0.1)
           
        lb5 = Label(roob, text='請輸入列車的票價')
        lb5.place(relx=0.1, rely=0.6, relwidth=0.5, relheight=0.1)
        e_text=StringVar()
        e_text.set("")
        self.inp5 = Entry(roob,textvariable =e_text)
        self.inp5.place(relx=0.6, rely=0.6, relwidth=0.3, relheight=0.1)
        
        lb6 = Label(roob, text='請輸入列車的座位數')
        lb6.place(relx=0.1, rely=0.7, relwidth=0.5, relheight=0.1)
        f_text=StringVar()
        f_text.set("")
        self.inp6 = Entry(roob,textvariable =f_text)
        self.inp6.place(relx=0.6, rely=0.7, relwidth=0.3, relheight=0.1)
        
        self.bt1=Button(roob,text="清空數據",command=self.Delete)
        self.bt1.place(relx=0.1, rely=0.9, relwidth=0.1, relheight=0.1)
        
        self.bt2=Button(roob,text="修改",command=self.Change)
        self.bt2.place(relx=0.7, rely=0.8, relwidth=0.2, relheight=0.1)
        
        self.btClose=Button(roob,text='關閉',command=roob.destroy)
        self.btClose.place(relx=0.7, rely=0.9, relwidth=0.2, relheight=0.1)
        
        self.txt1 = Text(roob)
        self.txt1.place(relx=0.1, rely=0.8, relwidth=0.4, relheight=0.1)

        
    def sal(self):
        ID=self.inp.get()
        if self.searchByID(tralist, ID) == False:
            self.txt.delete(1.0,END)
            self.txt.insert(END,'沒有該列車信息!' )
            self.inp.delete(0, END)
            return
        self.txt.delete(1.0,END)
        for item in tralist:
            if item.ID == ID:
                self.txt.insert(END,"{0:^8}\t{1:{6}^6}\t{2:{6}^6}\t{3:^12}\t{4:^12}\t{5:^8}".format(item.ID,item.forname,item.lastname,item.time,item.price,item.seating,chr(12288)))
                break


    def searchByID(self,tralist,ID):
        for item in tralist:
            if item.ID == ID:
                return True
        return False
    
    def Delete(self):
        self.inp1.delete(0, END)
        self.inp2.delete(0, END)
        self.inp3.delete(0, END)
        self.inp4.delete(0, END)
        self.inp5.delete(0, END)
        self.inp6.delete(0, END)
        

    
  
    def Change(self):     
        ID=self.inp.get()
        for item in tralist:
            if item.ID == ID:
                tralist.remove(item)
                file_object = open("train.txt", "w")
                for tra in tralist:
                    file_object.write(tra.ID)
                    file_object.write(" ")
                    file_object.write(tra.forname)
                    file_object.write(" ")
                    file_object.write(tra.lastname)
                    file_object.write(" ")
                    file_object.write(str(tra.time))
                    file_object.write(" ")
                    file_object.write(str(tra.price))
                    file_object.write(" ")
                    file_object.write(str(tra.seating))
                    file_object.write("\n")
                file_object.close()
        tra = Train()
        tra.ID=str(self.inp1.get())
        tra.forname=str(self.inp2.get())
        tra.lastname=str(self.inp3.get())
        tra.time=str(self.inp4.get())
        tra.price=str(self.inp5.get())
        tra.seating=str(self.inp6.get())
        self.Add(tralist,tra)
        print(tra.ID)    
    def Add(self,tralist,tra): 
        if self.searchByID(tralist, tra.ID) == True:
            self.txt1.delete(1.0,END)
            self.txt1.insert(END,'車次重複,修改失敗' )
            return
        tralist.append(tra)
        file_object = open("train.txt", "a")
        file_object.write(tra.ID)
        file_object.write(" ")
        file_object.write(tra.forname)
        file_object.write(" ")
        file_object.write(tra.lastname)
        file_object.write(" ")
        file_object.write(tra.time)
        file_object.write(" ")
        file_object.write(tra.price)
        file_object.write(" ")
        file_object.write(tra.seating)
        file_object.write("\n")
        file_object.close()
        self.txt1.delete(1.0,END)
        self.txt1.insert(END,'修改成功' )
        
        return
        
                

       

class DIS():
    def display(self):
        roob=Toplevel(root)
        roob.title('列車信息顯示')
        roob.geometry('600x300')

        self.txt = Text(roob)
        self.txt.place(relwidth=1, relheight=1)
        
        self.txt.insert(END,"{0:^6}\t{1:{6}^6}\t{2:{6}^6}\t{3:{6}^6}\t{4:{6}^6}\t{5:{6}^6}\n".format("車次","起點","終點","時間","票價","座位數",chr(12288)))
        size = os.path.getsize('train.txt')
        if size == 0:
            self.txt.insert(END,'沒有車次信息')
            return
        self.Sort(tralist)
        for item in tralist:
            self.txt.insert(END,"{0:^8}\t{1:{6}^6}\t{2:{6}^6}\t{3:^12}\t{4:^12}\t{5:^8}\n".format(item.ID,item.forname,item.lastname,item.time,item.price,item.seating,chr(12288)))
        
    def Sort(self,tralist):
        tra = []
        time_count = []
        for li in tralist:
            temp = []
            temp.append(li.ID)
            temp.append(li.forname)
            temp.append(li.lastname)
            temp.append(float(li.time))
            temp.append(int(li.price))
            temp.append(int(li.seating))
            time_count.append(float(li.time))
            tra.append(temp)
        self.insertSort(time_count,tralist)

        
    def insertSort(self,a, tralist):  
        for i in range(len(a)-1):   
            for j in range(i+1,len(a)):  
                if a[i]>a[j]:  
                    temp = tralist[i]  
                    tralist[i] = tralist[j]  
                    tralist[j] = temp  


    
class DEL():
    def Del_f(self):
        roob=Toplevel(root)
        roob.title('列車信息刪除')
        roob.geometry('400x160')
        
        lb4 = Label(roob, text='請輸入列車車次')
        lb4.place(relx=0.1, rely=0.2, relwidth=0.4, relheight=0.2)
        d_text=StringVar()
        d_text.set("")
        self.inp4 = Entry(roob,textvariable =d_text)
        self.inp4.place(relx=0.6, rely=0.2, relwidth=0.3, relheight=0.2)
        
        self.bt=Button(roob,text="刪除",command=self.Del)
        self.bt.place(relx=0.6, rely=0.5, relwidth=0.3, relheight=0.2)
        
        self.btClose=Button(roob,text='關閉',command=roob.destroy)
        self.btClose.place(relx=0.6, rely=0.7, relwidth=0.3, relheight=0.2)
        
        self.txt = Text(roob)
        self.txt.place(relx=0.1, rely=0.6, relwidth=0.4, relheight=0.2)
        
        
    def searchByID(self,tralist,ID): 
        for item in tralist:
            if item.ID == ID:
                return True
        return False    
    def Del(self):
        ID=self.inp4.get()
        if self.searchByID(tralist, ID) == False:
            self.txt.delete(1.0,END)
            self.txt.insert(END,'沒有該列車信息!' )
            self.inp4.delete(0, END)
            return
        
        for item in tralist:
            if item.ID == ID:
                tralist.remove(item)
                self.txt.delete(1.0,END)
                self.txt.insert(END,'刪除成功!' )
                self.inp4.delete(0, END)
                break
        
        file_object = open("train.txt", "w")
        for tra in tralist:
            file_object.write(tra.ID)
            file_object.write(" ")
            file_object.write(tra.forname)
            file_object.write(" ")
            file_object.write(tra.lastname)
            file_object.write(" ")
            file_object.write(str(tra.time))
            file_object.write(" ")
            file_object.write(str(tra.price))
            file_object.write(" ")
            file_object.write(str(tra.seating))
            file_object.write("\n")
        file_object.close()
    

class ADD:
    def Add_f(self):
        roob=Toplevel(root)
        roob.title('列車信息輸入')
        roob.geometry('500x240')

        lb1 = Label(roob, text='請輸入列車的車次')
        lb1.place(relx=0.1, rely=0.1, relwidth=0.5, relheight=0.1)
        a_text=StringVar()
        a_text.set("")
        self.inp1 = Entry(roob,textvariable =a_text)
        self.inp1.place(relx=0.6, rely=0.1, relwidth=0.3, relheight=0.1)
            
        lb2 = Label(roob, text='請輸入列車的起點')
        lb2.place(relx=0.1, rely=0.2, relwidth=0.5, relheight=0.1)
        b_text=StringVar()
        b_text.set("")
        self.inp2 = Entry(roob,textvariable =b_text)
        self.inp2.place(relx=0.6, rely=0.2, relwidth=0.3, relheight=0.1)
        
        lb3 = Label(roob, text='請輸入列車的終點')
        lb3.place(relx=0.1, rely=0.3, relwidth=0.5, relheight=0.1)
        c_text=StringVar()
        c_text.set("")
        self.inp3 = Entry(roob,textvariable =c_text)
        self.inp3.place(relx=0.6, rely=0.3, relwidth=0.3, relheight=0.1)
          
        lb4 = Label(roob, text='請輸入列車的時間')
        lb4.place(relx=0.1, rely=0.4, relwidth=0.5, relheight=0.1)
        d_text=StringVar()
        d_text.set("")
        self.inp4 = Entry(roob,textvariable =d_text)
        self.inp4.place(relx=0.6, rely=0.4, relwidth=0.3, relheight=0.1)
           
        lb5 = Label(roob, text='請輸入列車的票價')
        lb5.place(relx=0.1, rely=0.5, relwidth=0.5, relheight=0.1)
        e_text=StringVar()
        e_text.set("")
        self.inp5 = Entry(roob,textvariable =e_text)
        self.inp5.place(relx=0.6, rely=0.5, relwidth=0.3, relheight=0.1)
        
        lb6 = Label(roob, text='請輸入列車的座位數')
        lb6.place(relx=0.1, rely=0.6, relwidth=0.5, relheight=0.1)
        f_text=StringVar()
        f_text.set("")
        self.inp6 = Entry(roob,textvariable =f_text)
        self.inp6.place(relx=0.6, rely=0.6, relwidth=0.3, relheight=0.1)
        
        self.bt1=Button(roob,text="清空數據",command=self.Delete)
        self.bt1.place(relx=0.1, rely=0.9, relwidth=0.1, relheight=0.1)
        
        self.bt=Button(roob,text="添加",command=lambda:self.Add(a_text,b_text,c_text,d_text,e_text,f_text))
        self.bt.place(relx=0.7, rely=0.8, relwidth=0.2, relheight=0.1)
        
        self.btClose=Button(roob,text='關閉',command=roob.destroy)
        self.btClose.place(relx=0.7, rely=0.9, relwidth=0.2, relheight=0.1)
        
        self.txt = Text(roob)
        self.txt.place(relx=0.1, rely=0.8, relwidth=0.4, relheight=0.1)
        
        
    def Delete(self):
        self.inp1.delete(0, END)
        self.inp2.delete(0, END)
        self.inp3.delete(0, END)
        self.inp4.delete(0, END)
        self.inp5.delete(0, END)
        self.inp6.delete(0, END)
        
    def Add(self,ID,forname,lastname,time,price,seating): 
        tra = Train()
        tra.ID=str(ID.get())
        tra.forname=str(forname.get())
        tra.lastname=str(lastname.get())
        tra.time=str(time.get())
        tra.price=str(price.get())
        tra.seating=str(seating.get())
        print(tra.ID,tra.forname,tra.lastname,tra.time,tra.price,tra.seating)
        if self.searchByID(tralist, tra.ID) == True:
            self.txt.delete(1.0,END)
            self.txt.insert(END,'車次重複,保存失敗' )
            return
        tralist.append(tra)
        file_object = open("train.txt", "a")
        file_object.write(tra.ID)
        file_object.write(" ")
        file_object.write(tra.forname)
        file_object.write(" ")
        file_object.write(tra.lastname)
        file_object.write(" ")
        file_object.write(tra.time)
        file_object.write(" ")
        file_object.write(tra.price)
        file_object.write(" ")
        file_object.write(tra.seating)
        file_object.write("\n")
        file_object.close()
        self.txt.delete(1.0,END)
        self.txt.insert(END,'保存成功' )
        
        return
    
    def searchByID(self,tralist,ID):
        for item in tralist:
            if item.ID == ID:
                return True
        return False

    


    
def main(): #主函數 該程序的入口函數
    #root=Toplevel(rooa)
    b.destroy()
    root.title('列車查詢系統')
    root.geometry('500x240')
    root.colors='#d3fbfb'
    add=ADD()
    bt1=Button(root,text="添加數據",command=add.Add_f)
    bt1.place(relx=0.3, rely=0.2, relwidth=0.3, relheight=0.1)
    
    
    dela=DEL()
    bt2=Button(root,text="刪除數據",command=dela.Del_f)
    bt2.place(relx=0.3, rely=0.3, relwidth=0.3, relheight=0.1)

    sea=SEA()
    bt6=Button(root,text="列車查詢",command=sea.Search_f)
    bt6.place(relx=0.3, rely=0.4, relwidth=0.3, relheight=0.1)
    
    dis=DIS()
    bt3=Button(root,text="列車顯示",command=dis.display)
    bt3.place(relx=0.3, rely=0.5, relwidth=0.3, relheight=0.1)

    cha=CHA()
    bt4=Button(root,text="修改數據",command=cha.Change_f)
    bt4.place(relx=0.3, rely=0.6, relwidth=0.3, relheight=0.1)
    
    bt5=Button(root,text="退出系統",command=root.destroy)
    bt5.place(relx=0.3, rely=0.7, relwidth=0.3, relheight=0.1)


Init(tralist)

b=Button(root,text="歡迎使用列車管理系統",bg='#d3fbfb',fg='red',font=('華文新魏',32),relief=SUNKEN,command=main,width=500,height=240)
b.pack()

root.mainloop()


 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章