Cédric Krier
f6d10a651b
We add the flake8 configuration used so we ensure everyone uses the same. We remove the usage of __all__ for non public API. When possible, we rationalize the class name according to its __name__ and module. issue9082 review297061002
61 lines
2.2 KiB
Python
61 lines
2.2 KiB
Python
# This file is part of Tryton. The COPYRIGHT file at the top level of
|
|
# this repository contains the full copyright notices and license terms.
|
|
from trytond.model import ModelView, fields
|
|
from trytond.wizard import Wizard, StateView, StateAction, Button
|
|
from trytond.transaction import Transaction
|
|
from trytond.pool import Pool
|
|
|
|
|
|
class CreateInventoriesStart(ModelView):
|
|
'Create Inventories'
|
|
__name__ = 'stock.inventory.create.start'
|
|
date = fields.Date('Date', required=True)
|
|
lost_found = fields.Many2One(
|
|
'stock.location', 'Lost and Found', required=True,
|
|
domain=[('type', '=', 'lost_found')])
|
|
company = fields.Many2One('company.company', 'Company', required=True,
|
|
select=True)
|
|
locations = fields.Many2Many('stock.location', None, None,
|
|
'Locations', required=True, domain=[('type', '=', 'storage')])
|
|
|
|
@classmethod
|
|
def default_lost_found(cls):
|
|
Location = Pool().get('stock.location')
|
|
locations = Location.search(cls.lost_found.domain)
|
|
if len(locations) == 1:
|
|
return locations[0].id
|
|
|
|
@staticmethod
|
|
def default_company():
|
|
return Transaction().context.get('company')
|
|
|
|
|
|
class CreateInventories(Wizard):
|
|
'Create Inventories'
|
|
__name__ = 'stock.inventory.create'
|
|
start = StateView('stock.inventory.create.start',
|
|
'stock_inventory_location.inventory_create_start_view_form', [
|
|
Button('Cancel', 'end', 'tryton-cancel'),
|
|
Button('Create', 'create_', 'tryton-ok', default=True),
|
|
])
|
|
create_ = StateAction('stock.act_inventory_form')
|
|
|
|
def do_create_(self, action):
|
|
Inventory = Pool().get('stock.inventory')
|
|
|
|
to_create = []
|
|
for location in self.start.locations:
|
|
to_create.append({
|
|
'location': location.id,
|
|
'date': self.start.date,
|
|
'lost_found': self.start.lost_found.id,
|
|
'company': self.start.company.id,
|
|
})
|
|
if to_create:
|
|
inventories = Inventory.create(to_create)
|
|
|
|
Inventory.complete_lines(inventories)
|
|
|
|
data = {'res_id': [i.id for i in inventories]}
|
|
return action, data
|