54 lines
1.2 KiB
PHP
54 lines
1.2 KiB
PHP
<?php
|
|
|
|
/**
|
|
* This file is part of the Propel package.
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*
|
|
* @license MIT License
|
|
*/
|
|
|
|
/**
|
|
* Adds a primary key to models defined without one
|
|
*
|
|
* @author François Zaninotto
|
|
* @version $Revision: 1745 $
|
|
* @package propel.generator.behavior
|
|
*/
|
|
class AutoAddPkBehavior extends Behavior
|
|
{
|
|
protected $isEarly = true;
|
|
|
|
// default parameters value
|
|
protected $parameters = array(
|
|
'name' => 'id',
|
|
'autoIncrement' => 'true',
|
|
'type' => 'INTEGER'
|
|
);
|
|
|
|
/**
|
|
* Copy the behavior to the database tables
|
|
* Only for tables that have no Pk
|
|
*/
|
|
public function modifyDatabase()
|
|
{
|
|
foreach ($this->getDatabase()->getTables() as $table) {
|
|
if(!$table->hasPrimaryKey()) {
|
|
$b = clone $this;
|
|
$table->addBehavior($b);
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Add the primary key to the current table
|
|
*/
|
|
public function modifyTable()
|
|
{
|
|
$table = $this->getTable();
|
|
if (!$table->hasPrimaryKey() && !$table->hasBehavior('concrete_inheritance')) {
|
|
$columnAttributes = array_merge(array('primaryKey' => 'true'), $this->getParameters());
|
|
$this->getTable()->addColumn($columnAttributes);
|
|
}
|
|
}
|
|
} |