diff --git a/app/code/Magento/Catalog/Setup/CategorySetup.php b/app/code/Magento/Catalog/Setup/CategorySetup.php index 0791f162cc172..02dda9a482f50 100644 --- a/app/code/Magento/Catalog/Setup/CategorySetup.php +++ b/app/code/Magento/Catalog/Setup/CategorySetup.php @@ -67,6 +67,7 @@ public function getDefaultEntities() { return [ 'catalog_category' => [ + 'entity_type_id' => 3, 'entity_model' => 'Magento\Catalog\Model\ResourceModel\Category', 'attribute_model' => 'Magento\Catalog\Model\ResourceModel\Eav\Attribute', 'table' => 'catalog_category_entity', @@ -335,6 +336,7 @@ public function getDefaultEntities() ], ], 'catalog_product' => [ + 'entity_type_id' => 4, 'entity_model' => 'Magento\Catalog\Model\ResourceModel\Product', 'attribute_model' => 'Magento\Catalog\Model\ResourceModel\Eav\Attribute', 'table' => 'catalog_product_entity', diff --git a/app/code/Magento/Customer/Setup/CustomerSetup.php b/app/code/Magento/Customer/Setup/CustomerSetup.php index f6daffa52d6c4..9df5de4ed0a3c 100644 --- a/app/code/Magento/Customer/Setup/CustomerSetup.php +++ b/app/code/Magento/Customer/Setup/CustomerSetup.php @@ -126,6 +126,7 @@ public function getDefaultEntities() { $entities = [ 'customer' => [ + 'entity_type_id' => \Magento\Customer\Api\CustomerMetadataInterface::ATTRIBUTE_SET_ID_CUSTOMER, 'entity_model' => 'Magento\Customer\Model\ResourceModel\Customer', 'attribute_model' => 'Magento\Customer\Model\Attribute', 'table' => 'customer_entity', @@ -338,6 +339,7 @@ public function getDefaultEntities() ], ], 'customer_address' => [ + 'entity_type_id' => \Magento\Customer\Api\AddressMetadataInterface::ATTRIBUTE_SET_ID_ADDRESS, 'entity_model' => 'Magento\Customer\Model\ResourceModel\Address', 'attribute_model' => 'Magento\Customer\Model\Attribute', 'table' => 'customer_address_entity', diff --git a/app/code/Magento/Eav/Setup/EavSetup.php b/app/code/Magento/Eav/Setup/EavSetup.php index fbd25e293084c..1f31e1cd16b92 100644 --- a/app/code/Magento/Eav/Setup/EavSetup.php +++ b/app/code/Magento/Eav/Setup/EavSetup.php @@ -192,6 +192,9 @@ public function addEntityType($code, array $params) 'additional_attribute_table' => $this->_getValue($params, 'additional_attribute_table'), 'entity_attribute_collection' => $this->_getValue($params, 'entity_attribute_collection'), ]; + if (isset($params['entity_type_id'])) { + $data['entity_type_id'] = $params['entity_type_id']; + } if ($this->getEntityType($code, 'entity_type_id')) { $this->updateEntityType($code, $data); @@ -199,7 +202,11 @@ public function addEntityType($code, array $params) $this->setup->getConnection()->insert($this->setup->getTable('eav_entity_type'), $data); } - $this->addAttributeSet($code, $this->_defaultAttributeSetName); + if (isset($params['entity_type_id'])) { + $this->addAttributeSet($code, $this->_defaultAttributeSetName, null, $params['entity_type_id']); + } else { + $this->addAttributeSet($code, $this->_defaultAttributeSetName); + } $this->addAttributeGroup($code, $this->_defaultGroupName, $this->_generalGroupName); return $this; @@ -310,9 +317,10 @@ public function getAttributeSetSortOrder($entityTypeId, $sortOrder = null) * @param int|string $entityTypeId * @param string $name * @param int $sortOrder + * @param int $setId * @return $this */ - public function addAttributeSet($entityTypeId, $name, $sortOrder = null) + public function addAttributeSet($entityTypeId, $name, $sortOrder = null, $setId = null) { $data = [ 'entity_type_id' => $this->getEntityTypeId($entityTypeId), @@ -320,6 +328,10 @@ public function addAttributeSet($entityTypeId, $name, $sortOrder = null) 'sort_order' => $this->getAttributeSetSortOrder($entityTypeId, $sortOrder), ]; + if ($setId !== null) { + $data['attribute_set_id'] = $setId; + } + $setId = $this->getAttributeSet($entityTypeId, $name, 'attribute_set_id'); if ($setId) { $this->updateAttributeSet($entityTypeId, $setId, $data); diff --git a/app/code/Magento/Sales/Setup/SalesSetup.php b/app/code/Magento/Sales/Setup/SalesSetup.php index d69147ba925f0..d0c73fdc69eb4 100644 --- a/app/code/Magento/Sales/Setup/SalesSetup.php +++ b/app/code/Magento/Sales/Setup/SalesSetup.php @@ -214,6 +214,7 @@ public function getDefaultEntities() { $entities = [ 'order' => [ + 'entity_type_id' => 5, 'entity_model' => 'Magento\Sales\Model\ResourceModel\Order', 'table' => 'sales_order', 'increment_model' => 'Magento\Eav\Model\Entity\Increment\NumericValue', @@ -221,6 +222,7 @@ public function getDefaultEntities() 'attributes' => [], ], 'invoice' => [ + 'entity_type_id' => 6, 'entity_model' => 'Magento\Sales\Model\ResourceModel\Order\Invoice', 'table' => 'sales_invoice', 'increment_model' => 'Magento\Eav\Model\Entity\Increment\NumericValue', @@ -228,6 +230,7 @@ public function getDefaultEntities() 'attributes' => [], ], 'creditmemo' => [ + 'entity_type_id' => 7, 'entity_model' => 'Magento\Sales\Model\ResourceModel\Order\Creditmemo', 'table' => 'sales_creditmemo', 'increment_model' => 'Magento\Eav\Model\Entity\Increment\NumericValue', @@ -235,6 +238,7 @@ public function getDefaultEntities() 'attributes' => [], ], 'shipment' => [ + 'entity_type_id' => 8, 'entity_model' => 'Magento\Sales\Model\ResourceModel\Order\Shipment', 'table' => 'sales_shipment', 'increment_model' => 'Magento\Eav\Model\Entity\Increment\NumericValue',