作者 开飞机的舒克

后台导入功能

... ... @@ -38,140 +38,143 @@ class Study extends Backend
$this->view->assign("genderList", $this->model->getGenderList());
}
/*
* 导入
*/
public function import(){
$file = $this->request->request('file');
if (!$file) {
$this->error(__('Parameter %s can not be empty', 'file'));
}
$filePath = ROOT_PATH . DS . 'public' . DS . $file;
if (!is_file($filePath)) {
$this->error(__('No results were found'));
}
//实例化reader
$ext = pathinfo($filePath, PATHINFO_EXTENSION);
if (!in_array($ext, ['csv', 'xls', 'xlsx'])) {
$this->error(__('Unknown data format'));
}
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();
}
//导入文件首行类型,默认是注释,如果需要使用字段名称请使用name
$importHeadType = isset($this->importHeadType) ? $this->importHeadType : 'comment';
$table = $this->model->getQuery()->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') {
$v['COLUMN_COMMENT'] = explode(':', $v['COLUMN_COMMENT'])[0]; //字段备注有:时截取
$fieldArr[$v['COLUMN_COMMENT']] = $v['COLUMN_NAME'];
} else {
$fieldArr[$v['COLUMN_NAME']] = $v['COLUMN_NAME'];
}
}
//加载文件
$insert = [];
try {
if (!$PHPExcel = $reader->load($filePath)) {
$this->error(__('Unknown data format'));
}
$currentSheet = $PHPExcel->getSheet(0); //读取文件中的第一个工作表
$allColumn = $currentSheet->getHighestDataColumn(); //取得最大的列号
$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)->getValue();
$fields[] = $val;
}
}
for ($currentRow = 2; $currentRow <= $allRow; $currentRow++) {
$values = [];
for ($currentColumn = 1; $currentColumn <= $maxColumnNumber; $currentColumn++) {
$val = $currentSheet->getCellByColumnAndRow($currentColumn, $currentRow)->getValue();
$values[] = is_null($val) ? '' : $val;
}
$row = [];
$temp = array_combine($fields, $values);
foreach ($temp as $k => $v) {
if (isset($fieldArr[$k]) && $k !== '') {
$row[$fieldArr[$k]] = $v;
}
}
if ($row) {
$insert[] = $row;
}
}
} catch (Exception $exception) {
$this->error($exception->getMessage());
}
if (!$insert) {
$this->error(__('No rows were updated'));
}
try {
//是否包含admin_id字段
$has_admin_id = false;
foreach ($fieldArr as $name => $key) {
if ($key == 'admin_id') {
$has_admin_id = true;
break;
}
}
if ($has_admin_id) {
$auth = Auth::instance();
foreach ($insert as &$val) {
if (!isset($val['admin_id']) || empty($val['admin_id'])) {
$val['admin_id'] = $auth->isLogin() ? $auth->id : 0;
}
}
}
$this->model->saveAll($insert);
} catch (PDOException $exception) {
$msg = $exception->getMessage();
if (preg_match("/.+Integrity constraint violation: 1062 Duplicate entry '(.+)' for key '(.+)'/is", $msg, $matches)) {
$msg = "导入失败,包含【{$matches[1]}】的记录已存在";
};
$this->error($msg);
} catch (Exception $e) {
$this->error($e->getMessage());
}
$this->success();
return parent::import();
}
// /*
// * 导入
// */
// public function import(){
// $file = $this->request->request('file');
// if (!$file) {
// $this->error(__('Parameter %s can not be empty', 'file'));
// }
// $filePath = ROOT_PATH . DS . 'public' . DS . $file;
// if (!is_file($filePath)) {
// $this->error(__('No results were found'));
// }
// //实例化reader
// $ext = pathinfo($filePath, PATHINFO_EXTENSION);
// if (!in_array($ext, ['csv', 'xls', 'xlsx'])) {
// $this->error(__('Unknown data format'));
// }
// 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();
// }
//
// //导入文件首行类型,默认是注释,如果需要使用字段名称请使用name
// $importHeadType = isset($this->importHeadType) ? $this->importHeadType : 'comment';
//
// $table = $this->model->getQuery()->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') {
// $v['COLUMN_COMMENT'] = explode(':', $v['COLUMN_COMMENT'])[0]; //字段备注有:时截取
// $fieldArr[$v['COLUMN_COMMENT']] = $v['COLUMN_NAME'];
// } else {
// $fieldArr[$v['COLUMN_NAME']] = $v['COLUMN_NAME'];
// }
// }
//
// //加载文件
// $insert = [];
// try {
// if (!$PHPExcel = $reader->load($filePath)) {
// $this->error(__('Unknown data format'));
// }
// $currentSheet = $PHPExcel->getSheet(0); //读取文件中的第一个工作表
// $allColumn = $currentSheet->getHighestDataColumn(); //取得最大的列号
// $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)->getValue();
// $fields[] = $val;
// }
// }
//
// for ($currentRow = 2; $currentRow <= $allRow; $currentRow++) {
// $values = [];
// for ($currentColumn = 1; $currentColumn <= $maxColumnNumber; $currentColumn++) {
// $val = $currentSheet->getCellByColumnAndRow($currentColumn, $currentRow)->getValue();
// $values[] = is_null($val) ? '' : $val;
// }
// $row = [];
// $temp = array_combine($fields, $values);
// foreach ($temp as $k => $v) {
// if (isset($fieldArr[$k]) && $k !== '') {
// $row[$fieldArr[$k]] = $v;
// }
// }
// if ($row) {
// $insert[] = $row;
// }
// }
// } catch (Exception $exception) {
// $this->error($exception->getMessage());
// }
// if (!$insert) {
// $this->error(__('No rows were updated'));
// }
//
// try {
// //是否包含admin_id字段
// $has_admin_id = false;
// foreach ($fieldArr as $name => $key) {
// if ($key == 'admin_id') {
// $has_admin_id = true;
// break;
// }
// }
// if ($has_admin_id) {
// $auth = Auth::instance();
// foreach ($insert as &$val) {
// if (!isset($val['admin_id']) || empty($val['admin_id'])) {
// $val['admin_id'] = $auth->isLogin() ? $auth->id : 0;
// }
// }
// }
// $this->model->saveAll($insert);
// } catch (PDOException $exception) {
// $msg = $exception->getMessage();
// if (preg_match("/.+Integrity constraint violation: 1062 Duplicate entry '(.+)' for key '(.+)'/is", $msg, $matches)) {
// $msg = "导入失败,包含【{$matches[1]}】的记录已存在";
// };
// $this->error($msg);
// } catch (Exception $e) {
// $this->error($e->getMessage());
// }
//
// $this->success();
// }
/**
* 默认生成的控制器所继承的父类中有index/add/edit/del/multi五个基础方法、destroy/restore/recyclebin三个回收站方法
* 因此在当前控制器中可不用编写增删改查的代码,除非需要自己控制这部分逻辑
... ...
... ... @@ -8,7 +8,8 @@
<div id="toolbar" class="toolbar">
<a href="javascript:;" class="btn btn-primary btn-refresh" title="{:__('Refresh')}" ><i class="fa fa-refresh"></i> </a>
<a href="javascript:;" class="btn btn-success btn-add {:$auth->check('study/add')?'':'hide'}" title="{:__('Add')}" ><i class="fa fa-plus"></i> {:__('Add')}</a>
<a href="javascript:;" class="btn btn-success btn-import {:$auth->check('study/import')?'':'hide'}" title="导入" id="btn-import-file" data-url="ajax/upload" data-mimetype="csv,xls,xlsx" data-multiple="false"><i class="fa fa-upload"></i> 导入</a>
<!-- <a href="javascript:;" class="btn btn-success btn-import {:$auth->check('study/import')?'':'hide'}" title="导入" id="btn-import-file" data-url="ajax/upload" data-mimetype="csv,xls,xlsx" data-multiple="false"><i class="fa fa-upload"></i> 导入</a>-->
<a href="javascript:;" class="btn btn-danger btn-import {:$auth->check('你的权限规则')?'':'hide'}" title="导入" id="btn-import-file" data-url="ajax/upload" data-mimetype="csv,xls,xlsx" data-multiple="false"><i class="fa fa-upload"></i> 导入</a>
<a href="javascript:;" class="btn btn-danger btn-export {:$auth->check('study/export')?'':'hide'}" title="{:__('Export')}" id="btn-export-file"><i class="fa fa-download"></i> {:__('Export')}</a>
<a href="#" title="下载分享码" class="btn btn-success btn-disabled download {:$auth->check('study/multi')?'':'hide'} ">下载条形码</a>
... ...