SystemExtension: Set driver for DataTransfer

Venne · Oct 10, 2014 · 01f4149 · 01f4149
1 parent edc2d02
commit 01f4149
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/System/DI/SystemExtension.php b/src/System/DI/SystemExtension.php
@@ -15,6 +15,7 @@
 use Nette\DI\ContainerBuilder;
 use Nette\DI\Statement;
 use Nette\PhpGenerator\PhpLiteral;
+use Venne\DataTransfer\DI\DataTransferExtension;
 use Venne\Widgets\DI\WidgetsExtension;
 
 /**
@@ -188,6 +189,12 @@ public function loadConfiguration()
 			}
 		}
 
+		foreach ($this->compiler->extensions as $extension) {
+			if ($extension instanceof DataTransferExtension) {
+				$extension->setDriverClass('Venne\Bridges\Kdyby\Doctrine\DataTransfer\EntityDriver');
+			}
+		}
+
 		$container->addDefinition($this->prefix('trayComponent'))
 			->setImplement('Venne\System\AdminModule\Components\ITrayControlFactory')
 			->setInject(true);