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


import paho.mqtt.client as mqtt
import json
from apps.AppInfoManage.models import Equip, Equip_type, TCCBdata, TCCBstatus, MyUser,Alarm_record, RTUstatus, RTUdata

import re
import datetime
import shutil,os
import threading

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))
    client.subscribe("yfkj/cbd_rtu/c2s/#")
    client.subscribe("yfkj/cbd_rtu/offline/#")


# 发布消息完成回调函数：
def on_publish(msg, rc):
    if rc == 0:
        print("publish success,msg = " + msg)


def msg_thread(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)]
    reg = re.compile(r"(?<=/)\d+")
    imei = reg.search(msg.topic).group(0)
    print("<-----imei:", imei, "----->")

    # 判断主题：
    if "c2s" in msg.topic:
        # 将json字符串解析：
        try:
            payload = json.loads(msg.payload.decode())
        except:
            pass
        if payload.get("cmd") == "data":
            print("<-----uploading status!----->")
            extdata = payload.get("ext")
            # print("extdata:", extdata)
            rtu_exist = Equip.objects.filter(equip_id=imei)
            # 设备存在，进一步判断状态表是否存在：
            if rtu_exist.exists():
                # 设备状态表存在、刷新状态表：
                print("<-----this equip is existed!----->")
                e_id = Equip.objects.get(equip_id=imei)
                RTUdata.objects.create(equip_id=e_id, rtu_data=extdata)
                if RTUstatus.objects.filter(equip_id=imei).exists():
                    print("<-----this equip's status is existed!----->")
                    try:
                        sta = RTUstatus.objects.get(equip_id=imei)
                        sta.rtu_status=extdata
                        sta.is_online = '1'
                        sta.save()
                        print("<-----status update success!----->")
                    except:
                        print("<-----status update failed!----->")
                else:
                    # 设备状态表不存在、创建状态表：
                    print("<-----this equip's status is not existed!----->")
                    try:
                        e_id = Equip.objects.get(equip_id=imei)
                        try:
                            RTUstatus.objects.create(equip_id=e_id, rtu_status=extdata, is_online = '1')
                            print("<-----this equip's status table re-create successed!----->")
                        except:
                            print("<-----this equip's status table re-create failed!----->")
                    except:
                        print("<-----this equip didn't exist!----->") 
            else:
                # 设备不存在，在设备列表中创建：
                try:
                    # 得到设备类型实例：
                    equip_t = Equip_type.objects.get(type_id="10")
                    try:
                        e_id = Equip.objects.create(equip_id=imei, equip_type=equip_t)
                        print("<-----this imei add successed!----->")
                        
                        RTUstatus.objects.create(equip_id=e_id, rtu_status=extdata, is_online = '1')
                        RTUdata.objects.create(equip_id=e_id, rtu_data=extdata)
                    except:
                        print("<-----this imei add failed!----->")             
                except:
                    print("<-----register failed because this equip type is not existed!----->")

        # 设备异常预警
        elif payload.get("cmd") == "warn":
            print("<-----uploading warn!----->")
            print("%s is warn ! ! !" % imei)
            extdata = payload.get("ext")
            cbd_exist = Equip.objects.filter(equip_id=imei)
            # 设备存在，进一步判断状态表是否存在：
            if cbd_exist.exists():
                try:
                    e_id = Equip.objects.get(equip_id=imei)
                    # 创建预警记录：
                    Alarm_record.objects.create(equip_id=e_id, alarm_desc=extdata, e_type="10")
                    print("update warn ok!")
                except:
                    print("update warn failed!")
            else:
                print("this imei is not exist!")

    # 离线消息：
    elif "offline" in msg.topic:
        try:
            # 将json字符串解析：
            payload = json.loads(msg.payload.decode())
            if payload.get("cmd") == "offline":
                print("<-----离线消息!----->")
                print("%s is offline!" % imei)
                cbd_exist = Equip.objects.filter(equip_id=imei)
                # 设备存在，进一步判断状态表是否存在：
                if cbd_exist.exists():
                    try:
                        e_id = Equip.objects.get(equip_id=imei)
                        RTUstatus.objects.filter(equip_id=imei).update(is_online = '0',off_time = datetime.datetime.now())
                        # 创建预警记录：
                        Alarm_record.objects.create(equip_id=e_id, alarm_desc="{'status':0,'type':'offline'}", e_type="10")
                        print("update offline ok!")
                    except:
                        print("update offline failed!")
                else:
                    print("this imei is not exist!")
        except:
            pass


# 从服务器接收消息的回调函数 ：
def on_message(client, userdata, msg):
    t = threading.Thread(target=msg_thread,args=(msg,))
    #打印出当前线程的名称和id
    # print(threading.currentThread().name)
    # t.setDaemon(True)
    t.start()
    return

if __name__ == '__main__':
    client = mqtt.Client(
        client_id="PY_MQTT_CLIENT_RTU",
        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
    # #设置与服务器断开连接回调函数
    # client.on_disconnect = on_disconnect 
    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}))