From 07f67c5d1a18b1a167dd898fdd8491c7a871e755 Mon Sep 17 00:00:00 2001
From: Sergey Polischook <SPolischook@gmail.com>
Date: Mon, 30 Dec 2013 04:43:35 +0200
Subject: [PATCH] Allow to not generate extra use

For case when we not generate annotation (by default at doctrine orm:generate-entities) allow to not generate extra use for it -
```php
use Doctrine\ORM\Mapping as ORM;
```
For example if generate entities for my project that use only dbal in pord but use orm for generate entities in dev mode.
---
 lib/Doctrine/ORM/Tools/EntityGenerator.php | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/lib/Doctrine/ORM/Tools/EntityGenerator.php b/lib/Doctrine/ORM/Tools/EntityGenerator.php
index 8e93a2817..7a9f4b34f 100644
--- a/lib/Doctrine/ORM/Tools/EntityGenerator.php
+++ b/lib/Doctrine/ORM/Tools/EntityGenerator.php
@@ -202,9 +202,7 @@ class EntityGenerator
 '<?php
 
 <namespace>
-
-use Doctrine\ORM\Mapping as ORM;
-
+<useStatement>
 <entityAnnotation>
 <entityClassName>
 {
@@ -380,6 +378,7 @@ public function __construct()
     {
         $placeHolders = array(
             '<namespace>',
+            '<useStatement>',
             '<entityAnnotation>',
             '<entityClassName>',
             '<entityBody>'
@@ -387,6 +386,7 @@ public function __construct()
 
         $replacements = array(
             $this->generateEntityNamespace($metadata),
+            $this->generateEntityUse(),
             $this->generateEntityDocBlock($metadata),
             $this->generateEntityClassName($metadata),
             $this->generateEntityBody($metadata)
@@ -568,6 +568,15 @@ public function __construct()
             return 'namespace ' . $this->getNamespace($metadata) .';';
         }
     }
+    
+    protected function generateEntityUse()
+    {
+        if ($this->generateAnnotations) {
+            return "\n".'use Doctrine\ORM\Mapping as ORM;'."\n";
+        } else {
+            return "";
+        }
+    }
 
     /**
      * @param ClassMetadataInfo $metadata