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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
#!/usr/bin/env perl
use warnings;
use v5.14;
use utf8;
BEGIN {
use File::Basename qw(dirname);
use File::Spec;
my $d = dirname(File::Spec->rel2abs($0));
require "$d/../setenv.pl";
}
use Term::ANSIColor;
use FixMyStreet;
use FixMyStreet::DB;
use Getopt::Long::Descriptive;
use JSON::MaybeXS;
use Path::Tiny;
my ($opt, $usage) = describe_options(
'%c %o',
[ 'body=s', "Name of body to add categories to" ],
[ 'commit', "Actually commit changes to the database" ],
[ 'delete', "Delete all existing body categories first" ],
[ 'help', "print usage message and exit", { shortcircuit => 1 } ],
);
print($usage->text), exit if $opt->help;
die "Usage: $0 <path/to/categories.json>" unless $opt->body;
die "Usage: $0 <path/to/categories.json>" unless -f $ARGV[0];
my $db;
END {
if ($db) {
$opt->commit ? $db->txn_commit : $db->txn_rollback;
}
}
$db = FixMyStreet::DB->schema->storage;
$db->txn_begin;
if (!$opt->commit) {
say colored("NOT COMMITTING TO DATABASE", 'cyan');
}
my $config = decode_json(path($ARGV[0])->slurp_utf8);
my $body = FixMyStreet::DB->resultset('Body')->find({ name => $opt->body });
$body->contacts->delete_all if $opt->delete;
die "Couldn't find body" unless $body;
my $groups = $config->{groups};
if ($groups) {
for my $group (keys %$groups) {
my $cats = $groups->{$group};
make_categories($cats, $group);
say "Created $group group";
}
} else {
my $categories = $config->{categories};
make_categories($categories);
say "Created non group categories";
}
sub make_categories {
my ($cats, $group) = @_;
for my $cat (@$cats) {
my $child_cat = FixMyStreet::DB->resultset("Contact")->find_or_new({
body => $body,
category => $cat->{category}
});
$child_cat->email($cat->{email});
$child_cat->state('confirmed');
$child_cat->editor($0);
$child_cat->whenedited(\'current_timestamp');
$child_cat->note($child_cat->in_storage ? 'Updated by import_categories' : 'Created by import_categories');
say colored("WARNING", 'red') . " " . $child_cat->category . " already exists" if $child_cat->in_storage and $child_cat->category ne 'Other (TfL)';
$child_cat->extra(undef) if $child_cat->in_storage;
if ($group) {
my $groups = $child_cat->groups;
my %groups = map { $_ => 1} @$groups;
$groups{$group} = 1;
my @groups = keys %groups;
$child_cat->set_extra_metadata(group => \@groups);
}
if ($cat->{disable}) {
$child_cat->update_extra_field({
code => "_fms_disable_",
disable_form => "true",
variable => "false",
protected => "true",
description => $cat->{disable} eq 1 ? $config->{disabled_message} : $cat->{disable},
order => 0,
});
}
$child_cat->set_extra_fields(@{ $cat->{extra_fields} }) if $cat->{extra_fields};
if (my $asset_field = $cat->{asset_field}) {
my ($description, $code) = @$asset_field;
$child_cat->update_extra_field({
code => $code,
description => $description,
automated => "hidden_field",
order => 1,
});
}
$child_cat->in_storage ? $child_cat->update : $child_cat->insert;
}
}
|
