# -*- 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_transpond_scd is run----->")


import paho.mqtt.client as mqtt
import json
from apps.AppInfoManage.models import Alarm_record, Equip, Equip_Forward, Equip_type, MyUser, RecentSCDdata, SCDdata, SCDstatus, SCDstatus_all
from apps.ReportManage.all_dict import transpont_equip_scd
import re
import requests
import datetime

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_scd:
    #     client.subscribe("/yfkj/scd/pub/%s"%x)
    #     client.subscribe("/yfkj/scd/offline/%s"%x)
    client.subscribe("/yfkj/scd/pub/#")
    client.subscribe("/yfkj/scd/offline/#")

# *****成功发布******
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 = re.sub("\D", "", msg.topic)
    print("<-----imei:", imei, "----->")
    # url = "http://127.0.0.1:8000/test"
    payload = json.loads(msg.payload.decode())
    
    # payload = json.dumps(payload, cls=CJSONEncoder)

    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)
            if equip_id.equip_data_between == "1":
                res_1 = requests.post(equip_id.equip_data_url, data=data, timeout=3)
            elif equip_id.equip_data_between == "2":
                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_TRANSPOND_SCD",
        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()

    # # 输入发布的话题名称：
    # # user = input("请输入名称:")
    # topic = "/yfkj/scd/cmd/2001"
    # client.user_data_set(topic)

    # client.loop_start()

    # while True:
    #     str = input()
    #     if str:
    #         client.publish("/yfkj/scd/cmd/2001", json.dumps({"topic": topic, "cmd": str}))