64 |
|
|
65 |
=cut |
=cut |
66 |
|
|
67 |
use Spreadsheet::WriteExcel; |
use Excel::Writer::XLSX; |
68 |
use DBI; |
use DBI; |
69 |
use CGI::Carp qw(fatalsToBrowser); |
use CGI::Carp qw(fatalsToBrowser); |
70 |
use Encode qw/decode/; |
use Encode qw/decode/; |
74 |
our $database = 'template1'; |
our $database = 'template1'; |
75 |
our $user = 'dpavlin'; |
our $user = 'dpavlin'; |
76 |
our $passwd = ''; |
our $passwd = ''; |
77 |
our $path = 'sql_reports.xls'; |
our $path = 'sql_reports.xlsx'; |
78 |
|
|
79 |
our $db_encoding = 'iso-8859-2'; |
our $db_encoding = 'iso-8859-2'; |
80 |
our $xls_date_format = 'dd.mm.yyyy'; |
our $xls_date_format = 'dd.mm.yyyy'; |
81 |
|
|
82 |
our $debug = 1; |
our $debug = $ENV{DEBUG} || 0; |
83 |
|
|
84 |
my $sql_dir = $ENV{SCRIPT_FILENAME} || '.'; |
my $sql_dir = $ENV{SCRIPT_FILENAME} || '.'; |
85 |
$sql_dir =~ s,/[^/]+$,,; |
$sql_dir =~ s,/[^/]+$,,; |
99 |
|
|
100 |
require_config; |
require_config; |
101 |
|
|
102 |
warn "# reading SQL queries from $sql_dir\n" if $debug; |
warn "SQL queries from $sql_dir\n"; |
103 |
|
|
104 |
opendir(DIR, $sql_dir) || die "can't opendir $sql_dir: $!"; |
opendir(DIR, $sql_dir) || die "can't opendir $sql_dir: $!"; |
105 |
my @sql_files = sort grep { /\.sql$/i && -f "$sql_dir/$_" } readdir(DIR); |
my @sql_files = sort grep { /\.sql$/i && -f "$sql_dir/$_" } readdir(DIR); |
109 |
if ($ENV{GATEWAY_INTERFACE} && $ENV{GATEWAY_INTERFACE} =~ m/CGI/i) { |
if ($ENV{GATEWAY_INTERFACE} && $ENV{GATEWAY_INTERFACE} =~ m/CGI/i) { |
110 |
# use as cgi script |
# use as cgi script |
111 |
print "Content-type: application/vnd.ms-excel\n\n"; |
print "Content-type: application/vnd.ms-excel\n\n"; |
112 |
$workbook = Spreadsheet::WriteExcel->new("-"); |
$workbook = Excel::Writer::XLSX->new("-"); |
113 |
} else { |
} else { |
114 |
# Create a new Excel workbook |
# Create a new Excel workbook |
115 |
$workbook = Spreadsheet::WriteExcel->new( $path ); |
$path =~ s{\.xls$}{\.xlsx}; |
116 |
|
$workbook = Excel::Writer::XLSX->new( $path ); |
117 |
warn "Creating XLS file $path\n"; |
warn "Creating XLS file $path\n"; |
118 |
} |
} |
119 |
|
|
125 |
my $database = shift || return; |
my $database = shift || return; |
126 |
print STDERR "## connect to $database\n" if $debug; |
print STDERR "## connect to $database\n" if $debug; |
127 |
$dbh = DBI->connect($dsn . $database,$user,$passwd, { RaiseError => 1, AutoCommit => 0 }) || die $DBI::errstr; |
$dbh = DBI->connect($dsn . $database,$user,$passwd, { RaiseError => 1, AutoCommit => 0 }) || die $DBI::errstr; |
128 |
$dbh->do( qq{ set names '$db_encoding'; } ) if $db_encoding && $dsn =~ m{mysql}; |
if ( $db_encoding ) { |
129 |
|
if ( $dsn =~ m{Pg} ) { |
130 |
|
$dbh->do( qq{ set client_encoding = '$db_encoding'; } ); |
131 |
|
} elsif ( $dsn =~ m{mysql} ) { |
132 |
|
$dbh->do( qq{ set names '$db_encoding'; } ); |
133 |
|
} else { |
134 |
|
warn "Don't know how to set encoding to $db_encoding for $dsn"; |
135 |
|
} |
136 |
|
} |
137 |
} |
} |
138 |
|
|
139 |
use_database( $database ); |
use_database( $database ); |
152 |
|
|
153 |
# Add a worksheet |
# Add a worksheet |
154 |
warn "# clipping sheet name '$sheet_name' to 31 char limit\n" if length $sheet_name > 31; |
warn "# clipping sheet name '$sheet_name' to 31 char limit\n" if length $sheet_name > 31; |
155 |
my $worksheet = $workbook->addworksheet( substr($sheet_name,0,31) ); |
my $worksheet = $workbook->add_worksheet( substr($sheet_name,0,31) ); |
156 |
|
|
157 |
print STDERR "working on $sql_file\n" if ($debug); |
print STDERR "working on $sql_file\n" if ($debug); |
158 |
|
|
181 |
if ($comment) { |
if ($comment) { |
182 |
|
|
183 |
# Add and define a format |
# Add and define a format |
184 |
my $fmt_comment = $workbook->addformat(); # Add a format |
my $fmt_comment = $workbook->add_format(); # Add a format |
185 |
$fmt_comment->set_bold(); |
$fmt_comment->set_bold(); |
186 |
|
|
187 |
$comment =~ s/^\s+//; |
$comment =~ s/^\s+//; |
191 |
$row+=2; |
$row+=2; |
192 |
} |
} |
193 |
|
|
194 |
my $fmt_header = $workbook->addformat(); # Add a format |
my $fmt_header = $workbook->add_format(); # Add a format |
195 |
$fmt_header->set_italic(); |
$fmt_header->set_italic(); |
196 |
|
|
197 |
foreach my $sql ( split(/;/, $full_sql ) ) { |
foreach my $sql ( split(/;/, $full_sql ) ) { |
198 |
|
|
199 |
warn "SQL: $sql\n" if $debug; |
warn "SQL: $sql\n"; |
200 |
|
|
201 |
my $sth = $dbh->prepare($sql); |
my $sth = $dbh->prepare($sql); |
202 |
$sth->execute(); |
$sth->execute(); |
203 |
|
|
204 |
next unless $sth->{NAME}; # $sth->rows doesn't work for insert into with MySQL |
next unless $sth->{NAME} && $sth->rows > 0; # $sth->rows alone doesn't work for insert into with MySQL |
205 |
|
|
206 |
my @types = eval { |
my @types = eval { |
207 |
map { $dbh->type_info($_) ? $dbh->type_info($_)->{TYPE_NAME} : '?' } @{ $sth->{TYPE} }; |
map { $dbh->type_info($_) ? $dbh->type_info($_)->{TYPE_NAME} : '?' } @{ $sth->{TYPE} }; |
232 |
} |
} |
233 |
|
|
234 |
$row++; # separete queries by one row |
$row++; # separete queries by one row |
235 |
|
warn "# row $row\n"; |
236 |
} |
} |
237 |
} |
} |
238 |
|
|