Added XML output

[tpope-extra.git] / perl / sct6

diff --git a/perl/sct6 b/perl/sct6
index 9178a710bb1cd30260537d11f12610dbdb728871..dc5aad2c2827c82465db6f47e2b3fd6025277c18 100755 (executable)
--- a/perl/sct6
+++ b/perl/sct6
@@ -7,11 +7,13 @@
 # You'll need to change the url below if you go anywhere but TAMUK.
 
 use strict;
-use LWP::UserAgent;
-use HTML::TableExtract;
 use Date::Calc::Object qw(Day_of_Week Decode_Day_of_Week Decode_Month Week_of_Year Monday_of_Week Day_of_Week_Abbreviation Delta_Days Add_Delta_Days Nth_Weekday_of_Month_Year Gmtime Mktime);
 use Date::Calendar::Profiles qw($Profiles);
 use Date::Calendar::Year;
+use HTML::TableExtract;
+use LWP::UserAgent;
+use IO::File;
+use XML::Writer;
 use vars qw($ua %opts %faculty);
 
 my ($response);
@@ -269,7 +271,7 @@ sub get_schedule {
            $end =~ s/ ?[AP]M//;
            $row->[4] =~ s/([A-Za-z]{3,9})/(Decode_Month($1)<10?"0":"").Decode_Month($1)/eg;
            $row->[4] =~ s/(\d\d?) (\d\d), (\d\d\d\d)/$3$1$2/g;
-           $row->[4] =~ s/ - /\n/;
+           $row->[4] =~ s/ - /-/;
            $row->[3] =~ s/(ON|MAIN) CAMPUS\n|Palo Alto Building \d* //ig;
            $row->[3] =~ s/\n/ /g;
            push @schedule, {
@@ -277,16 +279,11 @@ sub get_schedule {
                title => $title,
                instructor => $row->[6],
                days => $row->[2],
-               'time' => $row->[1],
+               #'time' => $row->[1],
                begin => $begin,
                end => $end || undef,
                duration => $row->[4],
                location => $row->[3],
-               #session => $row->[1],
-               #credits => $row->[3],
-               #callnumber => $row->[4],
-               #gradetype => $row->[5],
-               #message => $row->[10],
            };
        }
     }
@@ -482,6 +479,29 @@ sub do_vcalendar_schedule {
     print "END:VCALENDAR\r\n";
 }
 
+sub do_xml_schedule {
+    my $file = shift if (defined $_[0] and $_[0] !~ /^\d/);
+    my ($output, $writer);
+    if($file) {
+       $output = new IO::File(">$file");
+       $writer = new XML::Writer(OUTPUT => $output, DATA_MODE => 1);
+    } else {
+       $writer = new XML::Writer(DATA_MODE => 1);
+    }
+    my @schedule = get_schedule(@_);
+    $writer->startTag("schedule");
+    foreach my $class (@schedule) {
+       $writer->startTag("class");
+       foreach my $key (sort keys %$class) {
+           $writer->dataElement($key, $class->{$key});
+       }
+       $writer->endTag("class");
+    }
+    $writer->endTag("schedule");
+    $writer->end();
+    close $output if $file;
+}
+
 sub do_html_schedule {
     my @showheaders = ("Section ID/Title", "Instructor", "Days", "Time", "Duration", "Location");
     my $shade = "dark";
@@ -492,6 +512,7 @@ sub do_html_schedule {
        foreach my $row (@schedule) {
            map { s/\n/<br \/>/g; $_} %$row;
            $row->{'time'} =~ s/-/-<wbr \/>/;
+           $row->{'duration'} =~ s/-/\n/g;
            $row->{'duration'} =~ s/\d\d(\d\d)(\d\d)(\d\d)/$2-$3-$1/g;
            print '<tr class="'.$shade.'">';
            $shade = ($shade eq "dark"?"light":"dark");
@@ -508,7 +529,7 @@ sub do_html_schedule {
                print "<td>" . $row->{'instructor'} . "</td>";
            #}
            print "<td>" . $row->{'days'} . "</td>";
-           print "<td>" . $row->{'time'} . "</td>";
+           print "<td>" . $row->{'begin'}."-<wbr />".$row->{'end'} . "</td>";
            print "<td>" . $row->{'duration'} . "</td>";
            print "<td>" . $row->{'location'} . "</td>";
            print "</tr>\n";
@@ -587,6 +608,8 @@ if ($arg eq "-s") {
     do_csv_schedule(@ARGV);
 } elsif ($arg eq "-v") {
     do_vcalendar_schedule(@ARGV);
+} elsif ($arg eq "-x") {
+    do_xml_schedule(@ARGV);
 } elsif ($arg eq "-g") {
     die "Grade support currently broken.\n";
     do_html_grades(@ARGV);