diff options
| -rwxr-xr-x | bin/tfl/import_categories | 114 |
1 files changed, 114 insertions, 0 deletions
diff --git a/bin/tfl/import_categories b/bin/tfl/import_categories new file mode 100755 index 000000000..b1cf23114 --- /dev/null +++ b/bin/tfl/import_categories @@ -0,0 +1,114 @@ +#!/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', + [ 'commit', "Actually commit changes to the database" ], + [ 'delete', "Delete all existing TfL 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 -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 => 'TfL' }); + +$body->contacts->delete_all if $opt->delete; + +die "Couldn't find TfL body" unless $body; + +my $groups = $config->{groups}; +for my $group (keys %$groups) { + my $cats = $groups->{$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'; + my $groups = $child_cat->groups; + my %groups = map { $_ => 1} @$groups; + $groups{$group} = 1; + my @groups = keys %groups; + $child_cat->extra(undef) if $child_cat->in_storage; + $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, + }); + } + # Add the safety critical hidden field + $child_cat->update_extra_field({ + code => "safety_critical", + description => "Safety critical", + automated => "hidden_field", + order => 1, + datatype => "singlevaluelist", + values => [ + { + name => "Yes", + key => "yes" + }, + { + name => "No", + key => "no" + } + ] + }); + $child_cat->in_storage ? $child_cat->update : $child_cat->insert; + } + + say "Created $group group"; +} |