Difference between revisions of "Import CSV data into a wiki"
(no need for perms) |
(Adding $::append variable) |
||
| Line 1: | Line 1: | ||
#!/usr/bin/perl | #!/usr/bin/perl | ||
| − | # {{perl}}{{Category:Robots}} | + | # {{perl}}{{#security:*|sysop}}{{Category:Robots}} |
# - Licenced under LGPL (http://www.gnu.org/copyleft/lesser.html) | # - Licenced under LGPL (http://www.gnu.org/copyleft/lesser.html) | ||
| − | # - | + | # - Authors: [http://www.organicdesign.co.nz/Nad Nad] [http://www.organicdesign.co.nz/Sven Sven] |
# - Source: http://www.organicdesign.co.nz/scraper.pl | # - Source: http://www.organicdesign.co.nz/scraper.pl | ||
# - Started: 2008-03-21 | # - Started: 2008-03-21 | ||
| Line 25: | Line 25: | ||
$::template = 'Record'; | $::template = 'Record'; | ||
$::prefix = ""; | $::prefix = ""; | ||
| + | $::append = 0; | ||
# Parse the job file | # Parse the job file | ||
| Line 37: | Line 38: | ||
if (/^\*?\s*template\s*:\s*(.+?)\s*$/i) { $::template = $1 } | if (/^\*?\s*template\s*:\s*(.+?)\s*$/i) { $::template = $1 } | ||
if (/^\*?\s*prefix\s*:\s*(.+?)\s*$/i) { $::prefix = $1 } | if (/^\*?\s*prefix\s*:\s*(.+?)\s*$/i) { $::prefix = $1 } | ||
| + | if (/^\*?\s*append\s*:\s*(.+?)\s*$/i) { $::append = $1 } | ||
} | } | ||
close JOB; | close JOB; | ||
| Line 100: | Line 102: | ||
$text = wikiRawPage($::wiki,$record[$::title],0); | $text = wikiRawPage($::wiki,$record[$::title],0); | ||
$text .= "\n$tmpl" unless $text =~ s/\{\{$template.+?\}\}/$tmpl/is; | $text .= "\n$tmpl" unless $text =~ s/\{\{$template.+?\}\}/$tmpl/is; | ||
| − | $done = | + | if($append) { |
| − | + | $done = wikiPageAppend($::wiki,$::prefix . $record[$::title],$text,"[[Template:$::template|$::template]] appended using csv2wiki.pl"); | |
| + | } else { | ||
| + | $done = wikiPageEdit($::wiki,$::prefix . $record[$::title],$text,"[[Template:$::template|$::template]] replacement using csv2wiki.pl"); | ||
| + | } | ||
# log a row error if any | # log a row error if any | ||
} | } | ||
close CSV; | close CSV; | ||
Revision as of 10:01, 22 June 2008
- !/usr/bin/perl
- Our Perl scripts.{{#security:*|sysop}}Automated scripts to perform batch automation.
- - Licenced under LGPL (http://www.gnu.org/copyleft/lesser.html)
- - Authors: Nad Sven
- - Source: http://www.organicdesign.co.nz/scraper.pl
- - Started: 2008-03-21
- - API: http://en.wikipedia.org/w/api.php
- Todo
- Make it so that if there is no title then it increments
- $hashref = { $wikitext =~ /\{{3}(.+?)(\|.*?)?\}{3}/g }
require('wiki.pl');
- Job, log and error files
$ARGV[0] or die "No job file specified!"; $ARGV[0] =~ /^(.+?)(\..+?)?$/;
- Set a debug conditional
$::debug = 0;
$::log = "$1.log"; $::err = "$1.err"; $::sep = ','; $::title = 0; $::template = 'Record'; $::prefix = ""; $::append = 0;
- Parse the job file
if (open JOB,'<',$ARGV[0]) { for (<JOB>) { if (/^\*?\s*csv\s*:\s*(.+?)\s*$/i) { $::csv = $1 } if (/^\*?\s*wiki\s*:\s*(.+?)\s*$/i) { $::wiki = $1 } if (/^\*?\s*user\s*:\s*(.+?)\s*$/i) { $::user = $1 } if (/^\*?\s*pass\s*:\s*(.+?)\s*$/i) { $::pass = $1 } if (/^\*?\s*separator\s*:\s*"(.+?)"\s*$/i) { $::sep = $1 } if (/^\*?\s*title\s*:\s*(.+?)\s*$/i) { $::title = $1 } if (/^\*?\s*template\s*:\s*(.+?)\s*$/i) { $::template = $1 } if (/^\*?\s*prefix\s*:\s*(.+?)\s*$/i) { $::prefix = $1 } if (/^\*?\s*append\s*:\s*(.+?)\s*$/i) { $::append = $1 } } close JOB; } else { die "Couldn't parse job file!" }
- Open CSV file and read in headings line
if (open CSV,'<',$::csv) { $_ = <CSV>; /^\s*(.+?)\s*$/; @headings = split /$::sep/i, $1; } else { die "Could not open CSV file!" }
- Log in to the wiki
wikiLogin($::wiki,$::user,$::pass) or exit;
- fetch the template if it exists
$response = $client->get("$wiki?title=Template:$template&action=raw"); if( $response->is_success ) {
$wikitext = $response->content;
# Remove noinclude areas $wikitext =~ s/.+?<\/noinclude>//gs;
# Find all unique {{{parameters}}}
# http://en.wikipedia.org/wiki/Help:Templates#Parameters
while ($wikitext =~ /\{{3}(.+?)(\|.*?)?\}{3}/g ) {
$params{$1} = undef;
}
# Create %{param=index} hash
foreach( $i = 0; $i <= $#headings; $i++ ) {
if(exists($params{$headings[$i]})) {
$params{$headings[$i]} = $i;
}
}
if($::debug) {
print "\@headings: @headings\n";
print "%params: @{[%params]}\n";
}
}
- Get batch size and current number (also later account for n-bots)
- todo: log batch start
- Process the records
$n = 1; while (<CSV>) { /^\s*(.+?)\s*$/; @record = split /$::sep/, $1; $tmpl = "{{$template\n"; $tmpl .= "|$_ = $record[$params{$_}]\n" foreach (keys %params); $tmpl .= "}}"; print "Processing record ".$n++."\n"; if($::debug) { print "\$tmpl = $tmpl\n"; die "[\$::debug set exiting]\n" ; } # Update the record $text = wikiRawPage($::wiki,$record[$::title],0); $text .= "\n$tmpl" unless $text =~ s/\{\{$template.+?\}\}/$tmpl/is; if($append) { $done = wikiPageAppend($::wiki,$::prefix . $record[$::title],$text,"$::template appended using csv2wiki.pl"); } else { $done = wikiPageEdit($::wiki,$::prefix . $record[$::title],$text,"$::template replacement using csv2wiki.pl");
}
# log a row error if any }
close CSV;
