-
Notifications
You must be signed in to change notification settings - Fork 821
/
MemberCsvBulkLoader.php
92 lines (77 loc) · 2.35 KB
/
MemberCsvBulkLoader.php
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
<?php
namespace SilverStripe\Security;
use SilverStripe\Core\Convert;
use SilverStripe\Dev\CsvBulkLoader;
use SilverStripe\ORM\DataObject;
/**
* Imports member records, and checks/updates duplicates based on their
* 'Email' property.
*/
class MemberCsvBulkLoader extends CsvBulkLoader
{
/**
* @var array Array of {@link Group} records. Import into a specific group.
* Is overruled by any "Groups" columns in the import.
*/
protected $groups = [];
public function __construct($objectClass = null)
{
if (!$objectClass) {
$objectClass = 'SilverStripe\\Security\\Member';
}
parent::__construct($objectClass);
}
/**
* @var array
*/
public $duplicateChecks = [
'ID' => 'ID',
'Email' => 'Email',
];
public function processRecord($record, $columnMap, &$results, $preview = false)
{
$objID = parent::processRecord($record, $columnMap, $results, $preview);
$_cache_groupByCode = [];
// Add to predefined groups
/** @var Member $member */
$member = DataObject::get_by_id($this->objectClass, $objID);
foreach ($this->groups as $group) {
$member->Groups()->add($group);
}
// Add to groups defined in CSV
if (isset($record['Groups']) && $record['Groups']) {
$groupCodes = explode(',', $record['Groups'] ?? '');
foreach ($groupCodes as $groupCode) {
$groupCode = Convert::raw2url($groupCode);
if (!isset($_cache_groupByCode[$groupCode])) {
$group = Group::get()->filter('Code', $groupCode)->first();
if (!$group) {
$group = new Group();
$group->Code = $groupCode;
$group->Title = $groupCode;
$group->write();
}
$member->Groups()->add($group);
$_cache_groupByCode[$groupCode] = $group;
}
}
}
$member->destroy();
unset($member);
return $objID;
}
/**
* @param array $groups
*/
public function setGroups($groups)
{
$this->groups = $groups;
}
/**
* @return array
*/
public function getGroups()
{
return $this->groups;
}
}