misc/devel/populate_db.pl

   1 #!/usr/bin/perl
   2
   3 # This file is part of Koha.
   4 #
   5 # Copyright 2016 Koha Development Team
   6 #
   7 # Koha is free software; you can redistribute it and/or modify it
   8 # under the terms of the GNU General Public License as published by
   9 # the Free Software Foundation; either version 3 of the License, or
  10 # (at your option) any later version.
  11 #
  12 # Koha is distributed in the hope that it will be useful, but
  13 # WITHOUT ANY WARRANTY; without even the implied warranty of
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15 # GNU General Public License for more details.
  16 #
  17 # You should have received a copy of the GNU General Public License
  18 # along with Koha; if not, see <http://www.gnu.org/licenses>.
  19
  20 use Modern::Perl;
  21
  22 use Getopt::Long;
  23 use Pod::Usage;
  24
  25 use Koha::Script;
  26 use C4::Installer;
  27 use C4::Context;
  28
  29 use Koha::SearchEngine::Elasticsearch;
  30
  31 =head1 NAME
  32
  33 populate_db.pl - Load included sample data into the DB
  34
  35 =head1 SYNOPSIS
  36
  37 populate_db.pl [--marcflavour MARCFLAVOUR]
  38
  39  Options:
  40    --help            Brief help message
  41    --marcflavour m   Specify the MARC flavour to use (MARC21|UNIMARC). Defaults
  42                                 to MARC21.
  43    -v                Be verbose.
  44
  45 =head1 OPTIONS
  46
  47 =over 8
  48
  49 =item B<--help>
  50
  51 Prints a brief help message and exits.
  52
  53 =item B<--marcflavour>
  54
  55 Lets you choose the desired MARC flavour for the sample data. Valid options are MARC21 and UNIMARC.
  56 It defaults to MARC21.
  57
  58 =item B<--verbose>
  59
  60 Make the output more verbose.
  61
  62 =back
  63
  64 =cut
  65
  66 my $help;
  67 my $verbose;
  68 my $marcflavour = 'MARC21';
  69
  70 GetOptions(
  71     'help|?'        => \$help,
  72     'verbose'       => \$verbose,
  73     'marcflavour=s' => \$marcflavour
  74 ) or pod2usage;
  75
  76 if ( $help ) {
  77     pod2usage;
  78 }
  79
  80 $marcflavour = uc($marcflavour);
  81
  82 if (     $marcflavour ne 'MARC21'
  83      and $marcflavour ne 'UNIMARC' ) {
  84     say "Invalid MARC flavour '$marcflavour' passed.";
  85     pod2usage;
  86 }
  87
  88 $ENV{KOHA_DB_DO_NOT_RAISE_OR_PRINT_ERROR} = 1;
  89 my $dbh = C4::Context->dbh; # At the beginning to die if DB does not exist.
  90
  91 my ( $prefs_count ) = $dbh->selectrow_array(q|SELECT COUNT(*) FROM systempreferences|);
  92 my ( $patrons_count ) = $dbh->selectrow_array(q|SELECT COUNT(*) FROM borrowers|);
  93 if ( $prefs_count or $patrons_count ) {
  94     die "Database is not empty!";
  95 }
  96 $dbh->disconnect;
  97 $ENV{KOHA_DB_DO_NOT_RAISE_OR_PRINT_ERROR} = 0;
  98
  99 our $root      = C4::Context->config('intranetdir');
 100 our $data_dir  = "$root/installer/data/mysql";
 101 our $installer = C4::Installer->new;
 102 my $lang                = 'en';
 103 my $koha_structure_file = "$data_dir/kohastructure.sql";
 104 my @sample_files_mandatory = (
 105     glob("$data_dir/mandatory/*.sql"),
 106     "$data_dir/audio_alerts.sql",
 107     "$data_dir/sysprefs.sql",
 108     "$data_dir/userflags.sql",
 109     "$data_dir/userpermissions.sql",
 110     "$data_dir/account_offset_types.sql",
 111     "$data_dir/account_debit_types.sql",
 112 );
 113 my @sample_lang_files_mandatory    = ( glob $root . "/installer/data/mysql/$lang/mandatory/*.sql" );
 114 my @sample_lang_files_optional     = ( glob $root . "/installer/data/mysql/$lang/optional/*.sql" );
 115 my @marc21_sample_files_mandatory  = ( glob $root . "/installer/data/mysql/$lang/marcflavour/marc21/*/*.sql" );
 116 my @unimarc_sample_files_mandatory = ( glob $root . "/installer/data/mysql/$lang/marcflavour/unimarc/*/*.sql" );
 117
 118 my $version = get_version();
 119
 120 initialize_data();
 121 update_database();
 122
 123 sub initialize_data {
 124     say "Inserting koha db structure..."
 125         if $verbose;
 126     my $error = $installer->load_db_schema;
 127     die $error if $error;
 128
 129     for my $f (@sample_files_mandatory) {
 130         execute_sqlfile($f);
 131     }
 132
 133     for my $f (@sample_lang_files_mandatory) {
 134         execute_sqlfile($f);
 135     }
 136
 137     for my $f (@sample_lang_files_optional) {
 138         execute_sqlfile($f);
 139     }
 140
 141     if ( $marcflavour eq 'UNIMARC' ) {
 142         for my $f (@unimarc_sample_files_mandatory) {
 143             execute_sqlfile($f);
 144         }
 145     } else {
 146         for my $f (@marc21_sample_files_mandatory) {
 147             execute_sqlfile($f);
 148         }
 149     }
 150
 151     # set marcflavour (MARC21)
 152     my $dbh = C4::Context->dbh;
 153
 154     say "Setting the MARC flavour on the sysprefs..."
 155         if $verbose;
 156     $dbh->do(qq{
 157         INSERT INTO `systempreferences` (variable,value,explanation,options,type)
 158         VALUES ('marcflavour',?,'Define global MARC flavor (MARC21 or UNIMARC) used for character encoding','MARC21|UNIMARC','Choice')
 159     },undef,$marcflavour);
 160
 161     # set version
 162     say "Setting Koha version to $version..."
 163         if $verbose;
 164     $dbh->do(qq{
 165         INSERT INTO systempreferences(variable, value, options, explanation, type)
 166         VALUES ('Version', '$version', NULL, 'The Koha database version. WARNING: Do not change this value manually, it is maintained by the webinstaller', NULL)
 167     });
 168
 169     # Initialize ES mappings
 170     Koha::SearchEngine::Elasticsearch->reset_elasticsearch_mappings;
 171 }
 172
 173 sub execute_sqlfile {
 174     my ($filepath) = @_;
 175     say "Inserting $filepath..."
 176         if $verbose;
 177     my $error = $installer->load_sql($filepath);
 178     die $error if $error;
 179 }
 180
 181 sub get_version {
 182     do $root . '/kohaversion.pl';
 183     my $version = kohaversion();
 184     $version =~ s/(\d)\.(\d{2})\.(\d{2})\.(\d{3})/$1.$2$3$4/;
 185     return $version;
 186 }
 187
 188 sub update_database {
 189     my $update_db_path = $root . '/installer/data/mysql/updatedatabase.pl';
 190     say "Updating database..."
 191         if $verbose;
 192     my $file = `cat $update_db_path`;
 193     $file =~ s/exit;//;
 194     eval $file;
 195     if ($@) {
 196         die "updatedatabase.pl process failed: $@";
 197     } else {
 198         say "updatedatabase.pl process succeeded.";
 199     }
 200 }