Mの備忘録 - Carnet de M -

pip install opencv-python
pip install numpy
pip install pyautogui

lemma,pos,count
chat,noun,5
ordinateur,noun,10
maison,noun,15
amour,noun,20
fille,noun,25

import os # OSに関連した機能を提供するモジュール
import cv2 # OpenCVのPythonバインディング
import csv # CSVファイルを扱うためのモジュール
import numpy as np # 数値計算を行うモジュール
import pyautogui # GUI操作を自動化するモジュール
import time # 時間に関連した機能を提供するモジュール

CSV_FILE = "tango_list.csv" # 単語リストのCSVファイル名
IMG_BOOKMARK = "bookmark.png" # ブックマークボタンの画像
IMG_INTERNAL_CAPTURE = "capture.png" # スクリーンショットの画像

# ブックマークボタンの位置を取得
def get_center_position():
    # スクリーンショット画像を読み込み
    _input = cv2.imread(IMG_INTERNAL_CAPTURE, cv2.IMREAD_COLOR)
    _temp = cv2.imread(IMG_BOOKMARK)
    # 画像のサイズを取得
    _h, _w = _temp.shape[:2]
    # テンプレートマッチングでブックマークボタンの位置を検索
    _match = cv2.matchTemplate(_input, _temp, cv2.TM_CCOEFF_NORMED)
    _loc = np.where(_match >= 0.80)
    try:
        _x = _loc[1][0]
        _y = _loc[0][0]
        return _x + _w / 2 , _y + _h /2
    except IndexError as e:
        # ブックマークボタンが見つからなかった場合
        return -1,-1

# ブックマークボタンをクリック
def img_click():
    os.system("screencapture " + IMG_INTERNAL_CAPTURE)
    _x,_y = get_center_position()
    if(_x ==-1):
        return -1
    pyautogui.moveTo(_x/2, _y/2)
    pyautogui.click()

def main():
    os.system(f'open mkdictionaries:///') # Dictionariesアプリを起動
    time.sleep(5) # Dictionariesアプリの起動を待つ
    with open(CSV_FILE, "r") as f:
        reader = csv.reader(f)
        next(reader) # ヘッダーをスキップ
        # データを一行ずつ読み込んで処理
        for row in reader:
            lemma = row[0]
            url = f"mkdictionaries:///?text={lemma}&category=fr-ja"
            os.system(f'open {url}')
            time.sleep(2)
            img_click()

# main関数の呼び出し
if __name__=="__main__":
    # プログラムのエントリーポイント
    main()