import requests
import os

prometheus_url = "http://192.168.10.109:9090/api/v1/query"
query = 'probe_http_status_code{instance="https://www.tcyunsc.cn/management/common/captcha/init"}'

response = requests.get(prometheus_url, params={'query': query})
if response.status_code != 200:
    data = response.json()
    status_code = data['data']['result'][0]['value'][1]
    if status_code == "200":
        cmd1 = 'ansible-playbook /root/ops_scripts/ansible_playbook/aqzq_restart_playbook/1_restart_aqzq_backend_port_container.yml '
        result1 = os.popen(cmd1).read().strip()
        cmd2 = 'ansible-playbook /root/ops_scripts/ansible_playbook/aqzq_restart_playbook/2_restart_aqzq_data_port_container.yml '
        result2 = os.popen(cmd2).read().strip()
        cmd3 = 'ansible-playbook /root/ops_scripts/ansible_playbook/aqzq_restart_playbook/3_restart_aqzq_frontend_container.yml '
        result3 = os.popen(cmd3).read().strip()
        mgs2 = "{}\n{}".format(result1, result2, result3)
        os.system('/usr/bin/python3 /data/backupscript/sendMessageQiyeweixin.py "{}"'.format(mgs2))