lambda_twitter_search/lambda_function.py at master · sparkgene/lambda_twitter_search · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
# -*- coding: utf-8 -*-
import boto3
import json
import ConfigParser
from time import gmtime, strftime

from searchkeyword import SearchKeyword
from slackclient import SlackClient

print('Loading function')

def lambda_handler(event, context):
    print(strftime("%a, %d %b %Y %H:%M:%S +0000", gmtime()))
    inifile = ConfigParser.SafeConfigParser()
    inifile.read("./config.ini")

    dynamodb = boto3.resource('dynamodb')
    table = dynamodb.Table(inifile.get('dynamodb', 'table_id'))
    key_id = int(inifile.get('dynamodb', 'id_value'))
    result = table.get_item(
        Key={
            'Id': key_id
        }
    )
    previous_last_id = result['Item']['LastId']

    last_id = previous_last_id
    print("previous_last_id: {0}".format(previous_last_id))
    max_send_count = inifile.get('search', 'max_tweet')
    if max_send_count.isdigit():
        max_send_count = int(max_send_count)
    else:
        raise TypeError("max_tweet is not numeric")

    send_count = 0
    for tweet in SearchKeyword(
        inifile.get('twitter_api', 'consumer_key'),
        inifile.get('twitter_api', 'consumer_secret'),
        inifile.get('twitter_api', 'access_token'),
        inifile.get('twitter_api', 'access_token_secret'),
        inifile.get('search', 'keyword'),
        previous_last_id):

        if last_id < tweet['id']:
            last_id = tweet['id']
        if send_count > max_send_count:
            break # sending many tweets to slack takes long time.
        send_count += 1

        sc = SlackClient(inifile.get('slack', 'token'))
        attachments = [
            {
                "fallback": tweet['text'],
                "pretext": "https://twitter.com/{0}/status/{1}".format(
                    tweet['screen_name'],
                    tweet['id']
                ),
                "author_name": tweet['user_name'],
                "author_link": "https://twitter.com/{0}".format(
                    tweet['screen_name']
                ),
                "author_icon": tweet['profile_image_url'],
                "text": tweet['text'],
                "color": "#439FE0"
            }
        ]
        res = sc.api_call("chat.postMessage",
                channel=inifile.get('slack', 'channel'),
                attachments=json.dumps(attachments),
                username=inifile.get('slack', 'username'),
                icon_emoji=inifile.get('slack', 'icon_emoji'))
        print(res)

    print("last_id: {0}".format(last_id))
    if last_id > previous_last_id:
        table.put_item(
            Item={
                "Id":key_id,
                "LastId": last_id,
                "source": inifile.get('dynamodb', 'source')}
        )