visitModelInit()

in lib/generator.js [1484:1505]

• 19 lines of code
• 5 McCabe index (conditional complexity)

  visitModelInit(models) {
    const modelNames = [];
    const moreExports = this.moreExports || [];
    for (let i = 0; i < models.length; i++) {
      const ast = models[i];
      const modelName = _subModelName(_name(ast.modelName));
      modelNames.push(modelName);
      this.emit(`from ._${_snakeCase(modelName)} import ${modelName}\n`);
    }
    
    this.emit('\n__all__ = [\n');
    this.emit(modelNames.join(',\n    '), 2);
    if(moreExports && moreExports.length > 0) {
      this.emit((',\n'));
      this.emit(moreExports.join(',\n    '), 2);
    }
    
    this.emit('\n]\n');

    this.save(path.join(this.modelPath, '__init__.py'));
    this.moreExports = [];
  }