Website.php

Go to the documentation of this file.

<?php
namespace Magento\Customer\Model\Customer\Attribute\Backend;

class Website extends \Magento\Eav\Model\Entity\Attribute\Backend\AbstractBackend
{
    protected $_storeManager;

    public function __construct(\Magento\Store\Model\StoreManagerInterface $storeManager)
    {
        $this->_storeManager = $storeManager;
    }

    public function beforeSave($object)
    {
        if ($object->getId()) {
            return $this;
        }

        if (!$object->hasData('website_id')) {
            $object->setData('website_id', $this->_storeManager->getStore()->getWebsiteId());
        }

        return $this;
    }
}