source: lab.git/Dev/twitter/dump_timeline.pl @ 52c76fd

trunk
Last change on this file since 52c76fd was 52c76fd, checked in by mitty <mitty@…>, 12 years ago
  • add search command
  • load search keyword from config.yaml and do OR search

git-svn-id: https://lab.mitty.jp/svn/lab/trunk@164 7d2118f6-f56c-43e7-95a2-4bb3031d96e7

  • Property mode set to 100755
File size: 4.2 KB
Line 
1#! /usr/bin/perl -w
2
3use strict;
4use warnings;
5use utf8;
6
7## IMPORTANT ##
8# When Net::Twitter::Lite encounters a Twitter API error or a network error,
9# it throws a Net::Twitter::Lite::Error object.
10# You can catch and process these exceptions by using eval blocks and testing $@
11## from http://search.cpan.org/perldoc?Net::Twitter::Lite#ERROR_HANDLING
12use Net::Twitter::Lite;
13use FindBin qw($Bin);
14use YAML::Tiny;
15use Data::Dumper;
16use Encode;
17
18my $help = sub {
19    die <<EOM;
20usage: $0
21    [{u}ser_timeline(default)|{r}etweeted_by_me|{m}sentions|{s}earch
22        [screen_name
23            [number_of_pages|all
24                [dump]
25            ]
26        ]
27    ]
28EOM
29};
30if ($ARGV[0] && ($ARGV[0] eq '--help' || $ARGV[0] eq '-h') ) {
31    &{$help};
32}
33
34my $method = $ARGV[0] || 'user_timeline';
35my $screen_name = $ARGV[1] || '';
36my $pages = $ARGV[2] || 1;
37if ($pages eq 'all') {
38    $pages = -1;
39}
40my $dump = $ARGV[3] || 0;
41
42my $conf = loadconf("$Bin/config.yml");
43if (! defined $conf) {
44    die "$0: cannot parse config file.";
45}
46
47my $bot = login($conf);
48if (! $bot->authorized) {
49    die "$0: this client is not yet authorized.";
50}
51
52
53eval {
54    my $page = 0;
55    while ($pages - $page && $page <= 160) {
56        $page++;
57       
58        my $param = ($screen_name)
59            ? { page => $page, screen_name => $screen_name, }
60            : { page => $page, }
61        ;
62   
63        my $res;
64        if ($method eq 'user_timeline' || $method eq 'u') {
65            $res = $bot->user_timeline($param);
66        }
67        elsif ($method eq 'retweeted_by_me' || $method eq 'r') {
68            $res = $bot->retweeted_by_me($param);
69        }
70        elsif ($method eq 'mentions' || $method eq 'm') {
71            $res = $bot->mentions($param);
72        }
73        elsif ($method eq 'search' || $method eq 's') {
74            my $key;
75            foreach my $word (@{ $conf->{hashtag} }) {
76                if ($key) {
77                    $key .= " OR $word";
78                }
79                else {
80                    $key = $word;
81                }
82            }
83            $param->{q} = $key;
84            $res = $bot->search($param)->{results};
85        }
86        else {
87            warn "$0: unknown method '$method'";
88            &{$help};
89        }
90       
91        if ($dump) {
92            foreach my $line (split /\n/, Dumper $res) {
93                if ($line =~ /undef/) { next; }
94                print $line, "\n";
95            }
96        }
97        else {
98            foreach my $status (@{$res}) {
99                my $text = "";
100                $text .= "(". $status->{id} . ") ";
101                $text .= $status->{user}{screen_name} . "|";
102                $text .= $status->{user}{name};
103                $text .= " [" . $status->{created_at} . "]";
104                $text .= " ".  $status->{text};
105                $text =~ s/\n//;
106                print encode('utf8', $text), "\n";
107            }
108        }
109    }
110};
111if ($@) {
112    evalrescue($@);
113}
114
115
116sub loadconf {
117    # load configration data from yaml formatted file
118    #   param   => scalar string of filename
119    #   ret     => hash object of yaml data
120   
121    my $file = shift @_;
122   
123    my $yaml = YAML::Tiny->read($file);
124   
125    if ($!) {
126        warn "$0: '$file' $!";
127    }
128   
129    return $yaml->[0];
130}
131
132sub login {
133    # make Net::Twitter::Lite object and login
134    #   param   => hash object of configration
135    #   ret     => Net::Twitter::Lite object
136   
137    my $conf = shift @_;
138   
139    my $bot = Net::Twitter::Lite->new(
140        consumer_key    => $conf->{consumer_key},
141        consumer_secret => $conf->{consumer_secret},
142    );
143   
144    $bot->access_token($conf->{access_token});
145    $bot->access_token_secret($conf->{access_token_secret});
146   
147    return $bot;
148}
149
150sub evalrescue {
151    # output error message at eval error
152   
153    use Scalar::Util qw(blessed);
154   
155    if (blessed $@ && $@->isa('Net::Twitter::Lite::Error')) {
156        warn $@->error;
157        if ($@->twitter_error) {
158            my %twitter_error = %{$@->twitter_error};
159            map {
160                $twitter_error{"$_ => "} = $twitter_error{$_} . "\n";
161                delete $twitter_error{$_}
162            } keys %twitter_error;
163            warn join("", %twitter_error);
164        }
165    }
166    else {
167        warn $@;
168    }
169}
Note: See TracBrowser for help on using the repository browser.