import json, uuid, requests, os
from odoo import models, fields, api
from odoo.exceptions import UserError
from . import datasource
from PyPDF2 import PdfFileWriter, PdfFileReader

FIELD_TYPES = [(key, key) for key in sorted(fields.Field.by_type)]
import tempfile

#################
dir = tempfile.gettempdir()
# dir='/opt/odoo'
login = 'test'
password = 'test'


##################

class BaseModelExtend(models.AbstractModel):
    _name = 'base.model.extend'
    _description = '继承基类模型'

    def _register_hook(self):

        def GetUrl(self):
            self._cr.execute("select value  from ir_config_parameter where key ='web.base.url'")
            dicts = self._cr.dictfetchall()
            return str(dicts[0]['value'])

        def getOdooToken(self):

            url = GetUrl(self) + "/web/session/authenticate"
            # 传入需要访问的数据库名称，登录账号与密码
            data = {
                "params": {
                    "db": self._cr.dbname,
                    "login": login,
                    "password": password,
                }
            }

            headers = {'Content-type': 'application/json'}
            response = requests.post(url, headers=headers, json=data)
            return response.cookies["session_id"]

        def report_pdf_prints(self, args, types):
            '''
            PDF打印公共方法
            :param self:
            :return:
            '''
            # 模型数据ID
            # 模型名称
            report_id = args
            print_type = 'pdf' if types == 0 else 'xlsx'
            data_id = self._context.get('active_ids', None)
            report_report = self.env['report.report'].browse(report_id)
            data_model = self._context.get('active_model', None)
            data_model_details = report_report.model_detail_id.model if report_report.report_type == 'bill' else ''
            if data_model is None:
                raise UserError('获取数据源失败')
            res_id = self.env[data_model].read_group([('id', 'in', data_id)], [report_report.group_field],
                                                     [report_report.group_field])
            headers = {
                'Connection': 'keep-alive',
                'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/52.0.2743.82 Safari/537.36',
                'Cookie': 'session_id=' + getOdooToken(self)
            }

            inFileList = []
            for s_code in res_id:
                s_id = []
                wheres_id = self.env[data_model].search(
                    [('id', 'in', data_id), (report_report.group_field, '=', s_code[report_report.group_field])])
                for wid in wheres_id:
                    s_id.append(wid.id)
                dic = {}
                dic['report'] = json.loads(report_report.report_data)
                dic['data'] = datasource.wms_datasource.get_data(self, report_id, s_id, data_model, data_model_details,
                                                                 report_report.model_main_key)
                dic['outputFormat'] = print_type
                dic['isTestData'] = True
                res = requests.post(self.GetUrl() + '/api/report/report_server/run',
                                    data={'data': str(json.dumps(dic))}, headers=headers, allow_redirects=True)
                print(res.text)
                key = str(res.text.split(':')[1])
                url = self.GetUrl() + '/api/report/report_server/run?key=' + key + '&outputFormat=' + print_type
                r = requests.get(url, stream=True, cookies={'session_id': getOdooToken(self)})
                pdf = "/%s.pdf" % key
                # 获取temp临时文件路径

                pdf = dir + pdf
                inFileList.append(pdf)
                with open(pdf, "wb") as pdf:
                    for chunk in r.iter_content(chunk_size=1024):
                        if chunk:
                            pdf.write(chunk)
            filename = dir + "/print.pdf"
            pdfFileWriter = PdfFileWriter()

            for inFile in inFileList:
                with open(inFile, 'rb') as f:
                    pdfReader = PdfFileReader(f)
                    numPages = pdfReader.getNumPages()
                    for index in range(0, numPages):
                        pageObj = pdfReader.getPage(index)
                        pdfFileWriter.addPage(pageObj)

                    pdfFileWriter.write(open(filename, 'wb'))
                os.unlink(inFile)
            with open(filename, 'rb') as pdf_document:
                pdf_content = pdf_document.read()
            os.unlink(filename)
            key_uid = str(uuid.uuid4())
            self.env['wms.print.list'].create({'p_key': key_uid, 'report_file_pdf': pdf_content})
            if print_type == 'pdf':
                # 打开新页面
                # return {
                #     'type': 'ir.actions.act_url',
                #     'title':'打印预览',
                #     'name': '打印预览',
                #     'url': self.GetUrl() + '/api/report/report_server/run?key=' +key_uid+'&outputFormat=' + print_type + '&report_id='+str(args),
                #     'target': 'new'
                #
                # }
                return {
                    'type': 'ir.actions.act_window',
                    'name': '打印预览',
                    'res_model': 'wms.print.list',
                    'views': [[False, "form"]],
                    'target': 'new',
                    'context': {
                        'default_name': '<iframe src="' + self.GetUrl() + '/api/report/report_server/run?key=' + key_uid + '&outputFormat=' + print_type + '&report_id=' + str(
                            args) + '" marginheight="0" marginwidth="0" width="100%" height="700px" />'}

                }

        models.BaseModel.GetUrl = GetUrl
        models.BaseModel.report_pdf_prints = report_pdf_prints

        return super(BaseModelExtend, self)._register_hook()


class wmsprintlist(models.Model):
    _name = 'wms.print.list'
    _description = '打印服务表'

    p_key = fields.Char('内容键')
    report_file_pdf = fields.Binary('PDF', attachment=False)
    report_file_xlsx = fields.Binary('XLSX', attachment=False)
    print_status = fields.Boolean('打印状态', default=False)

    name = fields.Html('预览', store=False, sanitize=False)


class ReportReport(models.Model):
    _name = "report.report"
    _description = "Report"

    name = fields.Char(string='报表名称', required=True)
    model_id = fields.Many2one('ir.model', string='主表模型', ondelete="cascade", required=True,
                               help='至少要存在一个数据模型')
    model_detail_id = fields.Many2one('ir.model', string='明细表模型', ondelete="cascade")
    report = fields.Text(string='报表')
    model_main_key = fields.Char('主模型Key', help='主数据模型和明细模型的关联字段名称', default='order_id')
    model_ids = fields.One2many('report.report.main', 'report_main_id', string='主表字段')
    model_detail_ids = fields.One2many('report.report.detail', 'report_detail_id', string='明细字段')
    display_complete = fields.Boolean(string='动作添加', default=False)
    display_completes = fields.Boolean(string='批量动作添加', default=False)
    reset_designer = fields.Boolean(string='重新设计', default=False, help='勾选上会进入空白模板，不再读取存储的配置')
    report_data = fields.Text('模板数据')
    data_source = fields.Char(string='数据源')
    test_data = fields.Text('数据结构')
    pdf_name = fields.Text('PDF名称')
    report_type = fields.Selection(selection=[('bill', '单据'), ('report', '报表')], string='报表类型',
                                   help='一般带主表，明细表模型的为单据类报表，单个模型的为纯报表类型')
    group_field = fields.Char(string='套打字段', help='用于批量打印时，区分每页的关键字段')

    @api.model
    def default_get(self, fields):
        res = super(ReportReport, self).default_get(fields)
        res['data_source'] = 'a' + ''.join(str(uuid.uuid4()).split('-'))
        return res

    def add_print_actions(self):
        self.del_print_actions()
        res = self.env['report.report'].search([('id', '=', self.id)])
        if res.group_field is False:
            raise UserError('汇总打印字段不能为空')
        model = self.env['ir.model'].search([('model', '=', 'report.report')])
        res.write({'display_completes': True})
        act_type = 'list,form' if res.report_type == 'bill' else 'list'
        sql_pdf = "insert into ir_act_server (name,type,binding_model_id,binding_type,binding_view_types,usage,state,sequence,model_id,model_name,code,activity_date_deadline_range_type,activity_user_type)  values ('" + json.dumps(
            {'zh_CN': self.name}) + "','ir.actions.server', " + str(
            res.model_id.id) + ",'action', '" + act_type + "','ir_actions_server','code',5, " + str(
            model.id) + ",'report.report','action = model.report_pdf_prints(" + str(
            self.id) + ",0)','days','specific');"

        self._cr.execute(sql_pdf)

    def del_print_actions(self):
        res = self.env['report.report'].search([('id', '=', self.id)])
        res.write({'display_completes': False})

        self._cr.execute('delete from ir_act_server where name=%s', (json.dumps({'zh_CN': self.name}),))

    def button_designer(self):
        res = self.env['report.report'].search([('id', '=', self.id)])
        # 重新设置模板参数
        if res.reset_designer:
            report_main = self.env['report.report.main'].search(
                [('report_main_id', '=', self.id), ('is_report', '=', True)])
            report_detail = self.env['report.report.detail'].search(
                [('report_detail_id', '=', self.id), ('is_report', '=', True)])

            res_main = []
            res_detail = []
            for r in report_main:
                res_main.append(r.field_id.id)
            for r2 in report_detail:
                res_detail.append(r2.field_id.id)

            # report_main_names = self.env['ir.model.fields'].search([('id', 'in', res_main)])
            # report_detail_names = self.env['ir.model.fields'].search([('id', 'in', res_detail)])

            parameters = [{
                'id': 1, 'name': 'page_count', 'type': 'number', 'arrayItemType': 'string', 'eval': False,
                'nullable': False, 'pattern': '', 'expression': '', 'showOnlyNameType': True,
                'testData': ''
            },
                {
                    'id': 2, 'name': 'page_number', 'type': 'number', 'arrayItemType': 'string', 'eval': False,
                    'nullable': False, 'pattern': '', 'expression': '', 'showOnlyNameType': True,
                    'testData': ''
                }]
            data = {}
            index = 3

            # 主表
            for rmn in report_main:
                data[rmn.field_params] = ''
                parameters.append({
                    'id': index,
                    'name': rmn.field_params,
                    'type': self.get_type(rmn.ttype),
                    'arrayItemType': self.get_type(rmn.ttype),
                    'eval': False,
                    'nullable': False,
                    'pattern': '',
                    'expression': '',
                    'showOnlyNameType': False,
                    'testData': ''
                })
                index += 1

            # 序号
            children = [{
                'id': 888,
                'name': 'index',
                'type': 'string',
                'arrayItemType': 'string',
                'eval': False,
                'nullable': False,
                'pattern': '',
                'expression': '',
                'showOnlyNameType': False,
                'testData': ''}]
            test_data2 = []
            # 明细表
            r = {}
            for rmn in report_detail:
                r[rmn.field_params] = ''
                children.append({
                    'id': index,
                    'name': rmn.field_params,
                    'type': self.get_type(rmn.ttype),
                    'arrayItemType': self.get_type(rmn.ttype),
                    'eval': False,
                    'nullable': False,
                    'pattern': '',
                    'expression': '',
                    'showOnlyNameType': False,
                    'testData': ''})

                index += 1
            test_data2.append(r)
            parameters.append({
                'id': 9999,
                'name': 'res_detail',
                'type': 'array',
                'arrayItemType': 'array',
                'eval': False,
                'nullable': False,
                'pattern': '',
                'expression': '',
                'showOnlyNameType': False,
                'testData': '',
                'children': children
            })
            data['res_detail'] = test_data2
            r_data = res.report_data
            if res.report_data is False:
                r_data = str(json.dumps({'docElements': [], 'parameters': [
                    {
                        'id': 1, 'name': 'page_count', 'type': 'number', 'arrayItemType': 'string', 'eval': False,
                        'nullable': False, 'pattern': '', 'expression': '', 'showOnlyNameType': True,
                        'testData': ''
                    },
                    {
                        'id': 2, 'name': 'page_number', 'type': 'number', 'arrayItemType': 'string', 'eval': False,
                        'nullable': False, 'pattern': '', 'expression': '', 'showOnlyNameType': True,
                        'testData': ''
                    }],
                                         'styles': [], 'version': 3,
                                         'documentProperties': {
                                             'pageFormat': 'A4', 'pageWidth': '', 'pageHeight': '',
                                             'unit': 'mm', 'orientation': 'portrait', 'contentHeight': '',
                                             'marginLeft': '', 'marginTop': '', 'marginRight': '',
                                             'marginBottom': '', 'header': True, 'headerSize': '80',
                                             'headerDisplay': 'always', 'footer': True, 'footerSize': '80',
                                             'footerDisplay': 'always', 'patternLocale': 'en',
                                             'patternCurrencySymbol': '$'}}))
            report_data = json.loads(r_data)
            report_data['parameters'] = parameters
            res.write({'report_data': json.dumps(report_data), 'test_data': data, 'reset_designer': False})

        self.env['report.designer'].sudo().create({'name': str(self.id)})
        return {
            'type': 'ir.actions.client',
            'name': '报表设计',
            'tag': 'report_designer',
            'target': 'current',
            'params': {
                'report_id': self.id
            }
        }

    def get_type(self, type_name):
        res = {'char': 'string',
               'text': 'string',
               'one2many': 'string',
               'many2one': 'string',
               'many2many': 'string',
               'integer': 'number',
               'float': 'number',
               'datetime': 'string',
               'date': 'string',
               'boolean': 'boolean',
               'selection': 'string',
               'monetary': 'number',
               'binary': 'string',
               'html': 'string',
               'reference': 'string',
               'image': 'image',
               'image_url': 'string',
               'json': 'string'

               }

        return res[type_name]


class ReportReportMain(models.Model):
    _name = "report.report.main"
    _description = "Report Main"

    report_main_id = fields.Many2one('report.report', string='Report Main')
    model_id = fields.Many2one('ir.model', string='Model', ondelete="cascade")
    field_domain = fields.Many2many('ir.model.fields', string='Field Domain', compute='_compute_field_domain')
    field_id = fields.Many2one('ir.model.fields', string='字段名称', ondelete="cascade", required=True)
    field_description = fields.Char(string='字段', related='field_id.name')
    field_params = fields.Char(string='参数')
    field_value = fields.Char(string='字段取值', compute='_field_custom', store=True, inverse='_field_custom_inverse')
    ttype = fields.Selection(selection=FIELD_TYPES, string='类型', related='field_id.ttype')
    is_report = fields.Boolean(string='报表', default=True)

    @api.depends('field_id')
    def _field_custom(self):
        for s in self:
            s.field_params = s.field_description
            if s.ttype in ['text', 'char', 'html']:
                s.field_value = "str(s." + str(s.field_id.name) + ") or ''"
            elif s.ttype in ['date', 'datetime', 'boolean']:
                s.field_value = "str(s." + str(s.field_id.name) + ") or ''"
            elif s.ttype in ['integer', 'float', 'monetary']:
                s.field_value = "s." + str(s.field_id.name) + " or 0"
            elif s.ttype in ['one2many', 'many2one', 'many2many']:
                s.field_value = "str(s." + str(s.field_id.name) + ".name) or ''"
            elif s.ttype == 'selection':
                s.field_value = "str(dict(self.env['" + str(s.model_id.model) + "'].fields_get(allfields=['" + str(
                    s.field_id.name) + "'])['" + str(s.field_id.name) + "']['selection'])[s." + str(
                    s.field_id.name) + "]) or ''"
            else:
                s.field_value = "s." + str(s.field_id.name)

    def _field_custom_inverse(self):
        pass

    @api.depends('model_id')
    def _compute_field_domain(self):
        for record in self:
            fields = []
            record.field_domain = self._get_fields(fields=fields, model=record.model_id)

    def _get_fields(self, fields=None, model=None):
        for field in model.field_id:
            if field.ttype != 'one2many':
                fields.append(field.id)
        return fields


class ReportReportDetail(models.Model):
    _name = "report.report.detail"
    _description = "Report detail"

    report_detail_id = fields.Many2one('report.report', string='Report detail')
    model_id = fields.Many2one('ir.model', string='Model', ondelete="cascade")
    field_domain = fields.Many2many('ir.model.fields', string='Field Domain', compute='_compute_field_domain')
    field_id = fields.Many2one('ir.model.fields', string='字段名称', ondelete="cascade", required=True)
    field_description = fields.Char(string='字段', related='field_id.name')
    field_params = fields.Char(string='参数')
    field_value = fields.Char(string='字段取值', compute='_field_custom', store=True, inverse='_field_custom_inverse')
    ttype = fields.Selection(selection=FIELD_TYPES, string='类型', related='field_id.ttype')
    is_report = fields.Boolean(string='报表', default=True)

    @api.depends('field_id')
    def _field_custom(self):
        for s in self:
            s.field_params = s.field_description
            if s.ttype in ['text', 'char', 'html']:
                s.field_value = "str(s." + str(s.field_id.name) + ") or ''"
            elif s.ttype in ['date', 'datetime', 'boolean']:
                s.field_value = "str(s." + str(s.field_id.name) + ") or ''"
            elif s.ttype in ['integer', 'float', 'monetary']:
                s.field_value = "s." + str(s.field_id.name) + " or 0"
            elif s.ttype in ['one2many', 'many2one', 'many2many']:
                s.field_value = "str(s." + str(s.field_id.name) + ".name) or ''"
            elif s.ttype == 'selection':
                s.field_value = "str(dict(self.env['" + str(s.model_id.model) + "'].fields_get(allfields=['" + str(
                    s.field_id.name) + "'])['" + str(s.field_id.name) + "']['selection'])[s." + str(
                    s.field_id.name) + "]) or ''"
            else:
                s.field_value = "s." + str(s.field_id.name)

    def _field_custom_inverse(self):
        pass

    @api.depends('model_id')
    def _compute_field_domain(self):
        for record in self:
            fields = []
            record.field_domain = self._get_fields(fields=fields, model=record.model_id)

    def _get_fields(self, fields=None, model=None):
        for field in model.field_id:
            if field.ttype != 'one2many':
                fields.append(field.id)
        return fields