REDASH_ADDITIONAL_QUERY_RUNNERS=redash.query_runner.python

- name: REDASH_ADDITIONAL_QUERY_RUNNERS
  value: redash.query_runner.python

import json
from datetime import datetime, timedelta
import urllib
import urllib2
import operator
import time
 
## アクセストークン
access_token = 'XXXXXXXXXXXXXXXX'
 
####### 対象チャネルの投稿を取得 #####
## 対象チャネルのID
channel_id = 'XXXXXXXXXXXXXXXX'
## 対象チャネルの投稿を取得するAPI
channel_url = 'https://chat.example.com/api/v4/channels/' + channel_id + '/posts'
## リクエストヘッダにアクセストークンを含める
channel_post_get_request_header = {
    'Authorization': 'Bearer ' + access_token,
}
## リクエストパラメータで、チャットの取得範囲を指定
since_time = datetime.now() - timedelta(days=1825)
since_time_unixtime = time.mktime(datetime.timetuple(since_time))
since_time_unixtime_milli = int(since_time_unixtime * 1000)
req_param = urllib.urlencode(
  {
    'since': since_time_unixtime_milli
  }
)
 
## 対象チャネルの投稿を取得するAPIにリクエストパラメータを付与
channel_url = channel_url +  "?{0}".format(req_param) # "+=演算子はRedashでは利用不可"
req = urllib2.Request(channel_url,headers=channel_post_get_request_header)
 
res = urllib2.urlopen(req)
post_data = res.read()
res.close()
 
## 対象チャネルの投稿データJSON
post_data_JSON = json.loads(post_data)
 
 
####### ユーザーIDとユーザー名の対応 #####
post_user_id_set = set() # 重複排除にsetを利用
for prop in post_data_JSON['posts']:
    post_user_id_set.add(post_data_JSON['posts'][prop]['user_id'])
post_user_id_list = list(post_user_id_set) # ただしlistにしないと後でjsonに渡せないため変換
 
 
####### ユーザーIDと表示名のディクショナリを作成 #####
## ユーザーIDを渡すと、そのユーザー情報を返すAPI
user_list_url = 'https://chat.example.com/api/v4/users/ids'
 
## リクエストヘッダにアクセストークンとContent-Typeを含める
user_list_get_request_header = {
    'Authorization': 'Bearer ' + access_token,
    'Content-Type': 'application/json'
}
## 投稿者のユーザーIDセットを渡し、各ユーザーの情報を取得
req = urllib2.Request(
    user_list_url, 
    headers=user_list_get_request_header,
    data=json.dumps(post_user_id_list).encode())
res = urllib2.urlopen(req)
user_data = res.read()
res.close()
user_data_JSON = json.loads(user_data)
 
 
###### 「ユーザーID:表示名」のディクショナリを作成 #####
post_user_dict = {}
## 取得した投稿の投稿者(user_id)のリストを1件ずつ見て、ユーザー名(username)の対応付け
for pu_id in post_user_id_list:
    for uid in user_data_JSON: # uidは、user_data_JSONの各要素
        if uid["id"] == pu_id:
            post_user_dict.update({pu_id:uid["username"]})
            break
    else:
        # ユーザー名が引けなかった場合、ユーザーIDを表示。
        post_user_dict.update({pu_id:pu_id})
 
 
###### ソート #####
## APIから取得したままだと、post_idでソートされている。
## post_idはランダム文字列のため、投稿日時でソートする。
post_data_list = []
for prop in post_data_JSON['posts']:
    # 'create_at'はUNIX時間のミリ秒
    dt = datetime.fromtimestamp(post_data_JSON['posts'][prop]['create_at'] / 1000) + timedelta(hours=+9)
    post_data_list.append(
        {
            'create_at':dt.strftime('%Y-%m-%d %H:%M.%S'), 
            # 「ユーザーID:表示名」ディクショナリから、ユーザーIDから表示名を取得
            'username':post_user_dict[ post_data_JSON['posts'][prop]['user_id'] ],
            'message':post_data_JSON['posts'][prop]['message']
        }
    )
sorted_post_data_list = sorted(post_data_list,key=operator.itemgetter('create_at'))
 
 
###### 表示する投稿の絞り込み #####
narrow_words_set = {'リモートワーク','Teleworker','機能'}
narrow_words_decoded_set = [nwd.decode('utf-8') for nwd in narrow_words_set]
 
## ソートされた投稿を1件ずつ、キーワードが部分一致するか確認する。
narrow_post_data_list = []
for spost in sorted_post_data_list:
    for nword in narrow_words_decoded_set:
        # 部分一致したら、リストに入れる。部分一致が0件ならリストは空のまま。
        if nword in spost['message']:
            narrow_post_data_list.append(spost)
            break
 
 
###### 画面表示 #####
result = {}
for disp_post in narrow_post_data_list:
    add_result_row(result,{
        'create_at':disp_post['create_at'], 
        'username':disp_post['username'],
        'message':disp_post['message']
    })
 
add_result_column(result,'create_at','','string')
add_result_column(result,'username','','string')
add_result_column(result,'message','','string')

narrow_post_data_list = [nl for nl in sorted_post_data_list if any(map(lambda x : x in nl['message'], narrow_words_set))]
## もしくは
narrow_post_data_list = [nl for nl in sorted_post_data_list if any(x in nl['message'] for x in narrow_wo