from requests import get, post
import json
import sys

entities = ['switch.pc_speakers']
payload = {}
post_url = 'https://hass_url/api/services/switch/toggle'
get_url = 'https://hass_url/api/states/'
headers = {'x-ha-access': 'pw_hass',
           'content-type': 'application/json'}

get_internal_url = '/ay?o=1'
speakers_ip = 'http://192.168.1.96'

use_hass = False

def toggle_state(entity):
    payload['entity_id'] = entity
    json_payload = json.dumps(payload)
    response = post(post_url, headers=headers, data=json_payload)
    return json.loads(response.text)


def get_state(entity):
    response = get(get_url + entity, headers=headers)
    json_response = json.loads(response.text)
    return json_response


if sys.argv[1] == 'shutdown':
    if not use_hass:
        get(speakers_ip + get_internal_url)
    else:
        if get_state(entities[0])['state'] == 'on':
            toggle_state(entities[0])
elif sys.argv[1] == 'startup':
    if not use_hass:
        get(speakers_ip + get_internal_url)
    else:
        if get_state(entities[0])['state'] == 'off':
            toggle_state(entities[0])