As a requirement, once Work Orders are created in an MO, all pickings related to that MO must be automatically booked and validated.

Below is the code I created for the task. It seems to work fine as all pickings are validated except when there aren't enough raw material quantity on hand. However and especially for the wood items with lot numbers, it seems the bookings are not properly being deducted from the inventory. Can someone review my code and check if the manner of how I reserve the items are valid or is there a better approach? Thanks in advance!

============================

# find out if this order has components
mo_records = env['mrp.production'].search(["&","&",["origin","=",record.name],["state","=","confirmed"],["routing_id.x_studio_bom_type","=","Component"]])

if mo_records:
  # components were found
  raise Warning('Cảnh báo: Bạn phải xác nhận và sản xuất tất cả các đơn đặt hàng linh kiện trước khi xác nhận đơn đặt hàng sản xuất chính này! (Warning: You must confirm and produce all component orders before confirming this main production order!)')
    
else:
  # no components were found, so this is main order
  records.button_plan()

#This sequence of codes will auto-reserve and confirm related picking orders of the MO after creating the work orders
pickings = env['stock.picking'].search(['&','&',('origin','=',record.name),('picking_type_id','=',21),('state','in',['confirmed','assigned'])])

if pickings:
  for picking in pickings:
    picking.action_confirm()
    picking.action_assign()
    
    # process and validate the material picking transaction
    move_items = env['stock.move'].search([('picking_id','=',picking\.id\)\]\)
\ \ \ \ 
\ \ \ \ for\ move\ in\ move_items:
\ \ \ \ \ \ has_lines\ =\ env\['stock\.move\.line'\]\.search\(\[\('move_id','=',move\.id\)\]\)

\ \ \ \ \ \ if\ has_lines:
\ \ \ \ \ \ \ \ for\ line\ in\ has_lines:
\ \ \ \ \ \ \ \ \ \ line\['qty_done'\]\ =\ line\.product_uom_qty

\ \ \ \ \ \ \ \ \ \ \#\ if\ a\ picking\ material\ is\ wood\ and\ has\ tracking\ for\ lot\ numbers,\ record\ the\ consumption\ into\ x_mrp_picking_lots\ for\ later\ consolidated\ consumption\ report
\ \ \ \ \ \ \ \ \ \ if\ line\.lot_id:
\ \ \ \ \ \ \ \ \ \ \ \ lot_move_lines\ =\ env\['stock\.move\.line'\]\.search\(\['\&','\&',\('lot_id','=',line\.lot_id\.id\),\('reference','like','WH/IN%'\),\('location_id','=',8\)\],\ limit=1\)
\ \ \ \ \ \ \ \ \ \ \ \ orders\ =\ env\['purchase\.order'\]\.search\(\[\('name','=',lot_move_lines\[0\]\.x_studio_po_reference\)\],\ limit=1\)
\ \ \ \ \ \ \ \ \ \ \ \ if\ orders:
\ \ \ \ \ \ \ \ \ \ \ \ \ \ order_id\ =\ orders\[0\]\.id
\ \ \ \ \ \ \ \ \ \ \ \ else:
\ \ \ \ \ \ \ \ \ \ \ \ \ \ order_id\ =\ False
\ \ \ \ \ \ \ \ \ \ \ \ 
\ \ \ \ \ \ \ \ \ \ \ \ is_fsc\ =\ False
\ \ \ \ \ \ \ \ \ \ \ \ 
\ \ \ \ \ \ \ \ \ \ \ \ if\ record\.product_id\.product_tmpl_id\.categ_id\.id\ ==\ 34\ and\ record\.product_id\.product_tmpl_id\.x_fsc_category\ ==\ True:\ \#\ is\ FG\ and\ FSC\ Category\ is\ set
\ \ \ \ \ \ \ \ \ \ \ \ \ \ is_fsc\ =\ True
\ \ \ \ \ \ \ \ \ \ \ \ else:
\ \ \ \ \ \ \ \ \ \ \ \ \ \ is_fsc\ =\ False
\ \ \ \ \ \ \ \ \ \ \ \ \ \ 
\ \ \ \ \ \ \ \ \ \ \ \ mrps\ =\ env\['mrp\.production'\]\.search\(\[\('name','=',record\.origin\)\]\)
\ \ \ \ \ \ \ \ \ \ \ \ if\ mrps:
\ \ \ \ \ \ \ \ \ \ \ \ \ \ source_mrp\ =\ mrps\[0\]\.id
\ \ \ \ \ \ \ \ \ \ \ \ \ \ 
\ \ \ \ \ \ \ \ \ \ \ \ env\['x_mrp_picking_lots'\]\.create\(\{
\ \ \ \ \ \ \ \ \ \ \ \ \ \ 'x_name':record\.name,
\ \ \ \ \ \ \ \ \ \ \ \ \ \ 'x_name2':\ record\.id,
\ \ \ \ \ \ \ \ \ \ \ \ \ \ 'x_source':\ record\.origin,
\ \ \ \ \ \ \ \ \ \ \ \ \ \ 'x_source2':\ source_mrp,
\ \ \ \ \ \ \ \ \ \ \ \ \ \ 'x_product_id':line\.product_id\.id,
\ \ \ \ \ \ \ \ \ \ \ \ \ \ 'x_lot_id':line\.lot_id\.id,
\ \ \ \ \ \ \ \ \ \ \ \ \ \ 'x_purchase_order':\ order_id,
\ \ \ \ \ \ \ \ \ \ \ \ \ \ 'x_qty_used':line\.product_uom_qty,
\ \ \ \ \ \ \ \ \ \ \ \ \ \ 'x_uom':line\.product_uom_id\.id,
\ \ \ \ \ \ \ \ \ \ \ \ \ \ 'x_fg_fsc':\ is_fsc,
\ \ \ \ \ \ \ \ \ \ \ \ \ \ 'x_state':'assigned'
\ \ \ \ \ \ \ \ \ \ \ \ \}\)
\ \ \ \ \ \ \ \ \ \ \ \ 
\ \ \ \ \ \ \ \ \ \ if\ move\.product_id\.type\ !=\ 'consu':
\ \ \ \ \ \ \ \ \ \ \ \ \#\ record\ the\ reserved\ quantity\ into\ Pre\-Production
\ \ \ \ \ \ \ \ \ \ \ \ quants\ =\ env\['stock\.quant'\]\.search\(\[\('product_id','=',line\.product_id\.id\),\('location_id','=',\ 34\),\('lot_id','=',\ False\)\],limit=1\)
\ \ \ \ \ \ \ \ \ \ \ \ 
\ \ \ \ \ \ \ \ \ \ \ \ if\ quants:
\ \ \ \ \ \ \ \ \ \ \ \ \ \ for\ quant\ in\ quants:
\ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ quants\['quantity'\]\ =\ quant\.quantity\ \+\ line\.product_uom_qty
\ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ quants\['reserved_quantity'\]\ =\ quant\.reserved_quantity\ \+\ line\.product_uom_qty
\ \ \ \ \ \ \ \ \ \ \ \ else:
\ \ \ \ \ \ \ \ \ \ \ \ \ \ env\['stock\.quant'\]\.sudo\(\)\.create\(\{'product_id':\ line\.product_id\.id,\ 'location_id':\ 34,\ 'quantity':\ line\.product_uom_qty,\ 'reserved_quantity':\ line\.product_uom_qty\}\)
\ \ \ \ \ \ \ \ \ \ 
\ \ \ \ \ \ \ \ \ \ \ \ \#\ deduct\ the\ reserved\ quantity\ from\ the\ storage\ location
\ \ \ \ \ \ \ \ \ \ \ \ quants\ =\ env\['stock\.quant'\]\.search\(\[\('product_id','=',line\.product_id\.id\),\('location_id','=',\ line\.location_id\.id\),\('lot_id','=',\ False\)\],limit=1\)
\ \ \ \ \ \ \ \ \ \ \ \ if\ quants:
\ \ \ \ \ \ \ \ \ \ \ \ \ \ for\ quant\ in\ quants:
\ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ quants\['quantity'\]\ =\ quant\.quantity\ \-\ line\.product_uom_qty
\ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ quants\['reserved_quantity'\]\ =\ quant\.reserved_quantity\ \-\ line\.product_uom_qty

\ \ \ \ \ \ \ \ move\['is_done'\]\ =\ True
\ \ \ \ \ \ \ \ move\['state'\]\ =\ 'done'
\ \ \ \ \ \ \#else:
\ \ \ \ \ \ \#\ \ raise\ Warning\('Could\ not\ reserve\ '\ \+\ move\.product_id\.name\ \ \+\ ',\ at\ this\ time,\ stopping\ auto\-booking\.'\)
\ \ 
\ \ \ \ date_done\ =\ datetime\.datetime\.today\(\)
\ \ \ \ picking\['date_done'\]\ =\ datetime.datetime.strftime(date_done, '%Y-%m-%d %H:%M:%S')
    picking['state'] = 'done'
    picking.action_done()
    picking.with_context(skip_backorder=True,skip_immediate=True).button_validate()
    
  # process the material pickings for material consumption of the production order
  record.action_assign()