diff --git a/main.py b/main.py index 794bbce..8e76b4b 100644 --- a/main.py +++ b/main.py @@ -1,4 +1,6 @@ import os +import sys +import json from lib.mqtt import create_client, update_disc, control_player, create_config # Define the MQTT server details @@ -12,10 +14,20 @@ client = create_client(broker, port, username, password) create_config(client) -# update_disc(client, {"type":"album", "id":"MPREb_2UBRpewr9ad"}) -# update_disc(client, {"type":"playlist", "id":"LM"}) - -# +# Check if any arguments are passed +if len(sys.argv) > 1: + # Get the JSON argument from the command line + json_str = sys.argv[1] + + try: + # Convert the JSON string into a dictionary + disc_data = json.loads(json_str) + # Pass the parsed data to the update_disc function + update_disc(client, disc_data) + except json.JSONDecodeError as e: + print(f"Invalid JSON format: {e}") +else: + print("No JSON argument passed, running default control_player action.") + +# You can still have the default behavior if no JSON is passed control_player(client, "EJECT") - -# \ No newline at end of file