2016-10-29 220 views
0

我正在製作一個程序,需要我遍歷文本文件並完成一筆金額,低於所需值的任何值都需要添加到發票文件中。然而,我創建的代碼只寫入一個產品,而不是每一個需要補貨的產品。Python - 搜索文本文件

下面是主要代碼:

def createRestockFile(productName,minimumStockLevel,currentStock, amountNeeded,costToUs): 
    with open("invoice.txt", 'r+') as f: 
     f.write("#Product Name\tMinimum Stock Level\tCurrent Stock Level\tAmount Needed\tCost To Re-Order \n") 
     f.write("%s\t%s\t%s\t%s\t%s" % (productName,minimumStockLevel,currentStock,amountNeeded,costToUs)) 

def checkStock(): 
    with open("stock.txt",'r+') as f: 
     for line in f: 
      if int(line.split()[2]) < int(line.split()[5]): 
       amountNeeded = int(line.split()[5]) - int(line.split()[2]) 
       total = '£{:,.2f}'.format(float(line.split()[3])*amountNeeded) 
       createRestockFile(line.split()[1],line.split()[5],line.split()[2],amountNeeded,total) 
       print(line.split()) 


def startProgramme(): 
    yesInput = ["yes", "yes please", "y"] 
    noInput = ["no","nope","n"] 
    print("Welcome to Sean's Stock re-order programme") 
    choice = input("Would you like to check which products need re-ordering ") 
    if choice in yesInput: 
     checkStock() 
    elif choice in noInput: 
     import time 
     print("Thank you for using Sean's re-order programme") 
     print("Ending Programme") 
     time.sleep(0.6) 
     exit() 



startProgramme() 

這裏是發票文件:

#Product Name Minimum Stock Level Current Stock Level Amount Needed Cost To Re-Order 
Wispa 16 6 10 £3.4003.40 

這裏是股票文件:

45678948 Twix 12 0.42 0.65 25 50 
12345670 Wispa 6 0.34 0.85 16 40 
26073125 Crunchie 37 0.37 0.69 8 43  
24785122 Flake 47 0.24 0.65 10 35 
45678914 Snickers 42 0.46 0.75 8 32  
78945617 Maltesers 78 0.32 0.56 12 65  
85146945 Galaxy 57 0.32 0.76 9 54 

在給定的股票文件中的值,該程序應該將twix和wispa添加到發票文件中,但僅添加wispa。任何幫助將不勝感激

回答

0

你需要改變你在打開invoice.txt模式對於功能,你需要將其從r+a+改變。它正在寫這個twix發票,然後刪除它,然後寫wispa。

0

以下代碼適用於我。

我已將您打開發票文件的代碼中的位置移動到主程序中,因此您可以保持打開模式「w +」。還請注意,我編寫的代碼只能分割一次輸入行(節省時間並縮短代碼)

def createRestockFile(productName,minimumStockLevel,currentStock, amountNeeded,costToUs, f): 
    f.write("%s\t%s\t%s\t%s\t%s" % (productName,minimumStockLevel,currentStock,amountNeeded,costToUs) + "\n") 


def checkStock(invoiceFile): 
    with open("stock.txt",'r+') as f: 
     for line in f: 
      splits = line.split() 
      if int(splits[2]) < int(splits[5]): 
       amountNeeded = int(splits[5]) - int(splits[2]) 
       total = '£{:,.2f}'.format(float(splits[3])*amountNeeded) 
       createRestockFile(splits[1],splits[5],splits[2],amountNeeded,total, invoiceFile) 
       print(splits) 


def startProgramme(): 
    yesInput = ["yes", "yes please", "y"] 
    noInput = ["no","nope","n"] 
    print("Welcome to Sean's Stock re-order programme") 
    choice = input("Would you like to check which products need re-ordering ") 
    if choice in yesInput: 
     invoice_f = open("invoice.txt", 'w+') 
     invoice_f.write("#Product Name\tMinimum Stock Level\tCurrent Stock Level\tAmount Needed\tCost To Re-Order \n") 
     checkStock(invoice_f) 
     invoice_f.close() 
    elif choice in noInput: 
     import time 
     print("Thank you for using Sean's re-order programme") 
     print("Ending Programme") 
     time.sleep(0.6) 
     exit() 

startProgramme()