actions/export_csv.php

Go to the documentation of this file.

<?php
if ( !function_exists('prepare_csv') ){
        function prepare_csv($str){
                return str_replace('"','""',$str);
        }
}

if ( !function_exists('fputcsv') ){
         function fputcsv($filePointer,$dataArray,$delimiter=',',$enclosure='"')
        {
                // Write a line to a file
                // $filePointer = the file resource to write to
                // $dataArray = the data to write out
                // $delimeter = the field separator
                
                
                // Build the string
                $dataArray = array_map('prepare_csv', $dataArray);
                $string = $enclosure.implode($enclosure.$delimiter.$enclosure, $dataArray).$enclosure;
                
                
                // Append new line
                $string .= "\n";
                
                // Write the string to the file
                fwrite($filePointer,$string);
        }
}



class dataface_actions_export_csv {
        
        function handle(&$params){
        
                $app =& Dataface_Application::getInstance();
                $query = $app->getQuery();
                $query['-limit'] = 9999999;
                $table =& Dataface_Table::loadTable($query['-table']);
                if ( isset($query['-relationship']) and @$query['--related'] ){
                        $query['-related:start'] = 0;
                        $query['-related:limit'] = 9999999;
                        $record =& $app->getRecord();
                        $relationship =& $table->getRelationship($query['-relationship']);
                        
                        $records =& df_get_related_records($query); //$record->getRelatedRecordObjects($query['-relationship']);
                        
                        $data = array(/*$relationship->_schema['short_columns']*/);
                        $headings = array();
                        foreach ( $relationship->_schema['short_columns'] as $colhead ){
                                $f =& $relationship->getField($colhead);
                                if ( @$f['visibility']['csv']  == 'hidden' ){
                                        unset($f);
                                        continue;
                                }
                                $headings[] = $colhead;
                                unset($f);
                        }
                        $data[] = $headings;
                        foreach ($records as $record){
                                if ( !$record->checkPermission('view') ) continue;
                                $data[] = $this->related_rec2data($record);
                        }
                } else {
                        $records =& df_get_records_array($query['-table'], $query,null,null,false);
                        $data = array();
                        $headings = array();
                        foreach (array_merge(array_keys($table->fields()), array_keys($table->graftedFields())) as $colhead){
                                $f =& $table->getField($colhead);
                                if ( @$f['visibility']['csv'] == 'hidden' ){
                                        unset($f);
                                        continue;
                                }
                                $headings[] = $colhead;
                                unset($f);
                        
                        }
                        $data[] = $headings;
                        foreach ($records as $record){
                                if ( !$record->checkPermission('view') ) continue;
                                $data[] = $this->rec2data($record);
                        }
                }
                
                $temp = tmpfile();
                foreach ($data as $row){
                        fputcsv($temp, $row,",",'"');
                }
                fseek($temp,0);
                header("Content-type: text/csv; charset={$app->_conf['oe']}");
                header('Content-disposition: attachment; filename="'.$query['-table'].'_results_'.date('Y_m_d_H_i_s').'.csv"');
                
                $fstats = fstat($temp);
                
                echo fread($temp, $fstats['size']);
                fclose($temp);
                exit;
                
                
                
        }
        
        function rec2data(&$record){
                $out = array();
                $columns = array_merge(array_keys($record->_table->fields()), array_keys($record->_table->graftedFields()));
                
                foreach ($columns as $key){
                        $f =& $record->_table->getField($key);
                        if ( @$f['visibility']['csv'] == 'hidden' ){
                                unset($f);
                                continue;
                        }
                        $out[] = $record->display($key);
                        unset($f);
                }
                return $out;
        }
        
        function related_rec2data(&$record){
                $out = array();
                $r =& $record->_relationship;
                foreach ($r->_schema['short_columns'] as $col){
                        $f =& $r->getField($col);
                        if ( @$f['visibility']['csv'] == 'hidden' ){
                                unset($f);
                                continue;
                        }
                        $out[] = $record->display($col);
                        unset($f);
                }
                return $out;
                
        }

}

?>