# --- coding: utf-8 ---
import tweepy
import boto3
from datetime import datetime

### Twitter API KEY
Consumer_key = 'XXXXXXXXXXXXXXXXXXXXXX'
Consumer_secret = 'XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX'
Access_token = 'XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX'
Access_secret = 'XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX'

### TwitterAPI認証用関数
def authTwitter():
    auth = tweepy.OAuthHandler(Consumer_key, Consumer_secret)
    auth.set_access_token(Access_token, Access_secret)
    api = tweepy.API(auth, wait_on_rate_limit = True) # API利用制限にかかった場合、解除まで待機する
    return(api)

def printS3(s):
    s3 = boto3.resource('s3') 
    bucket = 'XXXXXXXXXXXXXXX'
    key = 'test_' + datetime.now().strftime('%Y-%m-%d-%H-%M-%S') + '.csv'
    file_contents = s
    obj = s3.Object(bucket,key)
    obj.put(Body = file_contents)
    return

### Tweetの検索結果を標準出力
def printTweetBySearch(s):
    api = authTwitter() # 認証

    tweets = tweepy.Cursor(api.search_tweets, q = s,# APIの種類と検索文字列
                           include_entities = True, # 省略されたリンクを全て取得
                           tweet_mode = 'extended', # 省略されたツイートを全て取得
                           result_type="mixed",
                           ).items()
    p = 'id,created_at,screen_name,name\n'

    for tweet in tweets:
        p += str(tweet.id) + ','               # tweetのIDを出力。ユニークなもの
        p += str(tweet.created_at) + ','       # 呟いた日時
        p += str(tweet.user.screen_name) + ',' # ユーザー名
        p += str(tweet.user.name)              # アカウント名
        p += "\n"
        print('＝＝＝＝＝＝＝＝＝＝')
        print('twid : ',tweet.id)               # tweetのIDを出力。ユニークなもの
        print('date : ',tweet.created_at)       # 呟いた日時
        print('user : ',tweet.user.screen_name) # ユーザー名
        print('user : ',tweet.user.name)        # アカウント名
    #S3出力
    printS3(p)

def lambda_handler(event, context):
    text = '#PODB使ってみたい'
    printTweetBySearch(text)
    None

--クラウドストレージ統合を作成
CREATE OR REPLACE STORAGE INTEGRATION S3_role_integration
TYPE = EXTERNAL_STAGE
STORAGE_PROVIDER = S3
ENABLED = TRUE
STORAGE_AWS_ROLE_ARN = 'arn:aws:iam::XXXXXXXX:role/snowflake_role'
STORAGE_ALLOWED_LOCATIONS = ('s3://ts-snowpipetest/')

desc integration S3_role_integration;

--S3バケットを参照する外部ステージ(ステージエリア AWSSTAGE)を作成
create or replace stage awsstage
url = 's3://ts-snowpipetest/'
storage_integration = S3_role_integration;

--自動インジェストを有効にしたパイプを作成
create or replace pipe twitter.public.awspipe auto_ingest=true as
copy into twitter.public.TARGET_TWEETS_Lambda
from @twitter.public.awsstage
file_format = (TYPE = CSV skip_header = 1)

--セキュリティ
use role securityadmin;
create or replace role S3_role;
grant usage on database twitter to role S3_role;
grant usage on schema twitter.public to role S3_role;
grant insert, select on twitter.public.TARGET_TWEETS_Lambda to role S3_role;
grant usage on stage twitter.public.awsstage to role S3_role;
grant ownership on pipe twitter.public.awspipe to role accountadmin;
grant role S3_role to user MIMURA;
alter user MIMURA set default_role = S3_role;
--------------------------------------------------
--ここまででパイプ設定完了

--内部ステージの確認
LIST @twitter.public.awsstage

--内部ステージからファイルを削除 ここを消すとAWSのS3からも消える
REMOVE @twitter.public.awsstage

--パイプを開始 このタイミングではステージに滞留しているファイルをコピーしない。次のイベントでコピーする。
ALTER PIPE twitter.public.awspipe SET PIPE_EXECUTION_PAUSED=false