price.php

<?php
 
namespace Bitrix\Sale\Services\Company\Restrictions;
 
use Bitrix\Sale\Internals\CollectableEntity;
use Bitrix\Sale\Internals\Entity;
use Bitrix\Sale\Order;
use Bitrix\Sale\Payment;
use Bitrix\Sale\Services\PaySystem\Restrictions;
use Bitrix\Sale\Shipment;
 
class Price extends Restrictions\Price
{
    protected static function extractParams(Entity $entity)
    {
        $collection = null;
 
        if ($entity instanceof Payment)
            $collection = $entity->getCollection();
        elseif ($entity instanceof Shipment)
            $collection = $entity->getCollection();
        elseif ($entity instanceof Order)
        {
            return array('PRICE_PAYMENT' => $entity->getPrice());
        }
 
        if ($collection)
        {
            $order = $collection->getOrder();
 
            return array('PRICE_PAYMENT' => $order->getPrice());
        }
 
        return array('PRICE_PAYMENT' => null);
    }
}