entity generator - ignore trait properties and methods

doctrine · Mar 26, 2013 · 521276f · 521276f
1 parent a4db7c8
commit 521276f
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/lib/Doctrine/ORM/Tools/EntityGenerator.php b/lib/Doctrine/ORM/Tools/EntityGenerator.php
@@ -709,6 +709,15 @@ protected function hasProperty($property, ClassMetadataInfo $metadata)
             }
         }
 
+        // check traits for existing property
+        $reflClass = new \ReflectionClass($metadata->name);
+
+        foreach ($reflClass->getTraits() as $trait) {
+            if ($trait->hasProperty($property)) {
+                return true;
+            }
+        }
+
         return (
             isset($this->staticReflection[$metadata->name]) &&
             in_array($property, $this->staticReflection[$metadata->name]['properties'])
@@ -732,6 +741,15 @@ protected function hasMethod($method, ClassMetadataInfo $metadata)
             }
         }
 
+        // check traits for existing method
+        $reflClass = new \ReflectionClass($metadata->name);
+
+        foreach ($reflClass->getTraits() as $trait) {
+            if ($trait->hasMethod($method)) {
+                return true;
+            }
+        }
+
         return (
             isset($this->staticReflection[$metadata->name]) &&
             in_array($method, $this->staticReflection[$metadata->name]['methods'])