beauti-nof/application/common/service/MobileImport.php

<?php
namespace app\common\service;

use app\admin\model\Admin;
use app\admin\model\MobileUpload;
use app\common\model\Area;
use app\common\model\Mobile;
use PhpOffice\PhpSpreadsheet\Cell\Coordinate;
use PhpOffice\PhpSpreadsheet\IOFactory;
use PhpOffice\PhpSpreadsheet\Reader\Csv;
use PhpOffice\PhpSpreadsheet\Reader\Xls;
use PhpOffice\PhpSpreadsheet\Reader\Xlsx;
use think\Db;
use think\db\Query;
use think\Exception;

class MobileImport{
    public static function import($file,$admin_id,$type=1,$status=null,$mobileUpload=null){
        $time=time();
        if (!$file) {
            throw_user(__('Parameter %s can not be empty', 'file'));
        }
        if(!$mobileUpload) {
            $mobileUpload = new MobileUpload(['admin_id' => $admin_id]);
        }
        $file=parse_url($file,PHP_URL_PATH);
        $filePath = ROOT_PATH . DS . 'public' . DS . $file;
        $filenameArr=explode('.',$file);
        $filePathError=$filenameArr[0].'_err.'.$filenameArr[1];
        if (!is_file($filePath)) {
            throw_user(__('No results were found'));
        }
        //实例化reader
        $ext = pathinfo($filePath, PATHINFO_EXTENSION);
        if (!in_array($ext, ['csv', 'xls', 'xlsx'])) {
            throw_user(__('Unknown data format'));
        }
        $errLog=[];
        $succNum=0;
        $cities=[];
        if ($ext === 'csv') {
            $file = fopen($filePath, 'r');
            $filePath = tempnam(sys_get_temp_dir(), 'import_csv');
            $fp = fopen($filePath, "w");
            $n = 0;
            while ($line = fgets($file)) {
                $line = rtrim($line, "\n\r\0");
                $encoding = mb_detect_encoding($line, ['utf-8', 'gbk', 'latin1', 'big5']);
                if ($encoding != 'utf-8') {
                    $line = mb_convert_encoding($line, 'utf-8', $encoding);
                }
                if ($n == 0 || preg_match('/^".*"$/', $line)) {
                    fwrite($fp, $line . "\n");
                } else {
                    fwrite($fp, '"' . str_replace(['"', ','], ['""', '","'], $line) . "\"\n");
                }
                $n++;
            }
            fclose($file) || fclose($fp);

            $reader = new Csv();
        } elseif ($ext === 'xls') {
            $reader = new Xls();
        } else {
            $reader = new Xlsx();
        }
        //$newErrorExcel=IOFactory::createReader(ucfirst($ext));
        if(!empty($mobileUpload['filename_error'])){
            //$filePathErrorPath=ROOT_PATH . DS . 'public' . DS . parse_url($mobileUpload['filename_error'],PHP_URL_PATH);
        }else{
            /*$filePathErrorPath=ROOT_PATH . DS . 'public' . DS .$filePathError;
            copy($filePath,$filePathErrorPath);*/
            $mobileUpload['filename_error']=sprintf("%s%s",request()->domain(),$filePath);
        }
        //$newErrorExcelLoaded=$newErrorExcel->load($filePathErrorPath);
        //导入文件首行类型,默认是注释,如果需要使用字段名称请使用name
        $importHeadType = 'comment';

        /*$table = (new Mobile)->getTable();
        $database = \think\Config::get('database.database');*/
        //$fieldArr = [];
        /*  $list = db()->query("SELECT COLUMN_NAME,COLUMN_COMMENT FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_NAME = ? AND TABLE_SCHEMA = ?", [$table, $database]);
        foreach ($list as $k => $v) {
             if ($importHeadType == 'comment') {
                 $fieldArr[$v['COLUMN_COMMENT']] = $v['COLUMN_NAME'];
             } else {
                 $fieldArr[$v['COLUMN_NAME']] = $v['COLUMN_NAME'];
             }
         }*/
        $fieldArr=[
            "手机号"=>'no',
            "卡名称"=>'name',
            "省份"=>'province',
            "归属地"=>'city',
            "运营商"=>'network',
            "卡品牌"=>'brand',
            "原价"=>'amount_original',
            "底价"=>'amount_di',
            "售价"=>'amount_base',
            "预存话费"=>'amount_charge',
            "代理价"=>'amount_proxy',
            "备注(仅我可见)"=>'remark_me',
            "备注(全部可见)"=>'remark',
            "号码状态"=>'status',
            "供应商"=>'proxy_id',
            "批号"=>'batch_no',
        ];
        //dd($fieldArr);
        $infoArr=[
            '免流APP'=>'free_app',
            '资费套餐'=>'describe',
            '注意事项'=>'content',
            '每年流量(G'=>'flow_year',
            '费用'=>'fee',
            '首月免月租'=>'first_month_free',
            '前多少名免单'=>'flow_free_limit',
            '简介'=>'describe',
        ];
        $proxyInfo=[
            '供应商电话'=>'proxy_phone',
        ];
        //$admin=Admin::get($admin_id);
        //加载文件
        $insert = [];
        try {
            if (!$PHPExcel = $reader->load($filePath)) {
                throw_user(__('Unknown data format'));
            }

            $currentSheet = $PHPExcel->getSheet(0);  //读取文件中的第一个工作表
            $allColumn = $currentSheet->getHighestDataColumn(); //取得最大的列号
            $errorWriteColumn=++$allColumn;
            $PHPExcel->setActiveSheetIndex(0)->setCellValue($errorWriteColumn.'1',"错误信息");

            $allRow = $currentSheet->getHighestRow(); //取得一共有多少行
            $maxColumnNumber = Coordinate::columnIndexFromString($allColumn);
            $fields = [];
            for ($currentRow = 1; $currentRow <= 1; $currentRow++) {
                for ($currentColumn = 1; $currentColumn <= $maxColumnNumber; $currentColumn++) {
                    $val = $currentSheet->getCellByColumnAndRow($currentColumn, $currentRow)->getCalculatedValue();
                    $fields[] = preg_replace("/\s+/",'',$val);
                }
            }
            $dataRowNum=0;
            for ($currentRow = 2; $currentRow <= $allRow; $currentRow++) {
                $values = [];
                for ($currentColumn = 1; $currentColumn <= $maxColumnNumber; $currentColumn++) {
                    $val = $currentSheet->getCellByColumnAndRow($currentColumn, $currentRow)->getCalculatedValue();
                    $values[] = is_null($val) ? '' : $val;
                }
                $row = [];
                $row_info = [];
                $proxy_info = [];
                $temp = array_combine($fields, $values);
                if(!array_filter(array_values($temp))){
                    break;
                }
                foreach ($temp as $k => $v) {
                    if (isset($fieldArr[$k]) && $k !== '') {
                        $row[$fieldArr[$k]] = $v;
                    }
                    if(isset($infoArr[$k]) && $k!==''){
                        $row_info[$infoArr[$k]]=$v;
                    }
                    if(isset($proxyInfo[$k]) && $k!==''){
                        $proxy_info[$proxyInfo[$k]]=$v;
                    }
                }

                if(!$row['no'] || strlen($row['no'])!=11 || !is_numeric($row['no'])){
                    $errLog[]=[
                        'line' =>$currentRow,
                        'error'=>$errorMsg='手机号有误',
                    ];
                    $PHPExcel->setActiveSheetIndex(0)->setCellValue($errorWriteColumn.$currentRow,$errorMsg);
                    goto NEXT;
                }
                #价格
                if(!self::checkAmount($row,'amount_original')){
                    $errLog[]=[
                        'line' =>$currentRow,
                        'error'=>$errorMsg='原价有误',
                    ];
                    $PHPExcel->setActiveSheetIndex(0)->setCellValue($errorWriteColumn.$currentRow,$errorMsg);
                    goto NEXT;
                }
                if(!self::checkAmount($row,'amount_di')){
                    $errLog[]=[
                        'line' =>$currentRow,
                        'error'=>$errorMsg='底价有误',
                    ];
                    $PHPExcel->setActiveSheetIndex(0)->setCellValue($errorWriteColumn.$currentRow,$errorMsg);
                    goto NEXT;
                }
                if(!self::checkAmount($row,'amount_base')){
                    $errLog[]=[
                        'line' =>$currentRow,
                        'error'=>$errorMsg='售价有误',
                    ];
                    $PHPExcel->setActiveSheetIndex(0)->setCellValue($errorWriteColumn.$currentRow,$errorMsg);
                    goto NEXT;
                }
                if(!self::checkAmount($row,'amount_charge')){
                    $errLog[]=[
                        'line' =>$currentRow,
                        'error'=>$errorMsg='预充价格有误',
                    ];
                    $PHPExcel->setActiveSheetIndex(0)->setCellValue($errorWriteColumn.$currentRow,$errorMsg);
                    goto NEXT;
                }
                if(!self::checkAmount($row,'amount_proxy')){
                    $errLog[]=[
                        'line' =>$currentRow,
                        'error'=>$errorMsg='代理价格有误',
                    ];
                    $PHPExcel->setActiveSheetIndex(0)->setCellValue($errorWriteColumn.$currentRow,$errorMsg);
                    goto NEXT;
                }
                if($row['amount_di']>$row['amount_base']){
                    $errLog[]=[
                        'line' =>$currentRow,
                        'error'=>$errorMsg='底价不能大于售价',
                    ];
                    $PHPExcel->setActiveSheetIndex(0)->setCellValue($errorWriteColumn.$currentRow,$errorMsg);
                    goto NEXT;
                }
                $city=null;
                if(isset($cities[$row['city']])){
                    $city=$cities[$row['city']];
                }else{
                    $city=Area::shi($row['city'])->find();
                }
                if(!$city){
                    $errLog[]=[
                        'line' =>$currentRow,
                        'error'=>$errorMsg='城市有误',
                    ];
                    $PHPExcel->setActiveSheetIndex(0)->setCellValue($errorWriteColumn.$currentRow,$errorMsg);
                    goto NEXT;
                }
                $cities[$row['city']]=$city;
                $row['city_id']=$city['id'];
                $row['province_id']=$city['pid'];
                Db::startTrans();
                if($admin_id) {
                    if($type==1) {
                        if(empty($row['proxy_id'])){
                            $errLog[]=[
                                'line' =>$currentRow,
                                'error'=>$errorMsg='供应商不能为空',
                            ];
                            $PHPExcel->setActiveSheetIndex(0)->setCellValue($errorWriteColumn.$currentRow,$errorMsg);
                            goto NEXT;
                        }
                        $proxyName=$row['proxy_id'];
                        $row['proxy_id'] = Admin::where(function (Query $query)use ($proxyName,$proxy_info){
                            $query->where('username', $proxyName)->whereOr('username',$proxy_info['proxy_phone'])->whereOr('com_mobile',$proxy_info['proxy_phone']);
                        })->where('proxy',1)->value('id');
                        if (!$row['proxy_id']) {
                            $proxy=Admin::createProxy($proxyName,$proxy_info['proxy_phone']);
                            $row['proxy_id']=$proxy['id'];
                        }
                    }
                }else{
                    $row['proxy_id']=null;
                }

                $row['admin_id']=$admin_id;
                $row['type']=$type;
                if(!is_null($status)){
                    $row['status']=$status;
                }else{
                    $row['status']=Mobile::importStatus($row['status']);
                }

                $mobile=(new Mobile);
                $mobile->allowField(true)->save($row);
                if(!$mobile){
                    $errLog[]=[
                        'line' =>$currentRow,
                        'error'=>$errorMsg='保存手机号失败',
                    ];
                    $PHPExcel->setActiveSheetIndex(0)->setCellValue($errorWriteColumn.$currentRow,$errorMsg);
                    goto NEXT;
                }
                $info=$mobile->info()->save($row_info);
                if(!$info){
                    $errLog[]=[
                        'line' =>$currentRow,
                        'error'=>$errorMsg='保存手机号套餐失败',
                    ];
                    $PHPExcel->setActiveSheetIndex(0)->setCellValue($errorWriteColumn.$currentRow,$errorMsg);
                    goto NEXT;
                }
                $succNum++;
                $mobileUpload['rows_succ']=$succNum;
                $mobileUpload->save();
                Db::commit();

                NEXT:
                Db::rollback();
                $dataRowNum++;
                usleep(500000);
            }
            $mobileUpload['rows_total']=$dataRowNum;
            $mobileUpload['rows_fail']=count($errLog);
            $mobileUpload['run_time']=time()-$time;
            $mobileUpload['status']=2;
            $mobileUpload->save();
            if($errLog) {
                $mobileUpload->log()->saveAll($errLog);
            }
            $max=1;
            if($errLog) {
                while ($max < $allRow) {
                    for ($currentRow = 2; $currentRow <= $allRow; $currentRow++) {
                        if (!$PHPExcel->getSheet(0)->getCell($errorWriteColumn . $currentRow)->getValue()) {
                            //$PHPExcel->setActiveSheetIndex(0)->removeRow($currentRow);
                            break;
                        }
                    }
                    $max++;
                }
                $errorWriter = IOFactory::createWriter($PHPExcel, ucfirst($ext));
                $errorWriter->save($filePath);
            }
        } catch (Exception $exception) {
            throw $exception;
            throw_user($exception->getMessage());
        }
        unset($PHPExcel,$errorWriter,$errLog);
    }
    public static function saveFile($file,$admin_id,$type=1,$status=null){
        $file=parse_url($file,PHP_URL_PATH);
        //$filenameArr=explode('.',$file);
        //$filePathError=$filenameArr[0].'_err.'.$filenameArr[1];
        $filePath = ROOT_PATH . DS . 'public' . DS . $file;
        //$filePathErrorPath=ROOT_PATH . DS . 'public' . DS .$filePathError;
        //copy($filePath,$filePathErrorPath);
        MobileUpload::create([
            'admin_id'=>$admin_id,
            'file'=>$filePAth=parse_url($file,PHP_URL_PATH),
            'filename'=>basename($filePAth),
            'filename_error'=>sprintf("%s%s",request()->domain(),$file),
            'params'=>[
                'type'=>$type,
                'status'=>$status,
            ]
        ]);
    }

    public static function checkAmount($row,$field){
        if(!isset($row[$field])){
            return true;
        }
        if(!is_numeric($row[$field])){
            return false;
        }
        if($row[$field]<0){
            return false;
        }
        return true;
    }
}