#coding:utf-8
import sys
from openpyxl import load_workbook
from openpyxl import Workbook


#發貨數量
needNum = 200
#sku號
skuStr = 'sku0001'
#po所在列-1
poIdx = 0
#sku號所在列-1
skuIdx = 1
#數量所在列-1
quantityIdx = 2


wb = load_workbook('inventory.xlsx')
ws = wb.get_sheet_by_name('Sheet1')


skuList = []
poidList = []


for row in ws.rows:
if row[skuIdx].value == skuStr:
skuList.append(row)


skuList.sort(key=lambda x:x[quantityIdx].value)




for row in skuList:
if needNum > 0:
print row[poIdx].value
poidList.append(row[poIdx].value)
needNum = needNum - row[quantityIdx].value


if needNum > 0:
print 'not enough'
print needNum


#print sys.argv[0]


wb2 = Workbook()
ws2 = wb2.active 
for i in range(1,len(poidList)):
ws2['A' + str(i)] = poidList[i]
wb2.save('data.xlsx')