Merge pull request #480 from radekdostal/fix-support-dibi-2.x

Fixed support of dibi 2.x.
contributte · Mar 1, 2017 · b9026cf · b9026cf
2 parents 68d3bb8 + d695127
commit b9026cf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/DataModel.php b/src/DataModel.php
@@ -10,6 +10,7 @@
 
 use Nette;
 use Dibi;
+use DibiFluent;
 use Nette\Database\Table\Selection;
 use Doctrine\Common\Collections\Collection;
 use Doctrine\ORM\QueryBuilder;
@@ -61,7 +62,7 @@ public function __construct($source, $primary_key)
 		} else if (is_array($source)) {
 			$source = new DataSource\ArrayDataSource($source);
 
-		} else if ($source instanceof Dibi\Fluent) {
+		} else if ($source instanceof Dibi\Fluent || $source instanceof DibiFluent) {
 			$driver = $source->getConnection()->getDriver();
 
 			if ($driver instanceof Dibi\Drivers\OdbcDriver) {