L-electron-Rare/moodle
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

MDL-68500 dataformat: allow instances to export to local file.

Browse Source
This commit is contained in:
Paul Holden
2020-05-04 09:32:44 +01:00
parent cd391f9922
commit 1de3b81983
6 changed files with 262 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lib/dataformatlib.php
+82 -10
View File
@@ -33,8 +33,8 @@
* @param string $dataformat A dataformat name
* @param array $columns An ordered map of column keys and labels
* @param Iterator $iterator An iterator over the records, usually a RecordSet
* @param function $callback An option function applied to each record before writing
* @param mixed $extra An optional value which is passed into the callback function
* @param callable $callback An option function applied to each record before writing
* @throws coding_exception
*/
function download_as_dataformat($filename, $dataformat, $columns, $iterator, $callback = null) {
@@ -46,10 +46,9 @@ function download_as_dataformat($filename, $dataformat, $columns, $iterator, $ca
if (!class_exists($classname)) {
throw new coding_exception("Unable to locate dataformat/$dataformat/classes/writer.php");
}
$format = new $classname;
// The data format export could take a while to generate...
set_time_limit(0);
core_php_time_limit::raise();
// Close the session so that the users other tabs in the same session are not blocked.
\core\session\manager::write_close();
@@ -57,17 +56,22 @@ function download_as_dataformat($filename, $dataformat, $columns, $iterator, $ca
// If this file was requested from a form, then mark download as complete (before sending headers).
\core_form\util::form_download_complete();
/** @var \core\dataformat\base $format */
$format = new $classname;
$format->set_filename($filename);
$format->send_http_headers();
// This exists to support all dataformats - see MDL-56046.
if (method_exists($format, 'write_header')) {
error_log('The function write_header() does not support multiple sheets. In order to support multiple sheets you ' .
'must implement start_output() and start_sheet() and remove write_header() in your dataformat.');
if (!method_exists($format, 'start_output') && method_exists($format, 'write_header')) {
debugging('The function write_header() does not support multiple sheets. In order to support multiple sheets you ' .
'must implement start_output() and start_sheet() and remove write_header() in your dataformat.', DEBUG_DEVELOPER);
$format->write_header($columns);
} else {
$format->start_output();
$format->start_sheet($columns);
}
$c = 0;
foreach ($iterator as $row) {
if ($callback) {
@@ -78,10 +82,11 @@ function download_as_dataformat($filename, $dataformat, $columns, $iterator, $ca
}
$format->write_record($row, $c++);
}
// This exists to support all dataformats - see MDL-56046.
if (method_exists($format, 'write_footer')) {
error_log('The function write_footer() does not support multiple sheets. In order to support multiple sheets you ' .
'must implement close_sheet() and close_output() and remove write_footer() in your dataformat.');
if (!method_exists($format, 'close_output') && method_exists($format, 'write_footer')) {
debugging('The function write_footer() does not support multiple sheets. In order to support multiple sheets you ' .
'must implement close_sheet() and close_output() and remove write_footer() in your dataformat.', DEBUG_DEVELOPER);
$format->write_footer($columns);
} else {
$format->close_sheet($columns);
@@ -89,3 +94,70 @@ function download_as_dataformat($filename, $dataformat, $columns, $iterator, $ca
}
}
/**
* Writes a formatted data file to specified path
*
* @package core
* @subpackage dataformat
*
* @param string $filename The base filename without an extension
* @param string $dataformat A dataformat name
* @param array $columns An ordered map of column keys and labels
* @param Iterator $iterator An iterator over the records, usually a RecordSet
* @param callable $callback An option function applied to each record before writing
* @return string Complete path to written file
* @throws coding_exception
*/
function write_file_as_dataformat(string $filename, string $dataformat, array $columns, $iterator,
callable $callback = null): string {
$classname = 'dataformat_' . $dataformat . '\writer';
if (!class_exists($classname)) {
throw new coding_exception("Unable to locate dataformat/$dataformat/classes/writer.php");
}
// The data format export could take a while to generate.
core_php_time_limit::raise();
// Close the session so that the users other tabs in the same session are not blocked.
\core\session\manager::write_close();
/** @var \core\dataformat\base $format */
$format = new $classname;
$filepath = make_request_directory() . '/' . $filename . $format->get_extension();
$format->set_filepath($filepath);
// This exists to support all dataformats - see MDL-56046.
if (!method_exists($format, 'start_output_to_file') && method_exists($format, 'write_header')) {
debugging('The function write_header() does not support multiple sheets. In order to support multiple sheets you must ' .
'implement start_output_to_file() and start_sheet() and remove write_header() in your dataformat.', DEBUG_DEVELOPER);
$format->write_header($columns);
} else {
$format->start_output_to_file();
$format->start_sheet($columns);
}
$c = 0;
foreach ($iterator as $row) {
if ($callback) {
$row = $callback($row);
}
if ($row === null) {
continue;
}
$format->write_record($row, $c++);
}
// This exists to support all dataformats - see MDL-56046.
if (!method_exists($format, 'close_output_to_file') && method_exists($format, 'write_footer')) {
debugging('The function write_footer() does not support multiple sheets. In order to support multiple sheets you must ' .
'implement close_sheet() and close_output_to_file() and remove write_footer() in your dataformat.', DEBUG_DEVELOPER);
$format->write_footer($columns);
} else {
$format->close_sheet($columns);
$format->close_output_to_file();
}
return $filepath;
}