X-Git-Url: http://lab.mitty.jp/git/?a=blobdiff_plain;f=Dev%2Ftwitter%2Fdump_timeline.pl;h=4e2db3ce3ac3021a7a02fc4ad3b72ed7058cfbfd;hb=HEAD;hp=36fd046f57ea3adf0f9d7bcb20617ed1939146e3;hpb=a7e2c5332b7738094d6de3c73c7393bddcf3db9f;p=lab.git

diff --git a/Dev/twitter/dump_timeline.pl b/Dev/twitter/dump_timeline.pl
index 36fd046..4e2db3c 100755
--- a/Dev/twitter/dump_timeline.pl
+++ b/Dev/twitter/dump_timeline.pl
@@ -9,43 +9,81 @@ use utf8;
 # it throws a Net::Twitter::Lite::Error object. 
 # You can catch and process these exceptions by using eval blocks and testing $@
 ## from http://search.cpan.org/perldoc?Net::Twitter::Lite#ERROR_HANDLING
-use Net::Twitter::Lite;
+use Net::Twitter::Lite::WithAPIv1_1;
 use FindBin qw($Bin);
 use YAML::Tiny;
 use Data::Dumper;
 use Encode;
 
-if (@ARGV < 1) {
-    die "usage: $0 screen_name [number_of_pages|all [dump]]\n";
+my $help = sub {
+    die <<EOM;
+usage: $0
+    [{u}ser_timeline(default)|{m}sentions|{s}earch
+        [screen_name
+            [number_of_pages|all
+                [dump]
+            ]
+        ]
+    ]
+EOM
+};
+if ($ARGV[0] && ($ARGV[0] eq '--help' || $ARGV[0] eq '-h') ) {
+    &{$help};
 }
-my $screen_name = $ARGV[0];
-my $pages = $ARGV[1] || 1;
+
+my $method = $ARGV[0] || 'user_timeline';
+my $screen_name = $ARGV[1] || '';
+my $pages = $ARGV[2] || 1;
 if ($pages eq 'all') {
     $pages = -1;
 }
-my $dump = $ARGV[2] || 0;
+my $dump = $ARGV[3] || 0;
 
 my $conf = loadconf("$Bin/config.yml");
 if (! defined $conf) {
-    die "$0: cannot parse config file.\n";
+    die "$0: cannot parse config file.";
 }
 
 my $bot = login($conf);
 if (! $bot->authorized) {
-    die "$0: this client is not yet authorized.\n";
+    die "$0: this client is not yet authorized.";
 }
 
 
 eval {
     my $page = 0;
-    while ($pages - $page && $page <= 160) {
+    while ($pages - $page && $page <= 20) {
         $page++;
-        my $res = $bot->user_timeline(
-            {
-                screen_name => $screen_name,
-                page        => $page,
+        
+        my $param = ($screen_name)
+            ? { page => $page, screen_name => $screen_name, count => 200, }
+            : { page => $page, count => 200, }
+        ;
+    
+        my $res;
+        if ($method eq 'user_timeline' || $method eq 'u') {
+            $res = $bot->user_timeline($param);
+        }
+        elsif ($method eq 'mentions' || $method eq 'm') {
+            $res = $bot->mentions($param);
+        }
+        elsif ($method eq 'search' || $method eq 's') {
+            my $key;
+            foreach my $word (@{ $conf->{hashtag} }) {
+                if ($key) {
+                    $key .= " OR $word";
+                }
+                else {
+                    $key = $word;
+                }
             }
-        );
+            $param->{q} = $key;
+            $res = $bot->search($param)->{results};
+        }
+        else {
+            warn "$0: unknown method '$method'";
+            &{$help};
+        }
         
         if ($dump) {
             foreach my $line (split /\n/, Dumper $res) {
@@ -56,12 +94,16 @@ eval {
         else {
             foreach my $status (@{$res}) {
                 my $text = "";
-                $text .= $status->{user}{name};
+                $text .= "(". $status->{id} . ") ";
+                $text .= ($status->{user}{screen_name}) ?
+                    $status->{user}{screen_name} : $status->{from_user};
+                $text .= "|";
+                $text .= ($status->{user}{name}) ?
+                    $status->{user}{name} : $status->{from_user_name};
                 $text .= " [" . $status->{created_at} . "]";
-                $text .= " (". $status->{id} . ")";
-                $text .= " ". encode('utf8', $status->{text});
+                $text .= " ".  $status->{text};
                 $text =~ s/\n//;
-                print $text, "\n";
+                print encode('utf8', $text), "\n";
             }
         }
     }
@@ -69,7 +111,6 @@ eval {
 if ($@) {
     evalrescue($@);
 }
-print "done\n";
 
 
 sub loadconf {
@@ -82,22 +123,24 @@ sub loadconf {
     my $yaml = YAML::Tiny->read($file);
     
     if ($!) {
-        warn "$0: '$file' $!\n";
+        warn "$0: '$file' $!";
     }
     
     return $yaml->[0];
 }
 
 sub login {
-    # make Net::Twitter::Lite object and login
+    # make Net::Twitter::Lite::WithAPIv1_1 object and login
     #   param   => hash object of configration
-    #   ret     => Net::Twitter::Lite object
+    #   ret     => Net::Twitter::Lite::WithAPIv1_1 object
     
     my $conf = shift @_;
     
-    my $bot = Net::Twitter::Lite->new(
+    my $bot = Net::Twitter::Lite::WithAPIv1_1->new(
         consumer_key    => $conf->{consumer_key},
         consumer_secret => $conf->{consumer_secret},
+        legacy_lists_api => 0,
+        ssl             => 1,
     );
     
     $bot->access_token($conf->{access_token});
@@ -114,12 +157,18 @@ sub evalrescue {
     if (blessed $@ && $@->isa('Net::Twitter::Lite::Error')) {
         warn $@->error;
         if ($@->twitter_error) {
-            my %twitter_error = %{$@->twitter_error};
-            map {
-                $twitter_error{"$_ => "} = $twitter_error{$_} . "\n";
-                delete $twitter_error{$_}
-            } keys %twitter_error;
-            warn join("", %twitter_error);
+            my $twitter_error = $@->twitter_error;
+            if (defined $twitter_error->{errors}) {
+                foreach my $error (@{$twitter_error->{errors}}) {
+                    warn "code => "   , $error->{code}, "\n";
+                    warn "message => ", $error->{message}, "\n";
+                }
+            }
+            else {
+                # unknown HASH structure
+                use Data::Dumper;
+                warn Dumper $twitter_error;
+            }
         }
     }
     else {