Improve set_at_warehouse script

script/set_at_warehouse.py

@@ -61,6 +61,8 @@ if __name__ == '__main__':
                         wh = record.shipment.to_location.warehouse
                     elif key == 'receipt':
                         wh = record.goods_receipt_line.receipt.warehouse
+                    elif key == 'warehouse':
+                        wh = record.warehouse
                     elif not key:
                         wh = record.get_at_warehouse()
                     else:
@@ -101,5 +103,12 @@ if __name__ == '__main__':
                 ('goods_receipt_line.receipt.state', '=', 'done')
             ],
             key='receipt')
+        # warehouse production
+        set_warehouse(
+            extra_domain=[
+                ('shipment', '=', None),
+                ('warehouse', '!=', None),
+            ],
+            key='warehouse')
         # others
         set_warehouse()