Cédric Krier
04f5ba2f64
We need to commit after creating a fulfillment. Also the processing can be slow for the user when clicking on process button if many calls are needed. issue11081 review377741002
141 lines
4.8 KiB
Python
141 lines
4.8 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.
|
|
import shopify
|
|
|
|
from trytond.i18n import lazy_gettext
|
|
from trytond.model import ModelSQL, ModelView, Unique, fields
|
|
from trytond.pool import Pool, PoolMeta
|
|
|
|
from .common import IdentifierMixin
|
|
|
|
|
|
class ShipmentOut(metaclass=PoolMeta):
|
|
__name__ = 'stock.shipment.out'
|
|
|
|
shopify_identifiers = fields.One2Many(
|
|
'stock.shipment.shopify_identifier', 'shipment',
|
|
lazy_gettext('web_shop_shopify.msg_shopify_identifiers'))
|
|
|
|
def get_shopify(self, sale):
|
|
if self.state != 'done':
|
|
return
|
|
shopify_id = self.get_shopify_identifier(sale)
|
|
if shopify_id:
|
|
# Fulfillment can not be modified
|
|
return
|
|
else:
|
|
fulfillment = shopify.Fulfillment(
|
|
prefix_options={'order_id': sale.shopify_identifier})
|
|
for shop_warehouse in sale.web_shop.shopify_warehouses:
|
|
if shop_warehouse.warehouse == self.warehouse:
|
|
fulfillment.location_id = int(shop_warehouse.shopify_id)
|
|
break
|
|
line_items = []
|
|
for move in self.outgoing_moves:
|
|
if move.sale == sale:
|
|
line_item = move.get_shopify()
|
|
if line_item:
|
|
line_items.append(line_item)
|
|
if not line_items:
|
|
return
|
|
fulfillment.line_items = line_items
|
|
return fulfillment
|
|
|
|
def get_shopify_identifier(self, sale):
|
|
for record in self.shopify_identifiers:
|
|
if record.sale == sale:
|
|
return record.shopify_identifier
|
|
|
|
def set_shopify_identifier(self, sale, identifier=None):
|
|
pool = Pool()
|
|
Identifier = pool.get('stock.shipment.shopify_identifier')
|
|
for record in self.shopify_identifiers:
|
|
if record.sale == sale:
|
|
if not identifier:
|
|
Identifier.delete([record])
|
|
return
|
|
else:
|
|
if record.shopify_identifier != identifier:
|
|
record.shopify_identifier = identifier
|
|
record.save()
|
|
return record
|
|
if identifier:
|
|
record = Identifier(shipment=self, sale=sale)
|
|
record.shopify_identifier = identifier
|
|
record.save()
|
|
return record
|
|
|
|
@classmethod
|
|
def search_shopify_identifier(cls, sale, identifier):
|
|
records = cls.search([
|
|
('shopify_identifiers', 'where', [
|
|
('sale', '=', sale.id),
|
|
('shopify_identifier', '=', identifier),
|
|
]),
|
|
])
|
|
if records:
|
|
record, = records
|
|
return record
|
|
|
|
@classmethod
|
|
def copy(cls, records, default=None):
|
|
if default is None:
|
|
default = {}
|
|
else:
|
|
default = default.copy()
|
|
default.setdefault('shopify_identifiers')
|
|
return super().copy(records, default=default)
|
|
|
|
|
|
class ShipmentShopifyIdentifier(IdentifierMixin, ModelSQL, ModelView):
|
|
"Shopify Shipment Identifier"
|
|
__name__ = 'stock.shipment.shopify_identifier'
|
|
|
|
shipment = fields.Reference("Shipment", [
|
|
('stock.shipment.out', "Customer Shipment"),
|
|
], required=True, select=True)
|
|
sale = fields.Many2One('sale.sale', "Sale", required=True)
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super().__setup__()
|
|
cls.shopify_identifier_signed.states = {
|
|
'required': True,
|
|
}
|
|
t = cls.__table__()
|
|
cls._sql_constraints += [
|
|
('shipment_sale_unique',
|
|
Unique(t, t.shipment, t.sale, t.shopify_identifier_signed),
|
|
'web_shop_shopify.msg_identifier_shipment_sale_unique'),
|
|
]
|
|
|
|
|
|
class ShipmentOut_PackageShipping(metaclass=PoolMeta):
|
|
__name__ = 'stock.shipment.out'
|
|
|
|
def get_shopify(self, sale):
|
|
fulfillment = super().get_shopify(sale)
|
|
if fulfillment and self.packages:
|
|
fulfillment.tracking_numbers = [
|
|
p.shipping_reference for p in self.packages
|
|
if p.shipping_reference]
|
|
fulfillment.tracking_urls = [
|
|
p.shipping_tracking_url for p in self.packages
|
|
if p.shipping_tracking_url]
|
|
return fulfillment
|
|
|
|
|
|
class Move(metaclass=PoolMeta):
|
|
__name__ = 'stock.move'
|
|
|
|
def get_shopify(self):
|
|
pool = Pool()
|
|
SaleLine = pool.get('sale.line')
|
|
Uom = pool.get('product.uom')
|
|
if isinstance(self.origin, SaleLine):
|
|
return {
|
|
'id': self.origin.shopify_identifier,
|
|
'quantity': int(Uom.compute_qty(
|
|
self.uom, self.quantity, self.origin.unit)),
|
|
}
|