56 lines
1.6 KiB
PHP
56 lines
1.6 KiB
PHP
|
<?php
|
||
|
|
|
||
|
|
/*
|
||
|
|
* This file is part of the Doctrine Bundle
|
||
|
|
*
|
||
|
|
* The code was originally distributed inside the Symfony framework.
|
||
|
|
*
|
||
|
|
* (c) Fabien Potencier <fabien@symfony.com>
|
||
|
|
* (c) Doctrine Project, Benjamin Eberlei <kontakt@beberlei.de>
|
||
|
|
*
|
||
|
|
* For the full copyright and license information, please view the LICENSE
|
||
|
|
* file that was distributed with this source code.
|
||
|
|
*/
|
||
|
|
|
||
|
|
namespace Doctrine\Bundle\DoctrineBundle\Command;
|
||
|
|
|
||
|
|
use Symfony\Bundle\FrameworkBundle\Command\ContainerAwareCommand;
|
||
|
|
use Doctrine\ORM\Tools\EntityGenerator;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Base class for Doctrine console commands to extend from.
|
||
|
|
*
|
||
|
|
* @author Fabien Potencier <fabien@symfony.com>
|
||
|
|
*/
|
||
|
|
abstract class DoctrineCommand extends ContainerAwareCommand
|
||
|
|
{
|
||
|
|
protected function getEntityGenerator()
|
||
|
|
{
|
||
|
|
$entityGenerator = new EntityGenerator();
|
||
|
|
$entityGenerator->setGenerateAnnotations(false);
|
||
|
|
$entityGenerator->setGenerateStubMethods(true);
|
||
|
|
$entityGenerator->setRegenerateEntityIfExists(false);
|
||
|
|
$entityGenerator->setUpdateEntityIfExists(true);
|
||
|
|
$entityGenerator->setNumSpaces(4);
|
||
|
|
$entityGenerator->setAnnotationPrefix('ORM\\');
|
||
|
|
|
||
|
|
return $entityGenerator;
|
||
|
|
}
|
||
|
|
|
||
|
|
protected function getEntityManager($name)
|
||
|
|
{
|
||
|
|
return $this->getContainer()->get('doctrine')->getManager($name);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Get a doctrine dbal connection by symfony name.
|
||
|
|
*
|
||
|
|
* @param string $name
|
||
|
|
* @return Doctrine\DBAL\Connection
|
||
|
|
*/
|
||
|
|
protected function getDoctrineConnection($name)
|
||
|
|
{
|
||
|
|
return $this->getContainer()->get('doctrine')->getConnection($name);
|
||
|
|
}
|
||
|
|
}
|