| Server IP : 49.231.201.246 / Your IP : 216.73.216.149 Web Server : Apache/2.4.18 (Ubuntu) System : Linux 246 4.4.0-210-generic #242-Ubuntu SMP Fri Apr 16 09:57:56 UTC 2021 x86_64 User : root ( 0) PHP Version : 7.0.33-0ubuntu0.16.04.16 Disable Function : exec,passthru,shell_exec,system,proc_open,popen,pcntl_exec MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : ON | Pkexec : ON Directory : /proc/11585/cwd/html/ppaobm/vendor/codeception/base/src/Codeception/Lib/Driver/ |
Upload File : |
<?php
namespace Codeception\Lib\Driver;
use Codeception\Configuration;
use Codeception\Exception\ModuleException;
class Sqlite extends Db
{
protected $hasSnapshot = false;
protected $filename = '';
protected $con = null;
public function __construct($dsn, $user, $password, $options = null)
{
$filename = substr($dsn, 7);
if ($filename === ':memory:') {
throw new ModuleException(__CLASS__, ':memory: database is not supported');
}
$this->filename = Configuration::projectDir() . $filename;
$this->dsn = 'sqlite:' . $this->filename;
parent::__construct($this->dsn, $user, $password, $options);
}
public function cleanup()
{
$this->dbh = null;
file_put_contents($this->filename, '');
$this->dbh = self::connect($this->dsn, $this->user, $this->password);
}
public function load($sql)
{
if ($this->hasSnapshot) {
$this->dbh = null;
file_put_contents($this->filename, file_get_contents($this->filename . '_snapshot'));
$this->dbh = new \PDO($this->dsn, $this->user, $this->password);
} else {
if (file_exists($this->filename . '_snapshot')) {
unlink($this->filename . '_snapshot');
}
parent::load($sql);
copy($this->filename, $this->filename . '_snapshot');
$this->hasSnapshot = true;
}
}
/**
* @param string $tableName
*
* @return array[string]
*/
public function getPrimaryKey($tableName)
{
if (!isset($this->primaryKeys[$tableName])) {
if ($this->hasRowId($tableName)) {
return $this->primaryKeys[$tableName] = ['_ROWID_'];
}
$primaryKey = [];
$query = 'PRAGMA table_info(' . $this->getQuotedName($tableName) . ')';
$stmt = $this->executeQuery($query, []);
$columns = $stmt->fetchAll(\PDO::FETCH_ASSOC);
foreach ($columns as $column) {
if ($column['pk'] !== '0') {
$primaryKey []= $column['name'];
}
}
$this->primaryKeys[$tableName] = $primaryKey;
}
return $this->primaryKeys[$tableName];
}
/**
* @param $tableName
* @return bool
*/
private function hasRowId($tableName)
{
$params = ['type' => 'table', 'name' => $tableName];
$select = $this->select('sql', 'sqlite_master', $params);
$result = $this->executeQuery($select, $params);
$sql = $result->fetchColumn(0);
return strpos($sql, ') WITHOUT ROWID') === false;
}
}