Fix

drago-ex · Jan 31, 2024 · f9d7e35 · f9d7e35
1 parent 2403af7
commit f9d7e35
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/Drago/Generator/Generator/DataClassGenerator.php b/src/Drago/Generator/Generator/DataClassGenerator.php
@@ -11,7 +11,6 @@
 
 use Dibi\Exception;
 use Drago\Generator\Base;
-use Drago\Utils\CaseConverter;
 use Nette\PhpGenerator\ClassType;
 use Nette\PhpGenerator\PhpFile;
 use Nette\Utils\FileSystem;
@@ -86,8 +85,9 @@ public function createPhpFile(string $table): void
 
 			// Add constants.
 			if ($options->constantDataClass) {
-				$constant = $this->options->constantDataPrefix
-					? $this->inflector->classify($this->options->constantDataPrefix . $column)
+				$cp = $this->options->constantDataPrefix;
+				$constant = $cp
+					? $cp . $this->inflector->classify($column)
 					: $this->inflector->classify($column);
 
 				$class->addConstant($constant, $column)

diff --git a/src/Drago/Generator/Generator/EntityGenerator.php b/src/Drago/Generator/Generator/EntityGenerator.php
@@ -11,7 +11,6 @@
 
 use Dibi\Exception;
 use Drago\Generator\Base;
-use Drago\Utils\CaseConverter;
 use Nette\PhpGenerator\ClassType;
 use Nette\PhpGenerator\PhpFile;
 use Nette\Utils\FileSystem;
@@ -98,8 +97,9 @@ public function createPhpFile(string $table): void
 
 			// Add other constants.
 			if ($options->constant) {
-				$constant = $this->options->constantPrefix
-					? $this->inflector->classify($this->options->constantPrefix . $column)
+				$cp = $this->options->constantPrefix;
+				$constant = $cp
+					? $cp . $this->inflector->classify($column)
 					: $this->inflector->classify($column);
 
 				if (!$attr->isAutoIncrement()) {