import json
from odoo import http
from odoo.http import request
from odoo.addons.sf_machine_connect.models.ftp_operate import transfer_nc_files

class MainController(http.Controller):

    @http.route('/api/manual_download_program', type='json', methods=['POST'], auth='public', cors='*')
    def manual_download_program(self):
        """
            人工线下加工传输编程文件
        """
        data = json.loads(request.httprequest.data)
        maintenance_equipment_name = data.get('maintenance_equipment_name')
        model_id = data.get('model_id')
        if not maintenance_equipment_name or not model_id:
            return {'code': 400, 'message': '参数错误'}
        maintenance_equipment = request.env['maintenance.equipment'].sudo().search([('name', '=', maintenance_equipment_name)], limit=1)
        if not maintenance_equipment:
            return {'code': 400, 'message': '机床不存在'}
        # 获取刀具组
        tool_groups_id = request.env['sf.tool.groups'].sudo().search([('equipment_ids', 'in', maintenance_equipment.id)], limit=1)
        if not tool_groups_id:
            return {'code': 400, 'message': '刀具组不存在'}

        ftp_resconfig = request.env['res.config.settings'].sudo().get_values()
        source_ftp_info = {
            'host': ftp_resconfig['ftp_host'],
            'port': int(ftp_resconfig['ftp_port']),
            'username': ftp_resconfig['ftp_user'],
            'password': ftp_resconfig['ftp_password']
        }
        target_ftp_info = {
            'host': maintenance_equipment.ftp_host,
            'port': int(maintenance_equipment.ftp_port),
            'username': maintenance_equipment.ftp_username,
            'password': maintenance_equipment.ftp_password
        }
        # 传输nc文件
        if transfer_nc_files(
            source_ftp_info,
            target_ftp_info,
            '/' + str(model_id),
            '/home/jikimo/testdir',
            end_with=tool_groups_id.name + '-all.nc'):
            return {'code': 200, 'message': 'success'}
        else:
            return {'code': 500, 'message': '传输失败'}