perl/sct6

   1 #!/usr/bin/perl -w
   2 # $Id$
   3 # -*- perl -*- vim: ft=perl sw=4 sts=4
   4
   5 # Brief usage instructions:
   6 # Create a ~/.sct6rc that has SID=yourssn and PIN=yourpin
   7 # You'll need to change the url below if you go anywhere but TAMUK.
   8
   9 # Update: This tool now only outputs in XML.  For other formats, convert
  10 # with schedproc
  11
  12 use strict;
  13 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);
  14 use Date::Calendar::Profiles qw($Profiles);
  15 use Date::Calendar::Year;
  16 use HTML::TableExtract;
  17 use LWP::UserAgent;
  18 use XML::Simple;
  19 use vars qw($ua %opts %faculty);
  20
  21 my ($response);
  22
  23 $opts{'holidays'} = { # %{$Profiles->{'US-TX'}},
  24     "Martin Luther King Day"    => "3/Mon/Jan",
  25     "Good Friday"               => "-2",
  26     "Spring Break Monday"       => \&Spring_Break,
  27     "Spring Break Tuesday"      => \&Spring_Break,
  28     "Spring Break Wednesday"    => \&Spring_Break,
  29     "Spring Break Thursday"     => \&Spring_Break,
  30     "Spring Break Friday"       => \&Spring_Break,
  31     "Spring Break Saturday"     => \&Spring_Break,
  32     "Study Day"                 => "4/Thu/Apr", # ?
  33     "Memorial Day"              => "5/Mon/May",
  34     "Independence Day"          => \&US_Independence,
  35     "Labor Day"                 => \&US_Labor,
  36     #"Columbus Day"              => "2/Mon/Oct",
  37     "Thanksgiving Day"          => "4/Thu/Nov",
  38     "Thanksgiving Friday"       => \&Thanksgiving_Friday,
  39 };
  40
  41 sub Spring_Break {
  42     my($year,$label) = @_;
  43     $label =~ s/^Spring Break //;
  44     return( Add_Delta_Days(
  45             Nth_Weekday_of_Month_Year($year,1,1,1),
  46             7*(11-1) # This is for the 11th Monday of the year
  47             +Decode_Day_of_Week($label)-1) );
  48 }
  49
  50 sub Thanksgiving_Friday {
  51     my($year,$label) = @_;
  52     return( Add_Delta_Days(Nth_Weekday_of_Month_Year($year,11,4,4), 1) );
  53 }
  54 sub US_Independence # Fourth of July
  55 {
  56     my($year,$label) = @_;
  57     return( &Date::Calendar::Profiles::Nearest_Workday($year,7,4) );
  58 }
  59 sub US_Labor # First Monday after the first Sunday in September
  60 {
  61     my($year,$label) = @_;
  62     return( Add_Delta_Days(
  63         Nth_Weekday_of_Month_Year($year,9,7,1), +1) );
  64 }
  65
  66 my $config = $ENV{HOME} . "/.sct6rc";
  67 if (($ARGV[0] || "") eq '-F') {
  68     shift;
  69     $config = shift;
  70 }
  71
  72 my $arg = "";
  73 $arg = shift if (defined($ARGV[0]) && $ARGV[0] =~ /^-\w$/);
  74
  75 if (-r $config) {
  76     open CONFIG, $config;
  77     while(<CONFIG>) {
  78         s/\#.*//;
  79         next unless m/^([^=]*)=(.*)/;
  80         $opts{$1}=$2;
  81     }
  82     close CONFIG;
  83 }
  84 my $domain = $opts{'domain'} || "as1.tamuk.edu:9003";
  85 my $url = "https://$domain/pls/PROD/";
  86 $url = $opts{'url'} if(defined($opts{'url'}));
  87
  88 $ua = LWP::UserAgent->new;
  89 $ua->timeout(10);
  90 $ua->env_proxy;
  91 $ua->cookie_jar( {} );
  92 $response = $ua->get("${url}twbkwbis.P_WWWLogin") or die "$!";
  93 $response = $ua->post("${url}twbkwbis.P_ValLogin", { sid => $opts{SID}, PIN => $opts{PIN} }) or die "$!";
  94
  95 my %hiddenjunk;
  96 {
  97     my $content = $response->content;
  98     while($content =~ /<input type="hidden" name="(.*?)" value="(.*?)">/gi) {
  99         $hiddenjunk{$1} = $2;
 100     }
 101 }
 102 $hiddenjunk{usage_btn} = "Continue";
 103 $response = $ua->post("${url}twbkwbis.P_FirstMenu", \%hiddenjunk);
 104 die $response->status_line unless $response->is_success;
 105
 106 $response = $ua->get("${url}bwskflib.P_SelDefTerm"); # Valid terms
 107 die $response->status_line unless $response->is_success;
 108 my @terms = grep {s/^<option value="([^"]*)">.*/$1/i} (split( /\r?\n/, $response->content)); # "
 109 die "Site down. Try again later.\n" unless (@terms);
 110
 111 sub next_class {
 112     my %class = @_;
 113     $class{'duration'} =~ /(\d\d\d\d)(\d\d)(\d\d)-(\d\d\d\d)(\d\d)(\d\d)/;
 114     my $days = Delta_Days($1,$2,$3,$4,$5,$6);
 115     my $firstday = Date::Calc->new($1,$2,$3);
 116     my $lastday = Date::Calc->new($4,$5,$6);
 117     my $today = Date::Calc->new(Date::Calc->localtime(time+3600*6)->date);
 118     my @days=();
 119     foreach(split(" ",$class{'days'})) {
 120         push @days, Decode_Day_of_Week($_);
 121     }
 122     my @off = @{$class{'off'}};
 123     for($today = ($today > $firstday ? $today : $firstday); $today < $lastday; $today++) {
 124         next unless(grep($_ == Day_of_Week($today->date), @days));
 125         next if(grep($_ == $today, @off));
 126         return $today;
 127     }
 128     $today = Date::Calc->new(Date::Calc->gmtime->date);
 129     for($today = ($today < $lastday ? $today : $lastday); $today > $firstday; $today--) {
 130         next unless(grep($_ == Day_of_Week($today->date), @days));
 131         next if(grep($_ == $today, @off));
 132         return $today;
 133     }
 134     return undef;
 135 }
 136
 137 sub off_for_holidays {
 138     my %class = @_;
 139     my %days = (M => "Mon", T => "Tue", W => "Wed", R => "Thu", F => "Fri", S => "Sat", U => "SU");
 140     my $d = $class{'days'};
 141     $d =~ s/([MTWRFS])/ $days{$1}/g;
 142     $d =~ s/^ //;
 143     $class{'duration'} =~ /(\d\d\d\d)(\d\d)(\d\d)-(\d\d\d\d)(\d\d)(\d\d)/;
 144     my $days = Delta_Days($1,$2,$3,$4,$5,$6);
 145     my $firstday = Date::Calc->new($1,$2,$3);
 146     my $lastday = Date::Calc->new($4,$5,$6);
 147     my $year = Date::Calendar::Year->new($1, $opts{'holidays'});
 148     my @holidays=();
 149     my @days=();
 150     foreach(split(" ",$d)) {
 151         $days[Decode_Day_of_Week($_)] = 1;
 152     }
 153     foreach ($year->search("")) {
 154         my $good=1;
 155         #foreach my $x ($year->labels($_)) {
 156         #    $good=0 if ($x =~ /Veteran/ or $x =~ /President/);
 157         #}
 158         next unless ($year->is_full($_) && $_>=$firstday && $_<=$lastday);
 159         push @holidays, $_
 160             if (defined($days[Day_of_Week($_->date)]) && $good > 0);
 161     }
 162     return wantarray ? @holidays : "@holidays";
 163 }
 164
 165 sub off_for_exams {
 166     my %class = @_;
 167     my $time = $class{'begin'} . "-" . $class{'end'};
 168     $class{'duration'} =~ /\d{8}-(\d\d\d\d)(\d\d)(\d\d)/;
 169     my $lastday = Date::Calc->new($1,$2,$3);
 170     return wantarray ? () : undef unless (Day_of_Week($lastday->date) == 7);
 171     my $beginning="(08:00-09:15|11:00-12:15|14:00-15:15|17:00-18:15|17:30-18:45|20:00-21:15|20:30-21:45)";
 172     my $ending   ="(06:30-07:45|09:30-10:45|12:30-13:45|15:30-16:45|18:30-19:45|19:00-20:15)";
 173     my $days = join(" ",map {Decode_Day_of_Week($_)} split(/ /, $class{'days'}));
 174     if($days eq "1 3") {
 175         if($time =~ /$beginning/) {
 176             return $lastday-4;
 177         } elsif($time =~ /$ending/) {
 178             return $lastday-6;
 179         }
 180     } elsif($days eq "2 4") {
 181         if($time =~ /$beginning/) {
 182             return $lastday-3;
 183         } elsif($time =~ /$ending/) {
 184             return $lastday-5;
 185         }
 186     } elsif($days eq "5") {
 187         return $lastday-2;
 188     }
 189     return wantarray ? () : "";
 190 }
 191
 192 sub capitalize {
 193     local $_ = shift || "";
 194     s/ +$//;
 195     s/\b([A-Z])([A-Z]*)\b/$1\L$2/g;
 196     s/\b(I)(i*)\b/$1\U$2/g;
 197     s/\bUs\b/US/g;
 198     s/ (And|For|Of|Or|The|To|With) / \l$1 /g;
 199     s/\b(Mc)([a-z])/$1\u$2/g;
 200     s/\b(Tcp\/Ip|Pc|Tba)\b/\U$&/g;
 201     s/\bThru\b/Through/g;
 202     s/\bAcct\b/Accounting/g;
 203     s/\bAmer\b/American/g;
 204     s/\bChem\b/Chemistry/g;
 205     s/\bComp\b/Composition/g;
 206     s/\bFed\b/Federal/g;
 207     s/\bGen\b/General/g;
 208     s/\bIntro\b/Introduction/g;
 209     s/\bPrgm\b/Programming/g;
 210     s/\bOp Sys\b/Operating System/g;
 211     #s/\bGovt\b/Government/g;
 212     s/\bLit\b/Literature/g;
 213     s/\bPrin\b/Principles/g;
 214     s/\bBus\b/Business/g;
 215     s/\bSyst\b/Sys/g;
 216     return $_;
 217 }
 218
 219 sub get_schedule_terms {
 220     my (@sterms, @a);
 221     @a=('10','20','30','40');   # Fall, Spring, Summer I, Summer II
 222                                 # Last 3 are currently guesses
 223     my @localtime=localtime();
 224     if ($localtime[4]<3) { # Through Mar 31
 225         @sterms=((1900+$localtime[5]).$a[1]);
 226     } elsif ($localtime[4]<5) { # Through May 31
 227         @sterms=((1900+$localtime[5]).$a[1],(1900+$localtime[5]).$a[2]);
 228     } elsif ($localtime[4]<7) { # through July 31
 229         @sterms=((1900+$localtime[5]).$a[2],(1900+$localtime[5]).$a[3]);
 230     } elsif ($localtime[4]<8) { # through Aug 31
 231         @sterms=((1900+$localtime[5]).$a[3],(1901+$localtime[5]).$a[0]);
 232     } elsif ($localtime[4]<10) { # through Oct 31
 233         @sterms=((1901+$localtime[5]).$a[0]);
 234     } else {
 235         @sterms=((1901+$localtime[5]).$a[0],(1901+$localtime[5]).$a[1]);
 236     }
 237     return @sterms;
 238 }
 239
 240 sub get_schedule {
 241     my @readheaders = ("Type", "Time", "Days", "Where", "Date Range", "Schedule Type", "Instructors");
 242     my @class;
 243     my $te = new HTML::TableExtract( headers => [ @readheaders ] );
 244     my (@schedule, @terms, $classid, $title, $begin, $end, $times, $days, $session);
 245     @terms = get_schedule_terms();
 246     foreach (@_ ? @_ : @terms) {
 247         $response = $ua->get("${url}bwskfshd.P_CrseSchdDetl?term_in=$_" );
 248         die $response->status_line unless $response->is_success;
 249         $te->parse($response->content);
 250         foreach my $l (split (/\n/, $response->content)) {
 251             next unless $l =~ s/.*<CAPTION class=[^>]*>(.* - .... \d\d\d\d \d\d\d)<\/caption>.*/$1/i;
 252             push @class, $l;
 253         }
 254     }
 255     foreach my $ts ($te->table_states) {
 256         foreach my $row ($ts->rows) {
 257             #map { s/\xa0//g; $_} @$row;
 258             #$row->[0] =~ s/.*launchWebCT\("([^"]*)"\).*/$1/s;
 259             #$row->[0] =~ s/(.*) ?<[Bb][Rr][^>]*>(.*)/capitalize($2)/eg;
 260             $classid = (shift @class);
 261             $title = $classid;
 262             $classid =~ s/.* - //;
 263             $classid =~ s/ /-/g;
 264             $title =~ s/ - .*//;
 265             $row->[6] =~ s/\b([A-Z]r?)$/$1./;
 266             $row->[6] = capitalize($row->[6]);
 267             $row->[6] =~ s/ *\([A-Z]\)//;
 268             $row->[1] =~ s/ ?([ap])m/\u$1M/g;
 269             $row->[1] =~ s/ - /-/;
 270             ($begin, $end) = split(/ ?- ?/, $row->[1]);
 271             $begin =~ s/^(\d):/0$1:/;
 272             $end =~ s/^(\d):/0$1:/;
 273             if (($begin=~/PM$/ && $begin!~/^12/)||($begin=~/^12:..AM/)) {
 274                 $begin =~ s/^(\d?\d)/($1+12)%24/e;
 275             }
 276             if (($end=~/PM$/ && $end!~/^12/)||($end=~/^12:..AM/)) {
 277                 $end =~ s/^(\d?\d)/($1+12)%24/e;
 278             }
 279             $begin =~ s/ ?[AP]M//;
 280             $end =~ s/ ?[AP]M//;
 281             $row->[4] =~ s/([A-Za-z]{3,9})/(Decode_Month($1)<10?"0":"").Decode_Month($1)/eg;
 282             $row->[4] =~ s/(\d\d?) (\d\d), (\d\d\d\d)/$3$1$2/g;
 283             $row->[4] =~ s/ - /-/;
 284             $row->[3] =~ s/(ON|MAIN) CAMPUS\n|Palo Alto Building \d* //ig;
 285             $row->[3] =~ s/\n/ /g;
 286             my %class = (
 287                 id => $classid,
 288                 title => capitalize($title),
 289                 instructor => $row->[6],
 290                 days => $row->[2],
 291                 #'time' => $row->[1],
 292                 begin => $begin,
 293                 end => $end || undef,
 294                 duration => $row->[4],
 295                 location => $row->[3],
 296             );
 297             my @off = ( map { "$_"; } (off_for_holidays(%class), off_for_exams(%class)));
 298             $class{'off'} = [ @off ] if (@off);
 299             push @schedule, \%class;
 300         }
 301     }
 302     return @schedule;
 303 }
 304
 305 sub do_xml_schedule {
 306     my $file = shift if (defined $_[0] and $_[0] !~ /^\d/);
 307     my $schedule = { class => [ get_schedule(@_) ] };
 308     my $xml = XMLout($schedule, NoAttr => 1, RootName => 'schedule');
 309     if($file) {
 310         open FH, ">$file";
 311         print FH $xml;
 312         close FH;
 313     } else {
 314         print $xml;
 315     }
 316 }
 317
 318 #sub do_xml_grades {
 319 #    print "<grades>\n</grades>\n";
 320 #}
 321
 322 sub do_xml_grades {
 323     #my @readheaders = ("Subject", "Course", "Section", "Course Title", "Final Grade", "Attempted", "Earned", "GPA Hours", "Quality Points");
 324     my @readheaders = ("Subject", "Course", "Section", "CTitle", "Final Grade", "Attempted", "Earned", "GPA Hours", "Quality Points");
 325     my ($row, $lastrow);
 326     my $te = new HTML::TableExtract( headers => [ @readheaders ] );
 327     my $response = $ua->get("$url/bwskogrd.P_ViewTermGrde"); # Valid grading terms
 328     die $response->status_line unless $response->is_success;
 329     #my @terms = reverse grep {s/^<option value="([^"]*)">.*/$1/} (split( /\r\n/, $response->content));
 330     my @grades = ();
 331     my @terms = ();
 332     my $year = Date::Calc->localtime->year;
 333     for(my $y=$year-3;$y<=$year+1;$y++) { push @terms, ($y.10,$y.20,$y.30,$y.40); }
 334     foreach(@_ ? @_ : @terms) {
 335         $response = $ua->get("$url/bwskogrd.P_ViewGrde?term_in=".($_) );
 336         die $response->status_line unless $response->is_success;
 337         my $content = $response->content;
 338         $content =~ s/Course Title/CTitle/g;
 339         $te->parse($content);
 340     }
 341     foreach my $ts ($te->table_states) {
 342         foreach my $row ($ts->rows) {
 343             map { s/^\s+//; s/\xa0|\r//g; s/\n/<br \/>/g; $_} @$row;
 344             $row->[3] = capitalize($row->[3]);
 345             $row->[3] =~ s/&/&amp;/g;
 346             #$row->[0] =~ s/Cumulative through/Cumulative: through/;
 347             #$row->[0] =~ s/Cumulative:(.*) (\d{4}) (.*)/Cumulative:$1 $3 $2/;
 348             $lastrow=$row;
 349             next if $row->[0] =~ /:|Current Term/;
 350             push @grades, {
 351                 id => $row->[0]."-".$row->[1]."-".$row->[2],
 352                 title => $row->[3],
 353                 grade => $row->[4],
 354                 attempted => $row->[5],
 355                 earned => $row->[6],
 356                 hours => $row->[7],
 357                 points => $row->[8]
 358             };
 359         }
 360     }
 361     $lastrow->[0] =~ s/Cumulative:? *through *//i if($lastrow->[0]);
 362     my $cumulative = { term => capitalize($lastrow->[0]),
 363                        gpa => $lastrow->[6],
 364                        earned => $lastrow->[3],
 365                        hours => $lastrow->[4],
 366                        points => $lastrow->[5]
 367                    };
 368     my $xml = XMLout({class => [@grades], cumulative => []}, NoAttr => 1, RootName => 'grades');
 369     print $xml;
 370     return;
 371 }
 372
 373 sub do_transcripts {
 374     print "Not implemented.\n";
 375 }
 376
 377 if ($arg eq "-g") {
 378     do_xml_grades(@ARGV);
 379 } elsif ($arg eq "-t") {
 380     do_transcripts;
 381 } elsif ($arg eq "-x" || $arg eq "-s" || 1) {
 382     do_xml_schedule(@ARGV);
 383 }
 384
 385 $ua->get("${url}twbkwbis.P_Logout"); # Logout