# -*- coding: utf-8 -*-
# File Name:mqtt_chat_client.py
# Python Version:3.5.1

import os
import django
import sys

BASE_DIR = os.path.dirname(os.path.abspath(__file__))  # 定位到你的django根目录
sys.path.append(os.path.abspath(os.path.join(BASE_DIR, os.pardir)))

os.environ.setdefault("DJANGO_SETTINGS_MODULE",
                      "yfwlw_pro.settings")  # project_name 项目名称
django.setup()

print("<-----python_mqtt_client_qxz is run----->")


import paho.mqtt.client as mqtt
import json
from apps.AppInfoManage.models import Alarm_record, Equip, Equip_Forward, Equip_type, MyUser, QXZ_Base_Info, QXZdata, QXZstatus, QXZstatus_New, QXZswitchdata, QXZswitchstatus
from apps.ReportManage.all_dict import transpont_equip_ybq
import re
import datetime
import requests
import time

class CJSONEncoder(json.JSONEncoder):
    def default(self, obj):
        if isinstance(obj, datetime):
            return obj.strftime('%Y-%m-%d %H:%M:%S')
        elif isinstance(obj, date):
            return obj.strftime('%Y-%m-%d')
        else:
            return json.JSONEncoder.default(self, obj)


# 连接后的操作：  0为成功
def on_connect(client, userdata, flags, rc):
    # print("Connected with result code "+str(rc))
    # for x in transpont_equip_ybq:
    #     client.subscribe("yfkj/xycb/c2s/%s"%x)

    # equip_list = Equip_Forward.objects.all()
    # for x in equip_list:
    client.subscribe("yfkj/xycb/c2s/#")

# *****成功发布******
def on_publish(msg, rc):
    if rc == 0:
        print("publish success,msg = "+msg)


# 从服务器接受到消息后回调此函数 ：
def on_message(client, userdata, msg):
    print('\r')
    print('=================================================')
    print('\r')
    print("<-----topic:\n" + msg.topic + ';\n')
    print("Message:\n" + str(msg.payload) + "----->\n")
    # 从主题中获取imei
    # imei = msg.topic[14:len(msg.topic)]
    imei = msg.topic.split('/')[-1]
    # imei = re.sub("\D", "", msg.topic)
    print("<-----imei:", imei, "----->")

    payload = json.loads(msg.payload.decode())
    
    print("type_payload",type(payload))
    data = {"topic":msg.topic,"payload":payload}
    data = json.dumps(data, cls=CJSONEncoder)
    print("发送数据为:",data)
    try:
        if Equip_Forward.objects.filter(equip_id=imei).exists():
            equip_id = Equip_Forward.objects.get(equip_id=imei)
            headers = {"Content-Type": "application/json; charset=UTF-8", 'Connection': 'close'}
            res_1 = requests.post(equip_id.equip_data_url, data=data, timeout=3,headers=headers)
            print("res_1",res_1)
            print("res_1.text",res_1.text)
            print("send success")
    except Exception as e:
        print("错误信息为==================>>",e)



if __name__ == '__main__':

    client = mqtt.Client(
        client_id="PY_MQTT_TRANDSPOND_YBQ",
        clean_session=True,
        userdata=None,
        # protocol=MQTTv311,# 数据库版本
    )
    # 必须设置，否则会返回「Connected with result code 4」
    client.username_pw_set("admin", "password")
    client.on_connect = on_connect
    client.on_message = on_message

    HOST = "127.0.0.1"

    client.connect(HOST, 1883, 60)
    client.loop_forever()