csv_user_import.php

См. документацию.

<?php
 
use Bitrix\Main\Security\Random;
 
IncludeModuleLangFile(__FILE__);
 
class CSVUserImport
{
    var $csv;
    var $arHeader = false;
    var $isErrorOccured = false;
    var $errorMessage = "";
    var $ignoreDuplicate = false;
    var $userGroups = false;
    var $callback = null;
    var $defaultEmail = false;
    var $imageFilePath = null;
    var $externalAuthID = null;
 
    var $attachIBlockID = 0;
    var $userPropertyName = "UF_DEPARTMENT";
 
    var $arSectionCache = Array();
    var $isUserPropertyCreate = false;
 
    public function __construct($csvFilePath, $delimiter)
    {
        require_once($_SERVER["DOCUMENT_ROOT"]."/bitrix/modules/main/classes/general/csv_data.php");
 
        $this->csv = new CCSVData($fields_type = "R");
        $this->csv->LoadFile($csvFilePath);
        $this->csv->MoveFirst();
        $this->csv->SetDelimiter($delimiter);
        $this->csv->SetFirstHeader(false);
 
        if (!$this->arHeader = $this->csv->Fetch())
        {
            $this->isErrorOccured = true;
            $this->errorMessage = GetMessage("CSV_IMPORT_HEADER_NOT_FOUND");
            return;
        }
 
        foreach($this->arHeader as $key => $val)
            $this->arHeader[$key] = mb_strtoupper($val);
 
        if (!$this->CheckRequiredFields())
        {
            $this->isErrorOccured = true;
            return;
        }
    }
 
    function CheckRequiredFields()
    {
        if ($this->isErrorOccured || !is_array($this->arHeader) || count($this->arHeader) <= 1)
        {
            $this->errorMessage = GetMessage("CSV_IMPORT_DELIMETER_NOT_FOUND");
            return false;
        }
 
        $success = array_search("NAME", $this->arHeader);
        if ($success === false)
        {
            $this->errorMessage = GetMessage("CSV_IMPORT_NAME_NOT_FOUND");
            return false;
        }
 
        $success = array_search("LAST_NAME", $this->arHeader);
        if ($success === false)
        {
            $this->errorMessage = GetMessage("CSV_IMPORT_LAST_NAME_NOT_FOUND");
            return false;
        }
 
        return true;
    }
 
    function AttachUsersToIBlock($iblockID)
    {
        $iblockID = intval($iblockID);
        if (CModule::IncludeModule("iblock") && $iblockID > 0)
        {
            $dbIblock = CIBlock::GetByID($iblockID);
            if ($dbIblock->Fetch())
                $this->attachIBlockID = $iblockID;
        }
    }
 
    function SetUserPropertyName($userPropertyName)
    {
        $userPropertyName = trim($userPropertyName);
 
        if ($userPropertyName <> '')
            $this->userPropertyName = $userPropertyName;
    }
 
    function GenerateUserPassword($pass_len=10)
    {
        static $allchars = "abcdefghijklnmopqrstuvwxyzABCDEFGHIJKLNMOPQRSTUVWXYZ0123456789";
        $n = 61;
 
        $string = "";
        for ($i = 0; $i < $pass_len; $i++)
            $string .= $allchars[mt_rand(0, $n)];
 
        return $string;
    }
 
    function IsErrorOccured()
    {
        return $this->isErrorOccured;
    }
 
    function SetExternalAuthID($externalAuthID)
    {
        if ($externalAuthID <> '')
            $this->externalAuthID = $externalAuthID;
    }
 
    function GetErrorMessage()
    {
        return $this->errorMessage;
    }
 
    function IgnoreDuplicate($ignore = true)
    {
        $this->ignoreDuplicate = (bool)$ignore;
    }
 
    function SetCallback($functionName)
    {
        if (is_callable($functionName))
            $this->callback = $functionName;
    }
 
    function &GetCsvObject()
    {
        return $this->csv;
    }
 
    function SetDefaultEmail($email)
    {
        if (check_email($email))
            $this->defaultEmail = $email;
    }
 
    function GetDefaultEmail()
    {
        if ($this->defaultEmail !== false)
            return $this->defaultEmail;
 
        return COption::GetOptionString("main", "email_from", "admin@".$_SERVER["SERVER_NAME"]);
    }
 
    function SetUserGroups($arGroups)
    {
        if (!is_array($arGroups))
            return;
 
        foreach ($arGroups as $groupID)
        {
            $groupID = intval($groupID);
            $rsGroup = CGroup::GetByID($groupID);
            if (!$rsGroup->Fetch())
                continue;
 
            if (!is_array($this->userGroups))
                $this->userGroups = Array();
 
            $this->userGroups[] = $groupID;
        }
    }
 
    function SetImageFilePath($relativePath)
    {
        $relativePath = Rel2Abs("/", $relativePath);
        if (is_dir($_SERVER["DOCUMENT_ROOT"].$relativePath))
            $this->imageFilePath = rtrim($_SERVER["DOCUMENT_ROOT"].$relativePath, "/");
    }
 
    function __CreateUserProperty()
    {
        if ($this->attachIBlockID < 1)
            return false;
 
        $success = true;
        $dbRes = CUserTypeEntity::GetList(Array(), Array("ENTITY_ID" => "USER", "FIELD_NAME" => $this->userPropertyName));
        if (!$dbRes->Fetch())
        {
            $arLabelNames = Array();
            $rsLanguage = CLanguage::GetList();
            while($arLanguage = $rsLanguage->Fetch())
            {
                IncludeModuleLangFile($_SERVER["DOCUMENT_ROOT"].BX_ROOT."/modules/main/classes/general/csv_user_import_labels.php", $arLanguage["LID"]);
                $arLabelNames[$arLanguage["LID"]] = GetMessage("DEPARTMENT_USER_PROPERTY_NAME");
            }
 
            $arFields = Array(
                'ENTITY_ID' => 'USER',
                'FIELD_NAME' => $this->userPropertyName,
                'USER_TYPE_ID' => 'iblock_section',
                'XML_ID' => '',
                'SORT' => 100,
                'MULTIPLE' => 'Y',
                'MANDATORY' => 'N',
                'SHOW_FILTER' => 'I',
                'SHOW_IN_LIST' => 'Y',
                'EDIT_IN_LIST' => 'Y',
                'IS_SEARCHABLE' => 'Y',
                'SETTINGS' => array(
                    'DISPLAY' => 'LIST',
                    'LIST_HEIGHT' => '8',
                    'IBLOCK_ID' => $this->attachIBlockID,
                ),
 
                "EDIT_FORM_LABEL" => $arLabelNames,
                "LIST_COLUMN_LABEL" => $arLabelNames,
                "LIST_FILTER_LABEL" => $arLabelNames,
            );
 
            $userType = new CUserTypeEntity();
            $success = (bool)$userType->Add($arFields);
        }
 
        return $success;
    }
 
    function __GetIBlockSectionID(&$arFields)
    {
        $sectionID = 0;
        $i = 0;
 
        while(true)
        {
            $i++;
 
            $csvSectionCode = "IBLOCK_SECTION_NAME_".$i;
            if (!array_key_exists($csvSectionCode, $arFields))
                break;
 
            $sectionName = trim($arFields[$csvSectionCode]);
            if (mb_strlen($sectionName) < 1)
                break;
 
            $cacheID = md5($csvSectionCode."_".$sectionName."_".$sectionID);
            if (array_key_exists($cacheID, $this->arSectionCache))
            {
                $sectionID = $this->arSectionCache[$cacheID];
                continue;
            }
 
            $dbSection = CIBlockSection::GetList(Array(), Array("IBLOCK_ID" => $this->attachIBlockID, "NAME" => $sectionName, "SECTION_ID" => $sectionID));
            if ($arGroup = $dbSection->Fetch())
            {
                $sectionID = $arGroup["ID"];
                $this->arSectionCache[$cacheID] = $sectionID;
                continue;
            }
 
            $iblockSection = new CIBlockSection;
            $arSectionFields = Array(
                "ACTIVE" => "Y",
                "IBLOCK_SECTION_ID" => $sectionID,
                "IBLOCK_ID" => $this->attachIBlockID,
                "NAME" => $sectionName,
            );
 
            $sectionID = (int)$iblockSection->Add($arSectionFields);
            if ($sectionID > 1)
                $this->arSectionCache[$cacheID] = $sectionID;
            else
                return 0;
        }
 
        return $sectionID;
    }
 
    function ImportUser()
    {
        if ($this->isErrorOccured)
            return false;
 
        $this->errorMessage = "";
 
        $defaultEmail = $this->GetDefaultEmail();
 
        if (!$arUser = $this->csv->FetchDelimiter())
            return false;
 
        $arFields = Array();
        foreach($this->arHeader as $index => $key)
            if(($f = trim($arUser[$index])) <> '')
                $arFields[$key] = $f;
 
        if (!array_key_exists("NAME", $arFields) || mb_strlen($arFields["NAME"]) < 1)
        {
            $this->errorMessage = GetMessage("CSV_IMPORT_NO_NAME")." (".implode(", ", $arFields).").<br>";
            return true;
        }
 
        if (!array_key_exists("LAST_NAME", $arFields) || mb_strlen($arFields["LAST_NAME"]) < 1)
        {
            $this->errorMessage = GetMessage("CSV_IMPORT_NO_LASTNAME")." (".implode(", ", $arFields).").<br>";
            return true;
        }
 
        if (!array_key_exists("PASSWORD", $arFields) || mb_strlen($arFields["PASSWORD"]) < 1)
            $arFields["PASSWORD"] = $this->GenerateUserPassword(6);
        $arFields["CONFIRM_PASSWORD"] = $arFields["PASSWORD"];
 
        if (!array_key_exists("EMAIL", $arFields) || mb_strlen($arFields["EMAIL"]) < 3 || !check_email($arFields["EMAIL"]))
            $arFields["EMAIL"] = $defaultEmail;
 
        if (!array_key_exists("LOGIN", $arFields))
            $arFields["LOGIN"] = mb_strtolower($arFields["NAME"]." ".$arFields["LAST_NAME"]);
 
        if (array_key_exists("PERSONAL_BIRTHDAY", $arFields) && (mb_strlen($arFields["PERSONAL_BIRTHDAY"]) < 2 || !CheckDateTime($arFields["PERSONAL_BIRTHDAY"])))
            unset($arFields["PERSONAL_BIRTHDAY"]);
 
        if (array_key_exists("DATE_REGISTER", $arFields) && (mb_strlen($arFields["DATE_REGISTER"]) < 2 || !CheckDateTime($arFields["DATE_REGISTER"])))
            unset($arFields["DATE_REGISTER"]);
 
        if ($this->externalAuthID !== null && !array_key_exists("EXTERNAL_AUTH_ID", $arFields))
            $arFields["EXTERNAL_AUTH_ID"] = $this->externalAuthID;
 
        if (!array_key_exists("XML_ID", $arFields))
            $arFields["XML_ID"] = Random::getString(32);
 
        if(!array_key_exists("CHECKWORD", $arFields) || $arFields["CHECKWORD"] == '')
            $arFields["CHECKWORD"] = Random::getString(32);
 
        if ($this->imageFilePath !== null)
        {
            if (array_key_exists("PERSONAL_PHOTO", $arFields) && $arFields["PERSONAL_PHOTO"] <> '')
            {
                $arFile = CFile::MakeFileArray($this->imageFilePath."/".$arFields["PERSONAL_PHOTO"]);
                $arFile["MODULE_ID"] = "main";
                $arFields["PERSONAL_PHOTO"] = $arFile;
            }
 
            if (array_key_exists("WORK_LOGO", $arFields) && $arFields["WORK_LOGO"] <> '')
            {
                $arFile = CFile::MakeFileArray($this->imageFilePath."/".$arFields["WORK_LOGO"]);
                $arFile["MODULE_ID"] = "main";
                $arFields["WORK_LOGO"] = $arFile;
            }
        }
        else
        {
            unset($arFields["PERSONAL_PHOTO"]);
            unset($arFields["WORK_LOGO"]);
        }
 
        $arFields["GROUP_ID"] = $this->userGroups;
 
        $user = new CUser;
        $userID = (int)$user->Add($arFields);
 
        if($userID <= 0)
        {
            if($user->LAST_ERROR <> '')
                $this->errorMessage = $arFields["NAME"]." ".$arFields["LAST_NAME"].": ".$user->LAST_ERROR;
        }
 
        if ($userID <= 0 && $this->ignoreDuplicate === false)
        {
            $postFix = 2;
            $login = $arFields["LOGIN"];
            do
            {
                $rsUser = CUser::GetByLogin($arFields["LOGIN"]);
                if (!$rsUser->Fetch())
                    break;
 
                $arFields["LOGIN"] = $login.$postFix;
                $userID = (int)$user->Add($arFields);
                if ($userID > 1)
                    break;
 
                $postFix++;
 
            } while(true);
        }
 
        if ($userID > 0)
        {
            if ($this->attachIBlockID > 0)
            {
                $iblockSectionID = $this->__GetIBlockSectionID($arFields);
                if ($iblockSectionID > 0)
                {
                    if (!$this->isUserPropertyCreate)
                        $this->isUserPropertyCreate = $this->__CreateUserProperty();
 
                    $arUpdate = Array();
                    $arUpdate[$this->userPropertyName] = Array($iblockSectionID);
 
                    $user->Update($userID, $arUpdate);
                }
            }
 
            if ($this->callback !== null)
                call_user_func_array($this->callback, Array(&$arFields, &$userID));
        }
 
        return true;
 
    }
}