# -*- coding: utf-8 -*-



#主要的
__author__ = '上海色宁'

import os
import logging
import time
import datetime
import random
from logging.handlers import RotatingFileHandler
import eventlet
import json
from flask import Flask, render_template, request,Response
import requests
import cv2
from flask_mqtt import Mqtt
from flask_socketio import SocketIO
from flask_bootstrap import Bootstrap

from apscheduler.schedulers.background import BackgroundScheduler
from flask_apscheduler import APScheduler



from extract_voltage_task import extrct_digits_from_frame


eventlet.monkey_patch()

app = Flask(__name__, static_url_path='')
app.config.from_pyfile('settings.py')

mqtt = Mqtt(app)
socketio = SocketIO(app)
bootstrap = Bootstrap(app)

@app.route('/')
def index():
    return render_template('index.html')

@app.route('/image/')
def one_image():
    #需要改，换成实施的
    cap = cv2.VideoCapture("sample.mp4") # for testing
    ret, frame = cap.read()
    time_now = datetime.datetime.now().strftime("%Y-%m-%d-%H-%M-%S")
    cv2.imwrite("one_image_"+time_now+".jpg",frame)
    cap.release()
    return render_template('image.html',context = {"app_id":app.config['REMOTE_MQTT_PUBLISH_TOPIC'],"image_path":"one_image_"+time_now+".jpg"})
    
     
@socketio.on('publish')
def handle_publish(json_str):
    data = json.loads(json_str)
    mqtt.publish(data['topic'], data['message'])


@socketio.on('subscribe')
def handle_subscribe(json_str):
    data = json.loads(json_str)
    mqtt.subscribe(data['topic'])


@socketio.on('unsubscribe_all')
def handle_unsubscribe_all():
    mqtt.unsubscribe_all()


@mqtt.on_connect()
def handle_connect(client, userdata, flags, rc):
    mqtt.subscribe(app.config['REMOTE_MQTT_SUBSCRIBE_TOPIC'])


@mqtt.on_message()
def handle_mqtt_message(client, userdata, message):
    data = dict(
        topic=message.topic,
        payload=message.payload.decode()
    )
    socketio.emit('mqtt_message', data=data)
    # add receive message handler here
    print(data)
    download_config_file()


@mqtt.on_log()
def handle_logging(client, userdata, level, buf):
    print(level, buf)


@app.errorhandler(Exception)
def all_exception_handler(e):
    return 'Error', 500


def download_config_file():
	remote_config_file = "http://"+app.config['SERVER_IP']+":"+app.config['SERVER_PORT']+app.config['SERVER_CONFIG_PREFIX']+app.config['MQTT_CLIENT_ID']+"_config.py"
	r = requests.get(remote_config_file, stream=True)
	with open("settings.py",'wb') as f:
		# 分块写入文件
		for chunk in r.iter_content(chunk_size=1024):
			if chunk:
				f.write(chunk)
	f.close()
	print("request and download config file finished . programe will restarted.")
	socketio.run(app, host='0.0.0.0', port=8058, debug=True)

def AddJobConfig():
	  app.config.update(
        {"SCHEDULER_API_ENABLED": True,
         "SCHEDULER_TIMEZONE": 'Asia/Shanghai',
         "JOBS": [{"id": "my_job", # 任务ID
                    "func": "__main__:extract_voltage_send",#任务位置
                    "trigger": "interval", #触发器
                    "seconds": 60 # 时间间隔
                    },
                   {"id": "my_job2", # 任务ID
                    "func": "delete_task:delete_old_images",#任务位置
                    "trigger": "interval", #触发器
                    "seconds": 60 # 时间间隔
                    }
                ]}
        )


def extract_voltage_send():
    print('extract_voltage_send job started!')
    time_now = datetime.datetime.now().strftime("%Y-%m-%d-%H-%M-%S")
    app.logger.info(time_now+' extract_voltage_send job started!')
    #camera = cv2.VideoCapture(0) #use 0 for web camera  deplyoment
    cap = cv2.VideoCapture(0) # for testing
    #	以下设置显示屏的宽高
    cap.set(cv2.CAP_PROP_FRAME_WIDTH,	640)
    cap.set(cv2.CAP_PROP_FRAME_HEIGHT,	480)
    fails_count = 0 
    while True:
        ret, frame = cap.read()
        return_digital,job_status = extrct_digits_from_frame(frame,app.config['IMAGE_CLIP_COLS_START'],app.config['IMAGE_CLIP_COLS_END'],app.config['IMAGE_CLIP_ROWS_START'],app.config['IMAGE_CLIP_ROWS_END'])
        if job_status or fails_count > 10:
            mqtt.publish(app.config['REMOTE_MQTT_PUBLISH_TOPIC'], "{'voltage':"+str(return_digital)+"}")
            cv2.imwrite("./imageframe/image_frame_"+time_now+".jpg", cv2.putText(frame, str(return_digital), (app.config['IMAGE_CLIP_ROWS_START'], app.config['IMAGE_CLIP_COLS_START']), cv2.FONT_HERSHEY_COMPLEX, 1.5, (0, 255, 255), 2))
            cap.release()
            break
        else:
            fails_count +=1
            cv2.imwrite("./fails/image_frame_"+time_now+".jpg", frame)
    #if need_send _image :
    #	send_image

def make_dir(make_dir_path):
    path = make_dir_path.strip()
    if not os.path.exists(path):
        os.makedirs(path)

def init_logging():
    logFormatter = logging.Formatter(app.config["LOG_FORMAT"])
    fileHandler = RotatingFileHandler(
        app.config['LOG_FILE_FILENAME'],
        maxBytes=app.config['LOG_MAX_BYTES'],
        backupCount=3,
        encoding="UTF-8")
    fileHandler.setLevel(logging.DEBUG)
    fileHandler.setFormatter(logFormatter)
    app.logger.addHandler(fileHandler)
    streamHandler = logging.StreamHandler()
    streamHandler.setFormatter(logFormatter)
    streamHandler.setLevel(logging.INFO)
    app.logger.addHandler(streamHandler)
    app.logger.setLevel(logging.DEBUG) # set root log level


if __name__ == '__main__':
    init_logging()
    
    scheduler = APScheduler(BackgroundScheduler(timezone="Asia/Shanghai"))
    AddJobConfig()     # 配置任务，不然无法启动任务
    scheduler.init_app(app)
    scheduler.start()

    #app.run()
    socketio.run(app, host='0.0.0.0', port=8058, debug=True)
    #mqtt.subscribe(app.config['REMOTE_MQTT_SUBSCRIBE_TOPIC'])  // need to wait for mqtt client finished connection