frictionlessdata\tableschema\DataSources\CsvDataSource

Complexity

Total Complexity

18

Size/Duplication

Total Lines	115
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	2

Importance

Changes

0

7 Methods

Rating	Name	Duplication	Size	Complexity
C	open()	0	29	7
A	getSkippedRows()	0	4	1
A	getNextLine()	0	11	2
A	isEof()	0	8	2
A	close()	0	8	2
A	save()	0	8	2
A	getRow()	0	9	2

<?php

namespace frictionlessdata\tableschema\DataSources;

use frictionlessdata\tableschema\Exceptions\DataSourceException;

/**
 * handles reading data from a csv source
 * responsible for finding the header row based on options
 * support skipping rows from the csv.
 */
class CsvDataSource extends BaseDataSource
{
    /**
     * @throws DataSourceException
     */
    public function open()
    {
        $this->curRowNum = 0;
        try {
            $this->resource = fopen($this->dataSource, 'r');
        } catch (\Exception $e) {
            throw new DataSourceException($e->getMessage());
        }
        $this->headerRow = $this->getOption('headerRow');
        if ($this->headerRow) {
            $headerRowNum = 0;
            $defaultSkipRows = 0;
        } else {
            $defaultSkipRows = $headerRowNum = $this->getOption('headerRowNum', 1);
        }
        $skipRows = $this->getOption('skipRows', $defaultSkipRows);
        if ($skipRows > 0) {
            foreach (range(1, $skipRows) as $i) {
                $row = $this->getRow();
                $this->skippedRows[] = $row;
                if ($i == $headerRowNum) {
                    $this->headerRow = $row;
                }
            }
        }
        if (!$this->headerRow) {
            throw new DataSourceException('Failed to get header row');
        }
    }

    /**
     * @return array
     */
    public function getSkippedRows()
    {
        return $this->skippedRows;
    }

    /**
     * @return array
     *
     * @throws DataSourceException
     */
    public function getNextLine()
    {
        $row = $this->getRow();
        $colNum = 0;
        $obj = [];
        foreach ($this->headerRow as $fieldName) {
            $obj[$fieldName] = $row[$colNum++];
        }

        return $obj;
    }

    /**
     * @return bool
     *
     * @throws DataSourceException
     */
    public function isEof()
    {
        try {
            return feof($this->resource);
        } catch (\Exception $e) {
            throw new DataSourceException($e->getMessage(), $this->curRowNum);
        }
    }

    /**
     * @throws DataSourceException
     */
    public function close()
    {
        try {
            fclose($this->resource);
        } catch (\Exception $e) {
            throw new DataSourceException($e->getMessage(), $this->curRowNum);
        }
    }

    public function save($outputDataSource) {
        $file = fopen($outputDataSource, 'w');
        fputcsv($file, $this->headerRow);
        while (!$this->isEof()) {
            fputcsv($file, array_values($this->getNextLine()));
        }
        fclose($file);
    }

    protected $resource;
    protected $headerRow;
    protected $skippedRows;
    protected $curRowNum;

    /**
     * @return array
     *
     * @throws DataSourceException
     */
    protected function getRow()
    {
        ++$this->curRowNum;
        try {
            return fgetcsv($this->resource);
        } catch (\Exception $e) {
            throw new DataSourceException($e->getMessage(), $this->curRowNum);
        }
    }
}