refactor: pass a short lang to Generator.for()

Relate to #31

Author: php-coder

src/cli.js

@@ -353,7 +353,8 @@ const main = async (argv) => {
         fs.mkdirSync(destDir, {recursive: true})
     }
 
-    const generator = Generator.for(argv.lang)
+    const lang = lang2extension(argv.lang)
+    const generator = Generator.for(lang)
 
     await createApp(destDir, argv)
     await createDb(destDir, argv)

src/generator/Generator.js

@@ -13,7 +13,7 @@ module.exports = class Generator {
                 return new TsGenerator()
             case 'go':
                 return new GoGenerator()
-            case 'python':
+            case 'py':
                 return new PyGenerator()
             default:
                 throw new Error(`Unsupported language: ${lang}`)