← All changes | classes/Visualizer/Source/Csv.php +22 -75 3.10.12 → 1.4.2.2 View file →

		@@ -18,8 +18,9 @@
18	18	// | MA 02110-1301 USA |
19	19	// +----------------------------------------------------------------------+
20	20	// | Author: Eugene Manuilov <eugene@manuilov.org> |
21	21	// +----------------------------------------------------------------------+
	22	+
22	23	/**
23	24	* Source manager for local CSV files.
24	25	*
25	26	* @category Visualizer
		@@ -46,10 +47,10 @@
46	47	*
47	48	* @access public
48	49	* @param string $filename The path to the file.
49	50	*/
50		- public function __construct( $filename = '' ) {
51		- $this->_filename = trim( (string) $filename );
	51	+ public function __construct( $filename = null ) {
	52	+ $this->_filename = trim( $filename );
52	53	}
53	54
54	55	/**
55	56	* Fetches series information.
		@@ -61,45 +62,32 @@
61	62	*/
62	63	private function _fetchSeries( &$handle ) {
63	64	// read column titles
64	65	$labels = fgetcsv( $handle, 0, VISUALIZER_CSV_DELIMITER, VISUALIZER_CSV_ENCLOSURE );
65		- $types = null;
	66	+ // read series types
	67	+ $types = fgetcsv( $handle, 0, VISUALIZER_CSV_DELIMITER, VISUALIZER_CSV_ENCLOSURE );
66	68
67		- if ( false !== strpos( $this->_filename, 'tqx=out:csv' ) ) {
68		- $attributes = $this->_fetchSeriesForGoogleQueryLanguage( $labels );
69		- if ( ! $attributes['abort'] ) {
70		- $labels = $attributes['labels'];
71		- $types = $attributes['types'];
72		- }
73		- }
74		-
75		- if ( is_null( $types ) ) {
76		- // read series types
77		- $types = fgetcsv( $handle, 0, VISUALIZER_CSV_DELIMITER, VISUALIZER_CSV_ENCLOSURE );
78		- }
79		-
80		- $labels = array_filter( $labels );
81		- $types = array_filter( $types );
82		-
83		- if ( ! $labels || ! $types ) {
84		- $this->_error = esc_html__( 'File should have a heading row (1st row) and a data type row (2nd row). Please try again.', 'visualizer' );
	69	+ if ( !$labels || !$types ) {
85	70	return false;
86	71	}
87	72
	73	+ // if no types were setup, re read labels and empty types array
88	74	$types = array_map( 'trim', $types );
89		- if ( ! self::_validateTypes( $types ) ) {
90		- $this->_error = esc_html__( 'Invalid data types detected in the data type row (2nd row). Please try again.', 'visualizer' );
91		- return false;
	75	+ if ( !self::_validateTypes( $types ) ) {
	76	+ // re open the file
	77	+ fclose( $handle );
	78	+ $handle = $this->_get_file_handle();
	79	+
	80	+ // re read the labels and empty types array
	81	+ $labels = fgetcsv( $handle, 0, VISUALIZER_CSV_DELIMITER, VISUALIZER_CSV_ENCLOSURE );
	82	+ $types = array();
92	83	}
93	84
94	85	for ( $i = 0, $len = count( $labels ); $i < $len; $i++ ) {
95		- $default_type = $i === 0 ? 'string' : 'number';
96		-
97		- $labels[ $i ] = $this->toUTF8( $labels[ $i ] );
98		-
	86	+ $default_type = $i == 0 ? 'string' : 'number';
99	87	$this->_series[] = array(
100		- 'label' => $labels[ $i ],
101		- 'type' => isset( $types[ $i ] ) ? $types[ $i ] : $default_type,
	88	+ 'label' => $labels[$i],
	89	+ 'type' => isset( $types[$i] ) ? $types[$i] : $default_type,
102	90	);
103	91	}
104	92
105	93	return true;
		@@ -114,8 +102,10 @@
114	102	* @param string $filename Optional file name to get handle. If omitted, $_filename is used.
115	103	* @return resource File handle resource on success, otherwise FALSE.
116	104	*/
117	105	protected function _get_file_handle( $filename = false ) {
	106	+ // set line endings auto detect mode
	107	+ @ini_set( 'auto_detect_line_endings', true );
118	108	// open file and return handle
119	109	return fopen( $filename ? $filename : $this->_filename, 'rb' );
120	110	}
121	111
		@@ -129,9 +119,8 @@
129	119	*/
130	120	public function fetch() {
131	121	// if filename is empty return false
132	122	if ( empty( $this->_filename ) ) {
133		- $this->_error = esc_html__( 'No file provided. Please try again.', 'visualizer' );
134	123	return false;
135	124	}
136	125
137	126	// read file and fill arrays
		@@ -137,14 +126,13 @@
137	126	// read file and fill arrays
138	127	$handle = $this->_get_file_handle();
139	128	if ( $handle ) {
140	129	// fetch series
141		- if ( ! $this->_fetchSeries( $handle ) ) {
	130	+ if ( !$this->_fetchSeries( $handle ) ) {
142	131	return false;
143	132	}
144	133
145	134	// fetch data
146		- // phpcs:ignore WordPress.CodeAnalysis.AssignmentInCondition.FoundInWhileCondition
147	135	while ( ( $data = fgetcsv( $handle, 0, VISUALIZER_CSV_DELIMITER, VISUALIZER_CSV_ENCLOSURE ) ) !== false ) {
148	136	$this->_data[] = $this->_normalizeData( $data );
149	137	}
150	138
		@@ -166,46 +154,5 @@
166	154	public function getSourceName() {
167	155	return __CLASS__;
168	156	}
169	157
170		- /**
171		- * Adds support for QueryLanguage https://developers.google.com/chart/interactive/docs/querylanguage
172		- * where the user can provide something like /gviz/tq?tq=select%20A%2C%20B%20&tqx=out:csv after the URL of the raw spreadsheet
173		- * to get the subset of data specified by the query
174		- * this will conflate the heading and the type into one value viz. for heading XXX and type string, the value will become "XXX string"
175		- * so we need to split them apart logically
176		- * also the $types variable now contains the first row because the header and the type got conflated
177		- */
178		- private function _fetchSeriesForGoogleQueryLanguage( $labels, $types = array() ) {
179		- $new_labels = array();
180		- $new_types = array();
181		- $abort = false;
182		- foreach ( $labels as $label ) {
183		- // get the index of the last space
184		- $index = strrpos( $label, ' ' );
185		- if ( $index === false ) {
186		- // no space here? something has gone wrong; abort the entire process.
187		- $abort = true;
188		- break;
189		- }
190		- $type = trim( substr( $label, $index + 1 ) );
191		- if ( ! self::_validateTypes( array( $type ) ) ) {
192		- // some other data type? abort the entire process.
193		- $abort = true;
194		- break;
195		- }
196		- $label = substr( $label, 0, $index );
197		- $new_labels[] = $label;
198		- $new_types[] = $type;
199		- }
200		- if ( ! $abort ) {
201		- $labels = $new_labels;
202		- $types = $new_types;
203		- }
204		-
205		- return array(
206		- 'abort' => $abort,
207		- 'labels' => $labels,
208		- 'types' => $types,
209		- );
210		- }
211		-}
	158	+}