1
This commit is contained in:
85
vendor/topthink/think-migration/src/command/seed/Create.php
vendored
Normal file
85
vendor/topthink/think-migration/src/command/seed/Create.php
vendored
Normal file
@@ -0,0 +1,85 @@
|
||||
<?php
|
||||
// +----------------------------------------------------------------------
|
||||
// | TopThink [ WE CAN DO IT JUST THINK IT ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2016 http://www.topthink.com All rights reserved.
|
||||
// +----------------------------------------------------------------------
|
||||
// | Author: zhangyajun <448901948@qq.com>
|
||||
// +----------------------------------------------------------------------
|
||||
|
||||
namespace think\migration\command\seed;
|
||||
|
||||
use Phinx\Util\Util;
|
||||
use think\console\Input;
|
||||
use think\console\Output;
|
||||
use think\console\input\Argument as InputArgument;
|
||||
use think\migration\command\Seed;
|
||||
|
||||
class Create extends Seed
|
||||
{
|
||||
/**
|
||||
* {@inheritdoc}
|
||||
*/
|
||||
protected function configure()
|
||||
{
|
||||
$this->setName('seed:create')
|
||||
->setDescription('Create a new database seeder')
|
||||
->addArgument('name', InputArgument::REQUIRED, 'What is the name of the seeder?')
|
||||
->setHelp(sprintf('%sCreates a new database seeder%s', PHP_EOL, PHP_EOL));
|
||||
}
|
||||
|
||||
/**
|
||||
* Create the new seeder.
|
||||
*
|
||||
* @param Input $input
|
||||
* @param Output $output
|
||||
* @throws \RuntimeException
|
||||
* @throws \InvalidArgumentException
|
||||
* @return void
|
||||
*/
|
||||
protected function execute(Input $input, Output $output)
|
||||
{
|
||||
$path = $this->getPath();
|
||||
|
||||
if (!file_exists($path)) {
|
||||
if ($this->output->confirm($this->input, 'Create seeds directory? [y]/n')) {
|
||||
mkdir($path, 0755, true);
|
||||
}
|
||||
}
|
||||
|
||||
$this->verifyMigrationDirectory($path);
|
||||
|
||||
$path = realpath($path);
|
||||
|
||||
$className = $input->getArgument('name');
|
||||
|
||||
if (!Util::isValidPhinxClassName($className)) {
|
||||
throw new \InvalidArgumentException(sprintf('The seed class name "%s" is invalid. Please use CamelCase format', $className));
|
||||
}
|
||||
|
||||
// Compute the file path
|
||||
$filePath = $path . DS . $className . '.php';
|
||||
|
||||
if (is_file($filePath)) {
|
||||
throw new \InvalidArgumentException(sprintf('The file "%s" already exists', basename($filePath)));
|
||||
}
|
||||
|
||||
// inject the class names appropriate to this seeder
|
||||
$contents = file_get_contents($this->getTemplate());
|
||||
$classes = [
|
||||
'$className' => $className
|
||||
];
|
||||
$contents = strtr($contents, $classes);
|
||||
|
||||
if (false === file_put_contents($filePath, $contents)) {
|
||||
throw new \RuntimeException(sprintf('The file "%s" could not be written to', $path));
|
||||
}
|
||||
|
||||
$output->writeln('<info>created</info> .' . str_replace(getcwd(), '', $filePath));
|
||||
}
|
||||
|
||||
protected function getTemplate()
|
||||
{
|
||||
return __DIR__ . '/../stubs/seed.stub';
|
||||
}
|
||||
}
|
||||
107
vendor/topthink/think-migration/src/command/seed/Run.php
vendored
Normal file
107
vendor/topthink/think-migration/src/command/seed/Run.php
vendored
Normal file
@@ -0,0 +1,107 @@
|
||||
<?php
|
||||
// +----------------------------------------------------------------------
|
||||
// | TopThink [ WE CAN DO IT JUST THINK IT ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2016 http://www.topthink.com All rights reserved.
|
||||
// +----------------------------------------------------------------------
|
||||
// | Author: zhangyajun <448901948@qq.com>
|
||||
// +----------------------------------------------------------------------
|
||||
|
||||
namespace think\migration\command\seed;
|
||||
|
||||
use Phinx\Seed\SeedInterface;
|
||||
use think\console\Input;
|
||||
use think\console\input\Option as InputOption;
|
||||
use think\console\Output;
|
||||
use think\migration\command\Seed;
|
||||
|
||||
class Run extends Seed
|
||||
{
|
||||
/**
|
||||
* {@inheritdoc}
|
||||
*/
|
||||
protected function configure()
|
||||
{
|
||||
$this->setName('seed:run')
|
||||
->setDescription('Run database seeders')
|
||||
->addOption('--seed', '-s', InputOption::VALUE_REQUIRED, 'What is the name of the seeder?')
|
||||
->setHelp(<<<EOT
|
||||
The <info>seed:run</info> command runs all available or individual seeders
|
||||
|
||||
<info>php console seed:run</info>
|
||||
<info>php console seed:run -s UserSeeder</info>
|
||||
<info>php console seed:run -v</info>
|
||||
|
||||
EOT
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* Run database seeders.
|
||||
*
|
||||
* @param Input $input
|
||||
* @param Output $output
|
||||
* @return void
|
||||
*/
|
||||
protected function execute(Input $input, Output $output)
|
||||
{
|
||||
$seed = $input->getOption('seed');
|
||||
|
||||
// run the seed(ers)
|
||||
$start = microtime(true);
|
||||
$this->seed($seed);
|
||||
$end = microtime(true);
|
||||
|
||||
$output->writeln('');
|
||||
$output->writeln('<comment>All Done. Took ' . sprintf('%.4fs', $end - $start) . '</comment>');
|
||||
}
|
||||
|
||||
public function seed($seed = null)
|
||||
{
|
||||
$seeds = $this->getSeeds();
|
||||
|
||||
if (null === $seed) {
|
||||
// run all seeders
|
||||
foreach ($seeds as $seeder) {
|
||||
if (array_key_exists($seeder->getName(), $seeds)) {
|
||||
$this->executeSeed($seeder);
|
||||
}
|
||||
}
|
||||
} else {
|
||||
// run only one seeder
|
||||
if (array_key_exists($seed, $seeds)) {
|
||||
$this->executeSeed($seeds[$seed]);
|
||||
} else {
|
||||
throw new \InvalidArgumentException(sprintf('The seed class "%s" does not exist', $seed));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
protected function executeSeed(SeedInterface $seed)
|
||||
{
|
||||
$this->output->writeln('');
|
||||
$this->output->writeln(' ==' . ' <info>' . $seed->getName() . ':</info>' . ' <comment>seeding</comment>');
|
||||
|
||||
// Execute the seeder and log the time elapsed.
|
||||
$start = microtime(true);
|
||||
$seed->setAdapter($this->getAdapter());
|
||||
|
||||
// begin the transaction if the adapter supports it
|
||||
if ($this->getAdapter()->hasTransactions()) {
|
||||
$this->getAdapter()->beginTransaction();
|
||||
}
|
||||
|
||||
// Run the seeder
|
||||
if (method_exists($seed, SeedInterface::RUN)) {
|
||||
$seed->run();
|
||||
}
|
||||
|
||||
// commit the transaction if the adapter supports it
|
||||
if ($this->getAdapter()->hasTransactions()) {
|
||||
$this->getAdapter()->commitTransaction();
|
||||
}
|
||||
$end = microtime(true);
|
||||
|
||||
$this->output->writeln(' ==' . ' <info>' . $seed->getName() . ':</info>' . ' <comment>seeded' . ' ' . sprintf('%.4fs', $end - $start) . '</comment>');
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user