NaNtic
/
trytond-product_qty
mirror of https://github.com/NaN-tic/trytond-product_qty.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Get locations in context

This commit is contained in:
resteve 2015-09-16 16:00:03 +02:00
parent 9b71e0c70b
commit a3e89c9891
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
product.py
View File

@ -30,8 +30,9 @@ class Product:
def get_quantity(cls, products, name):
Location = Pool().get('stock.location')
context = Transaction().context
# not locations in context
if 'locations' not in Transaction().context:
if not context.get('locations'):
warehouses = Location.search([('type', '=', 'warehouse')])
location_ids = [w.storage_location.id for w in warehouses]
with Transaction().set_context(locations=location_ids):
@ -43,8 +44,9 @@ class Product:
def search_quantity(cls, name, domain=None):
Location = Pool().get('stock.location')
context = Transaction().context
# not locations in context
if 'locations' not in Transaction().context:
if not context.get('locations'):
warehouses = Location.search([('type', '=', 'warehouse')])
location_ids = [w.storage_location.id for w in warehouses]
with Transaction().set_context(locations=location_ids):