# codinf:utf-8
from warnings import catch_warnings
from flask import request
from app import app
import json
import time
import re
import timeit
import pexpect
import os
import csv
import logging
import requests
from config import Config
from werkzeug.utils import secure_filename
from app.utils.serial_communication import SerialCommunication
#from serial_config import ReadConfig
from flask_babel import Babel, gettext
import app.utils.g as gg

basepath = os.path.dirname(__file__)
ip_config = {'post':'wait'}
try:
    uploads_path = basepath + Config.STATE_PATH
    with open(uploads_path + 'ipconfig.csv','r') as f:
        reader = csv.reader(f)
        current = [row for row in reader]
        ip_config['ip'] = current[0][0]
        ip_config['cid'] = current[0][1]
        logging.warning("config:{}, ip&cid...:{}".format(uploads_path,ip_config))
except Exception as e:
    logging.warning("ip,cid未配置")

@app.route('/getstate',methods=['POST'])
def getstate():
    state_info = []
    version = read_version()
    ser = SerialCommunication()
    serial_state=ser.check_serial()
    # serial_cf = ReadConfig()
    addrs = read_addrs()
    addrs=re.sub('A1(\d)',r'B\1',addrs)
    serial_num = read_com()
    ad_serial = "{}：{},{}：{}".format(gettext("串口号"),serial_num,gettext("地址"),addrs)
    c_dict = {
        'state':gg.current_state,
        'msg':'未进行测试动作',
        'ipconfig':ip_config,
        'version':version,
        'serial_state':serial_state,
        'ad_serial':ad_serial
        }
    state_info.append(c_dict)
    return json.dumps(state_info)

@app.route('/taillog',methods=['POST'])
def taillog():
    a=os.popen("tail -n 20 /prog/smartshelf/logs/smart.log").readlines()
    s = '<br>'
    p = s.join(a)
    poststate = ip_config['post']
    jsondata = {'msg':p,'poststate':poststate}
    return json.dumps(jsondata)

@app.route('/updateip',methods=['POST'])
def updateip():
    data = request.get_json()
    confirm_ip(data)
    logging.warning("ip&cid:{}".format(ip_config))
    ip_info = []
    ip_info.append(ip_config)
    return json.dumps(ip_info)

def confirm_state(state):
    uploads_path = basepath + Config.STATE_PATH
    with open(uploads_path + "state.txt","w") as f:
        f.write(state)
def read_state():
    uploads_path = basepath + Config.STATE_PATH
    with open(uploads_path + "state.txt","r") as f:
        state=f.readline()
    if state.isspace():
        state='off'
    return state

from app.utils.g import ver,shelfconfig
def read_version():
    return ver()

def confirm_ip(data):
    uploads_path = basepath + Config.STATE_PATH
    sipconfig = read_ip()
    with open(uploads_path + 'ipconfig.csv','w') as f:
        if data['ip'] and not data['cid']:
            data['cid'] = sipconfig[0][1].strip()
        elif data['cid'] and not data['ip']:
            data['ip'] = sipconfig[0][0].strip()
        fieldnames = ['ip', 'cid']
        writer = csv.DictWriter(f,fieldnames=fieldnames)
        writer.writerow(data)
    ip_config['ip'] = data['ip']
    ip_config['cid'] = data['cid']
    shelfconfig.cid=ip_config['cid']
    shelfconfig.server_addr = ip_config['ip']
    shelfconfig.Save()
    confirm_autostart()


def confirm_autostart():
    # server_add = ip_config['ip'] + '/accshelf/' + ip_config['cid']
    result = re.findall(r"\b(?:(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.){3}(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\b", ip_config['ip'])
    if result:
        ip = result[0]
        server_add = "http://{}/smf".format(ip)
    else:
        server_add = "127.0.0.1:5000"
    line1 = "Comment = star chrome and enter the website for \"{}\"".format(server_add)
    line2 = "Exec = chromium-browser -kiosk \"{}\"".format(server_add)
    strs=["[Desktop Entry]","Name = Chrome Autostar","Type = Application"]
    strs.append(line1)
    strs.append(line2)
    filename = "/home/pi/.config/autostart/auto.desktop"
    if os.path.isfile(filename):
        os.remove(filename)
        os.mknod(filename)
        cmd = "chmod 777 " + filename
        run = pexpect.spawn('su -c "%s" root' %cmd)
        time.sleep(0.5)
        run.sendline('123')
        time.sleep(0.1)     
        with open("/home/pi/.config/autostart/auto.desktop","a") as f:
            for i in strs:
                f.write("{}\n".format(i))
            f.close()
    else:
        try:
            os.mknod(filename)
            cmd = "chmod 777 " + filename
            run = pexpect.spawn('su -c "%s" root' %cmd)
            time.sleep(0.5)
            run.sendline('123')
            time.sleep(0.1)     
            with open("/home/pi/.config/autostart/auto.desktop","a") as f:
                for i in strs:
                    f.write("{}\n".format(i))
                f.close()
        except Exception as e:
            logging.warning(e)

    logging.warning("自启动脚本更新成功")

def read_ip():
    uploads_path = basepath + Config.STATE_PATH
    if not os.path.exists(uploads_path + 'ipconfig.csv'):
        return [['','']]
    with open(uploads_path + 'ipconfig.csv','r') as f:
        reader = csv.reader(f)
        sipconfig = [row for row in reader]
    return sipconfig

# 上传配置文件
@app.route('/upload/', methods=['GET', 'POST'])
def upload_file():
    if request.method == 'POST':
        # check if the post request has the file part
        if 'file' not in request.files:
            flash('No file part')
            return redirect(request.url)
        file = request.files['file']
        # if user does not select file, browser also
        # submit an empty part without filename
        if file.filename == '':
            flash('No selected file')
            return redirect(request.url)
        if file:
            basepath = os.path.dirname(__file__)
            uploads_path = Config.UPLOAD_FOLDER
            # print (uploads_path)
            # print (Config.UPLOAD_FOLDER)
            filename = secure_filename(file.filename)
            file.save(os.path.join(uploads_path, filename))
            # reload_config()
            return  '{"filename":"%s"}' % filename
    return True
import re
# 更新串口以及地址配置
@app.route('/update_serial', methods=['POST'])
def update_serial():
    uploads_path = basepath + Config.STATE_PATH
    # serialconfig = read_serial()
    data = request.get_json()
    if data.get('serial_num','') and not data.get('addrs',''):
        confirm_com(data['serial_num'])
    elif data.get('addrs','') and not data.get('serial_num',''):
        confirm_addrs(data['addrs'])
    elif data.get('addrs','') and data.get('serial_num',''):
        confirm_com(data['serial_num'])
        confirm_addrs(data['addrs'])
    serial_num = read_com()
    addrs = read_addrs()
    addrs=re.sub('A1(\d)',r'B\1',addrs)
    ad_serial = "{}：{},{}：{}".format(gettext("串口号"),serial_num,gettext("地址"),addrs)
    return json.dumps({"addr":ad_serial})

def read_com():
    uploads_path = basepath + Config.STATE_PATH
    with open(uploads_path + "serialcom.txt","r") as f:
        serialcom=f.readline()
    shelfconfig.serial_port=serialcom
    return serialcom

def read_addrs():
    uploads_path = basepath + Config.STATE_PATH
    with open(uploads_path + "addrs.txt","r") as f:
        addrs=f.readline()
    shelfconfig.addrs = addrs
    addrs=addrs.replace('B','A1')
    addrs=addrs.replace('b','A1')
    return addrs

def confirm_com(state):
    uploads_path = basepath + Config.STATE_PATH
    with open(uploads_path + "serialcom.txt","w") as f:
        comnum="/dev/ttyUSB{}".format(state)
        f.write(comnum)
        f.close()
    shelfconfig.serial_port=comnum
    shelfconfig.Save()

def confirm_addrs(state):
    uploads_path = basepath + Config.STATE_PATH
    with open(uploads_path + "addrs.txt","w") as f:
        f.write(state)
        f.close()
    shelfconfig.addrs=state
    shelfconfig.Save()

@app.route('/sendcolor',methods=['POST'])
def sendcolor():
    COLORS = Config.COLORS
    color_list = list(COLORS.values())
    status_list = []
    # color_lists = [color_list[i:i+2] for i in range(0,len(color_list),2)]
    addrs = read_addrs()
    addrs_string=addrs.split('@')
    addr_list = filter(None,list(map(remove_a,addrs_string)))
    for saddr in addr_list:
        # for dlist in color_lists:
        colorset_dict = {"ADDR":saddr,"colorset":color_list}
        colorset_command = str(json.dumps(colorset_dict)).replace(" ","")
        colorset_command = colorset_command + '\n'
        print (colorset_command)
        ser = SerialCommunication()
        ser.send_data(colorset_command)
        status=ser.read_Line()
        status_list.append(status)
        print(status)
        time.sleep(0.5)
    logging.warning("颜色集合同步状态:{}".format(status_list))
    return json.dumps({"msg":"颜色集合同步状态:{}".format(status_list)})

from app.utils.g import shelfconfig
@app.route('/setconfig',methods=['POST'])
def setconfig():
    data = request.get_json()
    cf={}
    for k,v in shelfconfig.__dict__.items():
        if k in data:
            print(k,v,data.get(k))
            setattr(shelfconfig,k,data.get(k))
            cf[k]=data.get(k)
    if len(cf)>0:
       shelfconfig.Save()
    return json.dumps(cf)

@app.route('/setMC',methods=['POST'])
def setMC():
    data = request.data.decode('UTF-8')+'\n'
    ser = SerialCommunication()
    ser.send_data(data)
    return "ok"

def remove_a(s):
    number = re.findall("\d+",s)
    if number:
        return number[0]

def get_towerstate():
    # tower_state = {
    #             '1':{'action':False,'status':'off'},
    #             '2':{'action':False,'status':'off'},
    #             '3':{'action':True,'status':'on'},
    #             '4':{'action':False,'status':'off'},
    #             '5':{'action':False,'status':'off'},
    #             '6':{'action':True,'status':'on'}
    #             }
    tower_state = {
                '1@4':{'action':False,'status':'off'},
                '2@5':{'action':False,'status':'off'},
                '3@6':{'action':True,'status':'on'},
                '4@1':{'action':False,'status':'off'},
                '5@2':{'action':False,'status':'off'},
                '6@3':{'action':True,'status':'on'},
                '1@1':{'action':False,'status':'off'},
                '2@2':{'action':False,'status':'off'},
                '3@3':{'action':True,'status':'on'},
                '1':{'action':False,'status':'off'},
                '2':{'action':False,'status':'off'},
                '3':{'action':True,'status':'on'},
                '4':{'action':False,'status':'off'},
                '5':{'action':False,'status':'off'},
                '6':{'action':True,'status':'on'}
                }
    return tower_state

def testCidExists():
    cid = ip_config['cid']
    testurl = ip_config['ip'] + '/service/store/cidIsExists'
    data = {"cid":cid}
    try:
        response = requests.get(testurl, params=data,timeout=2)
        if response.status_code != 200:
            return None
        response=response.json() 
    except Exception as e:
        return None
       
    if response['okResult']:
        return True
    else:
        return False