DataInstaller.php 2.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. <?php
  2. /**
  3. * Copyright © Magento, Inc. All rights reserved.
  4. * See COPYING.txt for license details.
  5. */
  6. namespace Magento\Directory\Setup;
  7. use Magento\Framework\App\ResourceConnection;
  8. use Magento\Framework\DB\Adapter\AdapterInterface;
  9. /**
  10. * Class DataInstaller
  11. * @package Magento\Directory\Setup
  12. */
  13. class DataInstaller
  14. {
  15. /**
  16. * @var \Magento\Directory\Helper\Data
  17. */
  18. private $data;
  19. /**
  20. * @var ResourceConnection
  21. */
  22. private $resourceConnection;
  23. /**
  24. * DatInstaller constructor.
  25. * @param \Magento\Directory\Helper\Data $data
  26. * @param ResourceConnection $resourceConnection
  27. */
  28. public function __construct(
  29. \Magento\Directory\Helper\Data $data,
  30. ResourceConnection $resourceConnection
  31. ) {
  32. $this->data = $data;
  33. $this->resourceConnection = $resourceConnection;
  34. }
  35. /**
  36. * Add country-region data.
  37. *
  38. * @param AdapterInterface $adapter
  39. * @param array $data
  40. */
  41. public function addCountryRegions(AdapterInterface $adapter, array $data)
  42. {
  43. /**
  44. * Fill table directory/country_region
  45. * Fill table directory/country_region_name for en_US locale
  46. */
  47. foreach ($data as $row) {
  48. $bind = ['country_id' => $row[0], 'code' => $row[1], 'default_name' => $row[2]];
  49. $adapter->insert($this->resourceConnection->getTableName('directory_country_region'), $bind);
  50. $regionId = $adapter->lastInsertId($this->resourceConnection->getTableName('directory_country_region'));
  51. $bind = ['locale' => 'en_US', 'region_id' => $regionId, 'name' => $row[2]];
  52. $adapter->insert($this->resourceConnection->getTableName('directory_country_region_name'), $bind);
  53. }
  54. /**
  55. * Upgrade core_config_data general/region/state_required field.
  56. */
  57. $countries = $this->data->getCountryCollection()->getCountriesWithRequiredStates();
  58. $adapter->update(
  59. $this->resourceConnection->getTableName('core_config_data'),
  60. [
  61. 'value' => implode(',', array_keys($countries))
  62. ],
  63. [
  64. 'scope="default"',
  65. 'scope_id=0',
  66. 'path=?' => \Magento\Directory\Helper\Data::XML_PATH_STATES_REQUIRED
  67. ]
  68. );
  69. }
  70. }