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


import paho.mqtt.client as mqtt
import json
from apps.AppInfoManage.models import Equip, Equip_type, SCDdata, SCDstatus, MyUser,Alarm_record, RecentSCDdata, SCDstatus_all
import re
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))
    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):
    time_a = datetime.datetime.now()
    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, "----->")
    if imei == "862285036228686":
        pass
    else:
        # 判断主题：
        if "pub" in msg.topic:
            # 将json字符串解析：
            payload = json.loads(msg.payload.decode())

            if payload.get("cmd") == "status":
                print("<-----uploading status!----->")
                extdata = payload.get("ext")
                # print("extdata:", extdata)
                scd_exist = Equip.objects.filter(equip_id=imei)
                # 设备存在，进一步判断状态表是否存在：
                if scd_exist.exists():
                    # 新建数据记录：
                    try:
                        e_id = Equip.objects.get(equip_id=imei)
                        SCDstatus_all.objects.create(
                            equip_id=e_id, scd_status=extdata, lng=extdata['lng'], lat=extdata['lat'], is_online = '1', ds=extdata['ds'])

                        print("<-----status_all upload success!----->")
                    except:
                        print("<-----status_all upload failed!----->")
                    # 设备状态表存在、刷新状态表：
                    print("<-----this equip is existed!----->")
                    if SCDstatus.objects.filter(equip_id=imei).exists():
                        print("<-----this equip's status is existed!----->")
                        try:
                            sta = SCDstatus.objects.get(equip_id=imei)
                            sta.scd_status=extdata
                            sta.lng=extdata['lng']
                            sta.lat=extdata['lat']
                            sta.is_online = '1'
                            sta.save()
                            print("<-----status update success!----->")
                            time_b = datetime.datetime.now()
                            print("一共用时:",time_b-time_a)
                        except:
                            print("<-----status update failed!----->")
                    else:
                        # 设备状态表不存在、创建状态表：
                        print("<-----this equip's status is not existed!----->")
                        try:
                            e_id = Equip.objects.get(equip_id=imei)
                            try:
                                SCDstatus.objects.create(equip_id=e_id, ds=extdata['ds'],
                                scd_status=extdata, lng=extdata['lng'], lat=extdata['lat'],
                                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=extdata['dtype'])
                        try:
                            e_id = Equip.objects.create(equip_id=imei, equip_type=equip_t)
                            print("<-----this imei add successed!----->")
                            try:
                                SCDstatus.objects.create(equip_id=e_id, ds=extdata['ds'], 
                                scd_status=extdata,lng=extdata['lng'], lat=extdata['lat'],
                                is_online = '1')
                                print("<-----this imei register successed!----->")
                            except:
                                print("<-----this imei register failed!----->")
                        except:
                            print("<-----this imei add failed!----->")  
                    except:
                        print("<-----this imei register failed because this equip type is not exist,!----->")
                    # rtn="register success!!!"
                    # print("注册连接回应："+"/yfkj/scd/cmd/"+payload.get("equip_id"))
                    # client.publish("/yfkj/scd/cmd/"+payload.get("equip_id"), json.dumps({"msg": rtn}))

            elif payload.get("cmd") == "data":
                print("<-----uploading data!----->")
                extdata = payload.get("ext")
                # print("extdata:", extdata)
                # 新建数据记录：
                try:
                    e_id = Equip.objects.get(equip_id=imei)
                    SCDdata.objects.create(
                        equip_id=e_id, ds=extdata["ds"], scd_data=extdata)

                    print("<-----data upload success!----->")
                except:
                    print("<-----data upload failed!----->")
                
                # 如果是新设备，表里没有，需要做判断
                if RecentSCDdata.objects.filter(equip_id=imei).exists():
                    # 设备存在，更新数据
                    try:
                        Recdata = RecentSCDdata.objects.get(equip_id=imei)
                        Recdata.ds=extdata["ds"]
                        Recdata.scd_data=extdata
                        Recdata.upl_time=datetime.datetime.now()
                        Recdata.save()
                        print("<-----RecentSCDdata update success!----->")
                    except:
                        print("<-----RecentSCDdata update failed!----->")
                else:
                    # 如果是新设备，则新建数据
                    try:
                        e_id = Equip.objects.get(equip_id=imei)
                        RecentSCDdata.objects.create(
                            equip_id=e_id, ds=extdata["ds"], scd_data=extdata, upl_time=datetime.datetime.now())
                        print("<-----RecentSCDdata create success!----->")
                    except:
                        print("<-----RecentSCDdata create failed!----->")


            # 参数配置信息：
            elif payload.get("cmd") == "paramconf":
                print("<-----uploading paramconf!----->")
                extdata = payload.get("ext")
                print("extdata:", extdata)
                # 更新状态表中的参数配置信息；
                try:
                    sta = SCDstatus.objects.get(equip_id=imei)
                    sta.paramconf=extdata
                    sta.save()
                    print("<-----status.paramconf update success!----->")
                except:
                    print("<-----scdstatus table is not exist,status.paramconf upload failed!----->")
            
                # 设备异常预警
            elif payload.get("cmd") == "warn":
                print("<-----uploading warn!----->")
                print("%s is warn ! ! !" % imei)
                extdata = payload.get("ext")
                scd_exist = Equip.objects.filter(equip_id=imei)
                if scd_exist.exists():
                    try:
                        e_id = Equip.objects.get(equip_id=imei)
                        # 创建预警记录：
                        Alarm_record.objects.create(equip_id=e_id, alarm_desc=extdata, e_type="2")
                        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)
                    scd_exist = Equip.objects.filter(equip_id=imei)
                    # 设备存在，进一步判断状态表是否存在：
                    if scd_exist.exists():
                        try:
                            e_id = Equip.objects.get(equip_id=imei)
                            # now_time = json.dumps(datetime.datetime.now(), cls=CJSONEncoder)
                            # 更新状态表中未离线状态：
                            SCDstatus.objects.filter(equip_id=imei).update(is_online = '0',off_time = datetime.datetime.now())

                            # 增加状态历史的离线消息
                            sta1 = SCDstatus.objects.get(equip_id=imei)
                            # sta1.update(is_online = '0',off_time = datetime.datetime.now())
                            SCDstatus_all.objects.create(equip_id=e_id, ds=sta1.ds, off_time = datetime.datetime.now(),
                                    scd_status=sta1.scd_status,lng=sta1.lng, lat=sta1.lat,
                                    is_online = '0')


                            # 创建预警记录：
                            Alarm_record.objects.create(equip_id=e_id, alarm_desc="{'status':0,'type':'offline'}",e_type="2")
                            print("update offline ok!")
                        except Exception as e:
                            print(e)
                            print("update offline failed!")
                    else:
                        print("this imei is not exist!")
            except:
                pass

if __name__ == '__main__':

    client = mqtt.Client(
        client_id="PY_MQTT_CLIENTC_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}))