<h2>我设法构建了我想要的东西:</h2>
<pre><code>def dataRead():
with open("Inventory_List.csv", "r") as my_csv:
myFile = csv.reader(my_csv, delimiter=",")
global dataInventoryList
dataInventoryList = [[col[0], col[1], col[2], col[3], col[4], eval(col[5])] for col in myFile]
def dataWrite():
with open("Inventory_List.csv", "w+") as my_csv:
myFile = csv.writer(my_csv, delimiter=',', lineterminator='\n')
myFile.writerows(dataInventoryList)
def main():
while True:
found = False
dataRead()
print("==========================================================================")
print("Before update;")
for i in range(len(dataInventoryList)):
for j in range(6):
print(str(dataInventoryList[i][j]) + "\t", end="")
print("")
search = input("Enter product code: ")
quantity = int(input("Please enter quantity: "))
choice = int(input("Add stocks[1] Pull Stacks[2]\nChoice: "))
for i in range(len(dataInventoryList)):
if search == dataInventoryList[i][0]:
found = True
if choice == 1:
dataInventoryList[i][5] += quantity
break
elif choice == 2:
if dataInventoryList[i][5] == 0:
print("Requested order is out of stocks!\n")
break
elif quantity > dataInventoryList[i][5]:
print("Requested order is out of range!\n")
break
elif quantity <= dataInventoryList[i][5]:
dataInventoryList[i][5] -= quantity
break
if found == False:
print("Requested order was not found on the database!\n")
elif found == True:
dataWrite()
print("After update;")
for i in range(len(dataInventoryList)):
for j in range(6):
print(str(dataInventoryList[i][j]) + "\t", end="")
print("")
main()
</code></pre>