Make 'yo generate overwrite yo_db file (cloudspannerecosystem#61)

generator/generator.go

@@ -167,17 +167,12 @@ func (g *Generator) getFile(ds *basicDataSet, t *TBuf) (*os.File, error) {
 	// default open mode
 	mode := os.O_RDWR | os.O_CREATE | os.O_TRUNC
 
-	// stat file to determine if file already exists
+	// stat file to determine if file is a directory.
 	fi, err := os.Stat(filename)
 	if err == nil && fi.IsDir() {
 		return nil, errors.New("filename cannot be directory")
 	}
 
-	// skip
-	if t.TemplateType == YOTemplate && fi != nil {
-		return nil, nil
-	}
-
 	// open file
 	f, err = os.OpenFile(filename, mode, 0666)
 	if err != nil {
@@ -231,11 +226,6 @@ func (g *Generator) writeTypes(ds *basicDataSet) error {
 			return err
 		}
 
-		// should only be nil when type == yo
-		if f == nil {
-			continue
-		}
-
 		// write segment
 		if _, err = t.Buf.WriteTo(f); err != nil {
 			return err