import random
import string

import gevent
from flask import make_response, send_from_directory, current_app
from pypinyin import lazy_pinyin, Style

from models.inStockExcel import InStockExcel
from utils import instockHelper
from utils.BatchCode import GetBatchCode
from utils.apiDoc import *
import os
import pandas as pd


filePath = './file/'


@file.route('/inStock')
class FileInStock(Resource):
    @staticmethod
    def post():
        token = request.headers.get('X-Token')
        inStockTime = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
        excelFile = request.files.get('files')
        batchCode = GetBatchCode()
        batch_code_list = []
        is_his_file = False
        if not os.path.exists(filePath):
            os.makedirs(filePath)
        fileName = ''.join(random.choices(string.ascii_uppercase +
                                          string.digits, k=16)) + \
                   os.path.splitext(excelFile.filename)[-1]
        excelFile.save(filePath + fileName)
        df = pd.read_excel(filePath + fileName, sheet_name='Sheet1')
        if df.keys()[0] == '出库单号':
            is_his_file = True
            df = pd.read_excel(filePath + fileName, sheet_name='Sheet1',
                               names=['Code', 'Name', 'Specification', 'SupplierName', 'Maker', 'CertificationCode',
                                      'ProductionBatch', 'Position', 'TypeName', 'Unit', 'SmallestUnit',
                                      'Ratio', 'UnitPrice', 'SellingPrice', 'MaxValue', 'MinValue',
                                      'Number', 'Expiration'])
        else:
            df = pd.read_excel(filePath + fileName, sheet_name='Sheet1',
                               names=['Name', 'Specification', 'SupplierName', 'Maker', 'CertificationCode',
                                      'ProductionBatch', 'Position', 'TypeName', 'Unit', 'SmallestUnit',
                                      'Ratio', 'UnitPrice', 'SellingPrice', 'MaxValue', 'MinValue',
                                      'Number', 'Expiration'])
        # 文件转实体
        for index in df.index:
            data = InStockExcel()
            setattr(data, 'Name', str(df['Name'][index]).strip())
            setattr(data, 'Specification', str(df['Specification'][index]))
            setattr(data, 'SupplierName',
                    '' if pd.isnull(df['SupplierName'][index]) else str(df['SupplierName'][index]))
            setattr(data, 'Maker', str(df['Maker'][index]))
            setattr(data, 'CertificationCode',
                    '' if pd.isnull(df['CertificationCode'][index]) else str(df['CertificationCode'][index]))
            setattr(data, 'ProductionBatch',
                    '' if pd.isnull(df['ProductionBatch'][index]) else str(df['ProductionBatch'][index]))
            setattr(data, 'Position', str(df['Position'][index]))
            setattr(data, 'TypeName', str(df['TypeName'][index]))
            setattr(data, 'Unit', str(df['Unit'][index]))
            setattr(data, 'SmallestUnit',
                    None if pd.isnull(df['SmallestUnit'][index]) else str(df['SmallestUnit'][index]))
            setattr(data, 'Ratio',
                    None if pd.isnull(df['Ratio'][index]) else int(df['Ratio'][index]))
            setattr(data, 'UnitPrice', float(df['UnitPrice'][index]))
            setattr(data, 'SellingPrice', float(df['SellingPrice'][index]))
            setattr(data, 'MaxValue', int(df['MaxValue'][index]))
            setattr(data, 'MinValue', int(df['MinValue'][index]))
            setattr(data, 'Number', int(df['Number'][index]))
            setattr(data, 'Expiration', str(df['Expiration'][index]))
            if is_his_file:
                batchCode = str(df['Code'][index])
            batch_code_list.append(batchCode)
            # 入库信息写入
            instockHelper.in_stock(data, batchCode, inStockTime, token, request.json, remark='文件入库')
        try:
            # 是否入库成功
            db.session.commit()
        except gevent.Timeout:
            # 回调
            db.session.invalidate()
            raise
            return SuccessResponse(ResultCode.ERROR, None, None)
        except Exception:
            # 回调
            db.session.rollback()
            raise
            return SuccessResponse(ResultCode.ERROR, None, None)
        os.remove(filePath + fileName)
        return SuccessResponse(ResultCode.SUCCESS, None, fileName)


@file.route('/downloadInStock')
class DownloadInStock(Resource):
    @staticmethod
    def get():
        uploads = os.path.join(os.path.abspath(os.path.dirname(current_app.root_path)),
                               app.config[request.args.get('path')])
        app.logger.info(uploads)
        return send_from_directory(directory=uploads, filename=request.args.get('fileName'))