Added "off" days to XML support
[tpope-extra.git] / perl / sct6
index dc5aad2c2827c82465db6f47e2b3fd6025277c18..b2f5bff3ed7cde50eb26bd2b715268024202e7b1 100755 (executable)
--- a/perl/sct6
+++ b/perl/sct6
@@ -12,8 +12,7 @@ use Date::Calendar::Profiles qw($Profiles);
 use Date::Calendar::Year;
 use HTML::TableExtract;
 use LWP::UserAgent;
-use IO::File;
-use XML::Writer;
+use XML::Simple;
 use vars qw($ua %opts %faculty);
 
 my ($response);
@@ -114,7 +113,7 @@ sub next_class {
     foreach(split(" ",$class{'days'})) {
        push @days, Decode_Day_of_Week($_);
     }
-    my @off = (off_for_holidays(%class),off_for_exams(%class));
+    my @off = @{$row->{'off'}};
     for($today = ($today > $firstday ? $today : $firstday); $today < $lastday; $today++) {
        next unless(grep($_ == Day_of_Week($today->date), @days));
        next if(grep($_ == $today, @off));
@@ -131,6 +130,10 @@ sub next_class {
 
 sub off_for_holidays {
     my %class = @_;
+    my %days = (M => "Mon", T => "Tue", W => "Wed", R => "Thu", F => "Fri", S => "Sat", U => "SU");
+    my $d = $class{'days'};
+    $d =~ s/([MTWRFS])/ $days{$1}/g;
+    $d =~ s/^ //;
     $class{'duration'} =~ /(\d\d\d\d)(\d\d)(\d\d)-(\d\d\d\d)(\d\d)(\d\d)/;
     my $days = Delta_Days($1,$2,$3,$4,$5,$6);
     my $firstday = Date::Calc->new($1,$2,$3);
@@ -138,7 +141,7 @@ sub off_for_holidays {
     my $year = Date::Calendar::Year->new($1, $opts{'holidays'});
     my @holidays=();
     my @days=();
-    foreach(split(" ",$class{days})) {
+    foreach(split(" ",$d)) {
        $days[Decode_Day_of_Week($_)] = 1;
     }
     foreach ($year->search("")) {
@@ -188,8 +191,7 @@ sub capitalize {
     s/\bUs\b/US/g;
     s/ (And|For|Of|Or|The|To|With) / \l$1 /g;
     s/\b(Mc)([a-z])/$1\u$2/g;
-    s/\bTba\b/TBA/g;
-    s/\b(Tcp\/Ip|Pc)\b/\U$&/g;
+    s/\b(Tcp\/Ip|Pc|Tba)\b/\U$&/g;
     s/\bThru\b/Through/g;
     s/\bAcct\b/Accounting/g;
     s/\bAmer\b/American/g;
@@ -204,6 +206,7 @@ sub capitalize {
     s/\bLit\b/Literature/g;
     s/\bPrin\b/Principles/g;
     s/\bBus\b/Business/g;
+    s/\bSyst\b/Sys/g;
     return $_;
 }
 
@@ -274,9 +277,9 @@ sub get_schedule {
            $row->[4] =~ s/ - /-/;
            $row->[3] =~ s/(ON|MAIN) CAMPUS\n|Palo Alto Building \d* //ig;
            $row->[3] =~ s/\n/ /g;
-           push @schedule, {
+           my %class = (
                id => $classid,
-               title => $title,
+               title => capitalize($title),
                instructor => $row->[6],
                days => $row->[2],
                #'time' => $row->[1],
@@ -284,7 +287,10 @@ sub get_schedule {
                end => $end || undef,
                duration => $row->[4],
                location => $row->[3],
-           };
+           );
+           my @off = ( map { "$_"; } (off_for_holidays(%class), off_for_exams(%class)));
+           $class{'off'} = [ @off ] if (@off);
+           push @schedule, \%class;
        }
     }
     return @schedule;
@@ -298,7 +304,7 @@ sub get_faculty_email {
            chomp;
            m/"([^"]*)",([^,]*),([^,]*)/; # "
            ($name, $email, $school) = ($1, $2, $3);
-           $name =~ s/^([^,]*), ([^,]*)(.*)$/$2 $1$3/;
+           #$name =~ s/^([^,]*), ([^,]*)(.*)$/$2 $1$3/;
            $name =~ s/ [A-Z]\.//g;
            $name = lc $name;
            $name =~ s/\W//g;
@@ -346,12 +352,13 @@ sub do_mhc_schedule {
        @mhc = @mhc2;
     }
     foreach my $row (@schedule) {
+       #my @day = map { "!" . $_ } (off_for_holidays(%$row),off_for_exams(%$row));
+       my @day = @{$row->{'off'}};
        map {s/\n/-/g if defined; $_} %$row;
        my $id=generate_id($row->{'id'});
        $row->{'days'} =~ s/([MTWRFS])/ $days{$1}/g;
        $row->{'days'} =~ s/^ //;
        #$row->{'duration'} =~ s/(\d\d)-(\d\d)-(\d\d)/20$3$1$2/g;
-       my @day = map { "!" . $_ } (off_for_holidays(%$row),off_for_exams(%$row));
        $current = "";
        #print "# $id\n";
        $current .= "X-SC-Subject: " . $row->{'title'} . "\n";
@@ -364,7 +371,7 @@ sub do_mhc_schedule {
        $current .= "X-SC-Alarm: 15 minutes\n";
        $current .= "X-SC-Record-Id: <".$row->{'id'}."\@from.sctweb>\n";
        if(defined($file) && (-d $file)) {
-           $row->{'instructor'} =~ s/^([^,]*), ([^,]*)(.*)/$2 $1$3/;
+           #$row->{'instructor'} =~ s/^([^,]*), ([^,]*)(.*)/$2 $1$3/;
            my $email = get_faculty_email $row->{'instructor'};
            $row->{'instructor'} = '"' . $row->{'instructor'} . '" <'. ($email || ($1 || "unknown") . "\@from.sctweb") . ">";
            $row->{'duration'} =~ /^(\d\d\d\d)(\d\d)(\d\d)-\d{8}$/;
@@ -416,7 +423,7 @@ sub do_csv_schedule {
        my $id=generate_id($row->{'id'});
        $row->{'days'} =~ s/([MTWRFS])/ $days{$1}/g;
        $row->{'days'} =~ s/^ //;
-       $row->{'instructor'} =~ s/^([^,]*), ([^,]*)/$2 $1/;
+       #$row->{'instructor'} =~ s/^([^,]*), ([^,]*)(.*)$/$2 $1$3/;
        my $next = next_class(%$row);
        $current = "";
        #print "# $id\n";
@@ -439,11 +446,11 @@ sub do_vcalendar_schedule {
     print "BEGIN:VCALENDAR\r\nVERSION:1.0\r\n";
     foreach my $row (@schedule) {
        map { s/\n/-/g; $_} %$row;
-       $row->{'instructor'} =~ s/^([^,]*), ([^,]*)/$2 $1/;
+       #$row->{'instructor'} =~ s/^([^,]*), ([^,]*)/$2 $1/;
        $row->{'days'} =~ s/([MTWRFS])/ $days{$1}/g;
        $row->{'days'} =~ s/^ //;
        #$row->{'duration'} =~ s/(\d\d)-(\d\d)-(\d\d)/20$3$1$2/g;
-       my @day = (off_for_holidays(%$row),off_for_exams(%$row));
+       my @day = @{$row->{'off'}};
        my $day = "";
        if(exists($day[0])) {
            $day = join(";", @day);
@@ -481,25 +488,15 @@ sub do_vcalendar_schedule {
 
 sub do_xml_schedule {
     my $file = shift if (defined $_[0] and $_[0] !~ /^\d/);
-    my ($output, $writer);
+    my $schedule = { class => [ get_schedule(@_) ] };
+    my $xml = XMLout($schedule, NoAttr => 1, RootName => 'schedule');
     if($file) {
-       $output = new IO::File(">$file");
-       $writer = new XML::Writer(OUTPUT => $output, DATA_MODE => 1);
+       open FH, ">$file";
+       print FH $xml;
+       close FH;
     } 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");
+       print $xml;
     }
-    $writer->endTag("schedule");
-    $writer->end();
-    close $output if $file;
 }
 
 sub do_html_schedule {
@@ -511,8 +508,7 @@ sub do_html_schedule {
     print "</th></tr>\n";
        foreach my $row (@schedule) {
            map { s/\n/<br \/>/g; $_} %$row;
-           $row->{'time'} =~ s/-/-<wbr \/>/;
-           $row->{'duration'} =~ s/-/\n/g;
+           $row->{'duration'} =~ s/-/<br \/>/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");
@@ -520,7 +516,7 @@ sub do_html_schedule {
            print '<span class="sectionid">' .$row->{'id'}. '</span><br />';
            print '<span class="coursetitle">' .$row->{'title'}. '</span></td>';
            my $instructor = $row->{'instructor'};
-           $instructor =~ s/^([^,]*), ([^,]*)(.*)/$2 $1$3/;
+           #$instructor =~ s/^([^,]*), ([^,]*)(.*)/$2 $1$3/;
            my $email = get_faculty_email($instructor);
            #if($email) {
                #print '<td><a href="mailto:' . $email . '">'
@@ -600,7 +596,7 @@ sub do_transcripts {
     }
 }
 
-if ($arg eq "-s") {
+if ($arg eq "-s" || $arg eq "-h") {
     do_html_schedule(@ARGV);
 } elsif ($arg eq "-m") {
     do_mhc_schedule(@ARGV);
@@ -608,14 +604,14 @@ 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);
 } elsif ($arg eq "-t") {
     die "Transcript support currently broken.\n";
     do_transcripts;
+} elsif ($arg eq "-x" || 1) {
+    do_xml_schedule(@ARGV);
 }
 
 $ua->get("${url}twbkwbis.P_Logout"); # Logout