Python - Поиск текстового файла

Question

Я делаю программу, которая требует от меня итерации через текстовый файл и завершения суммы, любые значения, которые ниже необходимого значения, должны быть добавлены в файл счета. Однако созданный мной код записывает только один продукт, а не тот, который требуется для пополнения.

Вот основной код:

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. Любая помощь была бы принята с благодарностью

Answer 1

Вам нужно изменить режим открытия invoice.txt. Для этой функции вам необходимо изменить его с r+ на a+; он пишет счет-фактуру, затем удаляет его, а затем пишет wispa.

Answer 2

Для меня работает следующий код.

Я переместил позицию в код, в котором вы открываете файл счета в основной программе, чтобы сохранить режим открытия «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()