obs ws testing

This commit is contained in:
Chris kerr 2024-03-03 18:53:00 -05:00
parent 6cfae6abdf
commit a08ebd6002

76
obs_action.py Normal file
View File

@ -0,0 +1,76 @@
import json
import asyncio
import websockets
from obswebsocket import obsws, requests
def load_config(file_path):
try:
with open(file_path, 'r') as config_file:
config = json.load(config_file)
return config
except FileNotFoundError:
print(f"Config file not found at {file_path}.")
return None
async def listen_to_relay(your_pubkey, your_relay_url, obs_host, obs_port, obs_password):
uri = f"{your_relay_url}/ws"
# Create OBS connection
obs = obsws(obs_host, obs_port, obs_password)
obs.connect()
async with websockets.connect(uri) as websocket:
# Subscribe to all kind 3 events
subscription_id = "kind_3_events" # Replace with a unique identifier
await websocket.send(json.dumps(["REQ", subscription_id, {"kinds": [3]}]))
while True:
response = await websocket.recv()
await handle_event(response, your_pubkey, obs) # Pass the OBS connection to handle_event
async def handle_event(event_json, your_pubkey, obs):
event = json.loads(event_json)
if event[0] == "EVENT":
event_data = event[2]
if event_data["kind"] == 3: # Follow list event
await handle_follow_list(event_data, your_pubkey, obs) # Pass the OBS connection to handle_follow_list
async def handle_follow_list(follow_list_event, your_pubkey, obs):
if "tags" in follow_list_event:
# Extract the list of "p" tags from the follow list event
p_tags = [tag for tag in follow_list_event["tags"] if tag[0] == "p"]
# Check if your pubkey is the last entry in the "p" tags
if p_tags and p_tags[-1][1] == your_pubkey:
print(f"You have a new follower: {follow_list_event.get('pubkey', '')}")
# Toggle the specified source group in OBS
source_group_name = "heyguys" # Replace with the name of your source group
scene_name = "Escape From Tarkov" # Replace with the name of your scene
await toggle_source_group(obs, scene_name, source_group_name)
# Modify the toggle_source_group function to accept the scene name
async def toggle_source_group(obs, scene_name, source_group_name):
# Enable the source group in the specified scene
obs_response_enable = obs.call(requests.SetSceneItemProperties(item=source_group_name, scene=scene_name, visible=True))
print(f"Enable Response: {obs_response_enable}")
# Wait for a certain interval (e.g., 5 seconds)
await asyncio.sleep(5)
# Disable the source group in the specified scene
obs_response_disable = obs.call(requests.SetSceneItemProperties(item=source_group_name, scene=scene_name, visible=False))
print(f"Disable Response: {obs_response_disable}")
print(f"Toggled source group '{source_group_name}' in scene '{scene_name}'.")
if __name__ == "__main__":
config_path = 'config.json'
config = load_config(config_path)
if config:
asyncio.get_event_loop().run_until_complete(listen_to_relay(
config["your_pubkey"], config["your_relay_url"],
config["obs_host"], config["obs_port"], config["obs_password"]
))