280 lines
10 KiB
Python
280 lines
10 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 collections import defaultdict
|
|
|
|
import shopify
|
|
|
|
from trytond.i18n import gettext, lazy_gettext
|
|
from trytond.model import ModelSQL, ModelView, Unique, Workflow, fields
|
|
from trytond.model.exceptions import AccessError
|
|
from trytond.pool import Pool, PoolMeta
|
|
|
|
from . import graphql
|
|
from .common import IdentifierMixin, id2gid
|
|
from .exceptions import ShopifyError
|
|
|
|
QUERY_FULFILLMENT_ORDERS = '''\
|
|
query FulfillmentOrders($orderId: ID!) {
|
|
order(id: $orderId) %(fields)s
|
|
}'''
|
|
|
|
|
|
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, fulfillment_order_fields=None):
|
|
if self.state not in {'shipped', 'done'}:
|
|
return
|
|
shopify_id = self.get_shopify_identifier(sale)
|
|
if shopify_id:
|
|
# Fulfillment can not be modified
|
|
return
|
|
else:
|
|
fulfillment = {}
|
|
for shop_warehouse in sale.web_shop.shopify_warehouses:
|
|
if shop_warehouse.warehouse == self.warehouse:
|
|
location_id = int(shop_warehouse.shopify_id)
|
|
break
|
|
else:
|
|
location_id = None
|
|
fulfillment_order_fields = graphql.deep_merge(
|
|
fulfillment_order_fields or {}, {
|
|
'fulfillmentOrders(first: 250)': {
|
|
'nodes': {
|
|
'id': None,
|
|
'assignedLocation': {
|
|
'location': {
|
|
'id': None,
|
|
},
|
|
},
|
|
'lineItems(first: 250)': {
|
|
'nodes': {
|
|
'id': None,
|
|
'lineItem': {
|
|
'id': None,
|
|
'fulfillableQuantity': None,
|
|
},
|
|
},
|
|
},
|
|
'status': None,
|
|
},
|
|
},
|
|
})
|
|
order_id = id2gid('Order', sale.shopify_identifier)
|
|
fulfillment_orders = shopify.GraphQL().execute(
|
|
QUERY_FULFILLMENT_ORDERS % {
|
|
'fields': graphql.selection(fulfillment_order_fields),
|
|
}, {'orderId': order_id})['data']['order']['fulfillmentOrders']
|
|
line_items = defaultdict(list)
|
|
for move in self.outgoing_moves:
|
|
if move.sale == sale:
|
|
for order_id, line_item in move.get_shopify(
|
|
fulfillment_orders, location_id):
|
|
line_items[order_id].append(line_item)
|
|
if not line_items:
|
|
return
|
|
fulfillment['lineItemsByFulfillmentOrder'] = [{
|
|
'fulfillmentOrderId': order_id,
|
|
'fulfillmentOrderLineItems': line_items,
|
|
}
|
|
for order_id, line_items in line_items.items()]
|
|
fulfillment['notifyCustomer'] = bool(
|
|
sale.web_shop.shopify_fulfillment_notify_customer)
|
|
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)
|
|
|
|
@classmethod
|
|
@ModelView.button
|
|
@Workflow.transition('draft')
|
|
def draft(cls, shipments):
|
|
for shipment in shipments:
|
|
if shipment.state == 'cancelled' and shipment.shopify_identifiers:
|
|
raise AccessError(
|
|
gettext(
|
|
'web_shop_shopify.'
|
|
'msg_shipment_cancelled_draft_shopify',
|
|
shipment=shipment.rec_name))
|
|
super().draft(shipments)
|
|
|
|
|
|
class ShipmentShopifyIdentifier(IdentifierMixin, ModelSQL, ModelView):
|
|
__name__ = 'stock.shipment.shopify_identifier'
|
|
|
|
shipment = fields.Reference("Shipment", [
|
|
('stock.shipment.out', "Customer Shipment"),
|
|
], required=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_order_fields=None):
|
|
fulfillment = super().get_shopify(
|
|
sale, fulfillment_order_fields=fulfillment_order_fields)
|
|
if fulfillment and self.packages:
|
|
numbers, urls = [], []
|
|
fulfillment['trackingInfo'] = {
|
|
'numbers': numbers,
|
|
'urls': urls,
|
|
}
|
|
for package in self.packages:
|
|
if package.shipping_reference:
|
|
numbers.append(package.shipping_reference)
|
|
if package.shipping_tracking_url:
|
|
urls.append(package.shipping_tracking_url)
|
|
return fulfillment
|
|
|
|
|
|
class Move(metaclass=PoolMeta):
|
|
__name__ = 'stock.move'
|
|
|
|
def get_shopify(self, fulfillment_orders, location_id):
|
|
pool = Pool()
|
|
SaleLine = pool.get('sale.line')
|
|
Uom = pool.get('product.uom')
|
|
if (not isinstance(self.origin, SaleLine)
|
|
or not self.origin.shopify_identifier):
|
|
return
|
|
location_id = id2gid('Location', location_id)
|
|
identifier = id2gid('LineItem', self.origin.shopify_identifier)
|
|
quantity = round(Uom.compute_qty(
|
|
self.unit, self.quantity, self.origin.unit))
|
|
for fulfillment_order in fulfillment_orders['nodes']:
|
|
if fulfillment_order['status'] in {'CANCELLED', 'CLOSED'}:
|
|
continue
|
|
if (fulfillment_order['assignedLocation']['location']['id']
|
|
!= location_id):
|
|
continue
|
|
for line_item in fulfillment_order['lineItems']['nodes']:
|
|
if line_item['lineItem']['id'] == identifier:
|
|
qty = min(
|
|
quantity, line_item['lineItem']['fulfillableQuantity'])
|
|
if qty:
|
|
yield fulfillment_order['id'], {
|
|
'id': line_item['id'],
|
|
'quantity': qty,
|
|
}
|
|
quantity -= qty
|
|
if quantity <= 0:
|
|
return
|
|
else:
|
|
raise ShopifyError(gettext(
|
|
'web_shop_shopify.msg_fulfillment_order_line_not_found',
|
|
quantity=quantity,
|
|
move=self.rec_name,
|
|
))
|
|
|
|
|
|
class Move_Kit(metaclass=PoolMeta):
|
|
__name__ = 'stock.move'
|
|
|
|
def get_shopify(self, fulfillment_orders, location_id):
|
|
pool = Pool()
|
|
SaleLineComponent = pool.get('sale.line.component')
|
|
UoM = pool.get('product.uom')
|
|
yield from super().get_shopify(fulfillment_orders, location_id)
|
|
if not isinstance(self.origin, SaleLineComponent):
|
|
return
|
|
|
|
sale_line = self.origin.line
|
|
|
|
# Track only the first component
|
|
if min(c.id for c in sale_line.components) != self.origin.id:
|
|
return
|
|
|
|
location_id = id2gid('Location', location_id)
|
|
identifier = id2gid('LineItem', sale_line.shopify_identifier)
|
|
|
|
c_quantity = UoM.compute_qty(
|
|
self.unit, self.quantity, self.origin.unit, round=False)
|
|
if self.origin.quantity:
|
|
ratio = c_quantity / self.origin.quantity
|
|
else:
|
|
ratio = 1
|
|
quantity = round(sale_line.quantity * ratio)
|
|
for fulfillment_order in fulfillment_orders['nodes']:
|
|
if (fulfillment_order['assignedLocation']['location']['id']
|
|
!= location_id):
|
|
continue
|
|
for line_item in fulfillment_order['lineItems']['nodes']:
|
|
if line_item['lineItem']['id'] == identifier:
|
|
qty = min(
|
|
quantity, line_item['lineItem']['fulfillableQuantity'])
|
|
if qty:
|
|
yield fulfillment_order['id'], {
|
|
'id': line_item['id'],
|
|
'quantity': qty,
|
|
}
|
|
quantity -= qty
|
|
if quantity <= 0:
|
|
return
|
|
else:
|
|
raise ShopifyError(gettext(
|
|
'web_shop_shopify.msg_fulfillment_order_line_not_found',
|
|
quantity=quantity,
|
|
move=self.rec_name,
|
|
))
|