fix: migration generator does not change - to _ (#96)

postlight · May 18, 2016 · b165576 · b165576
1 parent 81cdd21
commit b165576
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/packages/cli/templates/model-migration.js b/src/packages/cli/templates/model-migration.js
@@ -1,6 +1,7 @@
-import { underscore, pluralize } from 'inflection';
+import { pluralize } from 'inflection';
 
 import indent from '../utils/indent';
+import underscore from '../../../utils/underscore';
 
 export default (name, attrs = []) => {
   const table = pluralize(underscore(name));
@@ -11,6 +12,8 @@ export default (name, attrs = []) => {
     .map(attr => attr.split(':'))
     .filter(([, type]) => !/^has-(one|many)$/g.test(type))
     .map(([column, type]) => {
+      column = underscore(column);
+
       if (type === 'belongs-to') {
         type = 'integer';
         column = `${column}_id`;