# 論文の条件を設定する
QUERY = "(cat:hep-lat)" #僕は格子やってるので。

# 翻訳のためのAPI KEY（DeepL翻訳版）
# **にはご自分で発行されたdeepL API KEYを入れてください
DeepL_API_KEY = '**'

# IFTTTを用いてarXivとLINEを繋ぐ
# webhook POST先URL
API_URL = "https://maker.ifttt.com/trigger/<IFTTTに設定するイベント名>/with/key/<IFTTTでの個人用KEY>"
# <IFTTTに設定するイベント名>は自分で決める。<IFTTTでの個人用KEY>はIFTTTの設定後にsettingからコピペする。

#モジュールをimport
import datetime, re, requests

# 論文の条件を設定する
QUERY = "(cat:hep-lat)" 

# 翻訳のためのAPI KEY（DeepL翻訳版）
DeepL_API_KEY = '**'

# IFTTTを用いてarXivとLINEを繋ぐ
# webhook POST先URL
API_URL = "https://maker.ifttt.com/trigger/<IFTTTに設定するイベント名>/with/key/<IFTTTでの個人用KEY>"

# 日付取得
today = datetime.date.today() 
yesterday = today - datetime.timedelta(days=1)

#　指定したHTMLtagの情報を返す e.g. Input :<tag>XYZ </tag> -> Output: XYZ
def parse(data, tag):
    pattern = "<" + tag + ">([\s\S]*?)<\/" + tag + ">"
    if all:
        obj = re.findall(pattern, data)
    return obj

#論文を探してきてwebhooksにPOSTする関数
def search_and_send(query, api_url):
    while True:
        counter = 0 #何個今日の論文があったかのカウンター

        url = 'http://export.arxiv.org/api/query?search_query=' + query + '&start=0&max_results=100&sortBy=submittedDate&sortOrder=descending'
        #　arXiv APIから指定したURLの情報を受け取る
        data = requests.get(url).text

        # 各論文ごとに分ける
        entries = parse(data, "entry")

        #　各論文に対し情報の分離と翻訳を行う
        for entry in entries:
            # Parse each entry
            url = parse(entry, "id")[0]

            # 情報の分離
            title = parse(entry, "title")[0]
            abstract = parse(entry, "summary")[0]
            date = parse(entry, "published")[0]

            if date[0:10]  == str(yesterday.strftime('%Y-%m-%d')): #今日の分の論文だけ得るという条件
                # abstの改行を取る
                abstract = abstract.replace('\n', '')

                # deepLで日本語に翻訳
                # https://deepblue-ts.co.jp/nlp/deepl-api-python/ を参照
                # DeepLのURLクエリに仕込むパラメータを作っておく
                params = {"auth_key": DeepL_API_KEY, "text": abstract, "source_lang": 'EN', "target_lang": 'JA'}
                # DeepLに翻訳してもらう
                request = requests.post("https://api-free.deepl.com/v2/translate", data=params)
                #Pro版ならクリエを出すURLは、https://api.deepl.com/v2/translate
                result = request.json()
                abstract_jap = result["translations"][0]["text"]
                message = "\n".join(["<br>Title:  " + title, "<br><br>URL: " + url, "<br><br>Published: " + date, "<br><br>JP_Abstract: " + abstract_jap]) 

                # webhookへPost 
                response = requests.post(api_url, data={"value1": message})

                counter += 1

            else:
                if counter==0 :
                    messageend = "今日の論文は無しです。"
                    requests.post(api_url, data={"value1": messageend})
                    return 0

                else:
                    messageend = "今日の論文は以上です。"
                    requests.post(api_url, data={"value1": messageend})
                    return 0

def main():
    # setup 
    # IFTTT APIを設定
    api_url = API_URL
    # arXiv APIのクエリ
    query = QUERY

    # start 
    # LINEに時間を送る
    dt = datetime.datetime.now().strftime("%Y/%m/%d") + ": arXiv 2 LINE"
    requests.post(api_url, data={"value1": dt})

    # 今日の論文をLINEに送る
    search_and_send(query, api_url)

# main関数を実行(Cloud functionsでは不要)
# main()

datetime
requests==2.23.0