【obs-websocket v5版】rekordbox+OBSでトラックタイトルの表示を自動切り替えする@Python

pip install obs-websocket-py opencv-python

{
  'name': <API名>,
  'datain': <APIから返された値>,
  'dataout': <APIに送ったパラメータ>,
  'status': <実行結果>
}

{
  'sceneItems': 
  [
    {
      'inputKind': <str>,
      'isGroup': <NoneType>,
      'sceneItemBlendMode': <str>,
      'sceneItemEnabled': <bool>,
      'sceneItemId': <int>,
      'sceneItemIndex': <int>,
      'sceneItemLocked': <bool>,
      'sceneItemTransform': 
      {
        'alignment': <int>,
        'boundsAlignment': <int>,
        'boundsHeight': <float>,
        'boundsType': <str>,
        'boundsWidth': <float>,
        'cropBottom': <int>,
        'cropLeft': <int>,
        'cropRight': <int>,
        'cropTop': <int>,
        'height': <float>,
        'positionX': <float>,
        'positionY': <float>,
        'rotation': <float>,
        'scaleX': <float>,
        'scaleY': <float>,
        'sourceHeight': <float>,
        'sourceWidth': <float>,
        'width': <float>
      },
      'sourceName': <str>,
      'sourceType': <str>
    },
    {
      ...
    },
    ...
  ]
}

from obswebsocket import obsws, requests as rq

# websocket接続先設定
host = "localhost" 
port = 4455
password = "xxxxxxxxxxxxxxxx"
# ----------

scene = input("シーン名 > ")  #ID取得対象のシーン名を入力させる

# OBSに接続
ws = obsws(host, port, password)
ws.connect()

items = ws.call(rq.GetSceneItemList(sceneName=scene)).datain["sceneItems"]  # アイテム取得

items.reverse() # 逆順

export_list = []  # 出力用リスト初期化

# リストの処理
for i in items:
  # グループなら
  if i["isGroup"] == True:
    export_list.append("ID: {: >3,d} | [G]{}".format(i["sceneItemId"],i["sourceName"])) # グループを出力用リストに追加

    group_items = ws.call(rq.GetGroupSceneItemList(sceneName=i["sourceName"])).datain["sceneItems"] # グループ内アイテム取得
    group_list = [] #グループ用リスト初期化

    # グループ内のアイテム処理
    for j in group_items:
      group_list.append("ID: {: >3,d} |  + {}".format(j["sceneItemId"],j["sourceName"]))  # グループ用リストに追加（グループ内アイテム）

    group_list.reverse()  # 逆順

    #グループ内アイテムを出力用リストに追加
    for j in group_list:
      export_list.append(j)

  # グループではないなら
  else:
    export_list.append("ID: {: >3,d} | {}".format(i["sceneItemId"],i["sourceName"]))  # 出力用リストに追加

# 出力
for i in export_list:
  print(i)

ws.disconnect() # OBSから切断

import os
import cv2
import datetime

from obswebsocket import obsws, requests as rq

# ======================================

# OBS接続設定
host = "localhost" 
port = 4455
password = "xxxxxxxxxxxxxxxx"

# シーン名
scene_name = "DJ"

# アイテムID
# ID_Analyzerで調べたやつ
deck1_id = 17
deck2_id = 18
track2_id = 19

# キャプチャデバイスID
device_id = 3

# 色のしきい値
threshold = 20

# キャプチャ解像度
capture_width = 1920
capture_height = 1080

# クロップエリア
deck_top = 353
deck_bottom = 370
deck1_left = 902
deck1_right = 958
deck2_left = 1862
deck2_right = 1918

# ======================================

# ソースの表示を切り替える
def setVisible(item_id, visible):
	ws.call(rq.SetSceneItemEnabled(sceneName=scene_name, sceneItemId=item_id, sceneItemEnabled=visible))

# VideoCapture オブジェクトを取得
capture = cv2.VideoCapture(device_id)
print(capture.isOpened())
capture.set(cv2.CAP_PROP_FRAME_WIDTH, capture_width)
capture.set(cv2.CAP_PROP_FRAME_HEIGHT, capture_height)

# OBSに接続
ws = obsws(host, port, password)
ws.connect()

# 処理フラグ
switch = 0

try:
	while True:
		# 映像を取得
		ret, frame = capture.read()
		
		# クロップ
		deck1_crop = frame[ deck_top : deck_bottom , deck1_left : deck1_right]
		deck2_crop = frame[ deck_top : deck_bottom , deck2_left : deck2_right]
		
		# 色情報を取得
		red_deck1 = deck1_crop.T[2].flatten().mean()
		red_deck2 = deck2_crop.T[2].flatten().mean()
		
		# DECK1がMASTERのとき
		if red_deck1 > threshold and (switch == 2 or switch == 0):
			os.system('cls')
			print(datetime.datetime.now().strftime('%Y/%m/%d %H:%M:%S'))
			print("\nNOW PLAYING: DECK1\n")
			setVisible(deck2_id, False)
			setVisible(deck1_id, True)
			setVisible(track2_id, False)
			switch = 1
		
		# DECK2がMASTERのとき
		elif red_deck2 > threshold and (switch == 1 or switch == 0):
			os.system('cls')
			print(datetime.datetime.now().strftime('%Y/%m/%d %H:%M:%S'))
			print("\nNOW PLAYING: DECK2\n")
			setVisible(deck1_id, False)
			setVisible(deck2_id, True)
			setVisible(track2_id, True)
			switch = 2
		
		# よくわからんとき
		elif (red_deck1 < threshold and red_deck2 < threshold) or (red_deck1 > threshold and red_deck2 > threshold):
			os.system('cls')
			print(datetime.datetime.now().strftime('%Y/%m/%d %H:%M:%S'))
			print("\nUndefined\n")
			setVisible(deck1_id, False)
			setVisible(deck2_id, False)
			setVisible(track2_id, False)
			switch = 0
		
		# なにもしない
		else:
			pass

# Ctrl+Cが押されたら終了
except KeyboardInterrupt:
	capture.release()
	ws.disconnect()