add own includes and imports
This commit is contained in:
@@ -13,15 +13,15 @@ function generateModels(model) {
|
||||
const properties = model.model?.properties ?? {};
|
||||
const ownIncludes = Object.values(properties)
|
||||
.filter(p => p.property.constructor.name === 'ConstrainedReferenceModel')
|
||||
.map(p => `from ${p.property.ref.name} import ${p.property.ref.name}`).join('\n')
|
||||
.map(p => `from ${p.property.ref.name} import ${p.property.ref.name}`)
|
||||
const anonymousModels = Object.values(properties)
|
||||
.filter(p => p.property.constructor.name === 'ConstrainedArrayModel')
|
||||
.map(p => `from ${p.property.valueModel.name} import ${p.property.valueModel.name}`)
|
||||
const imports = model.dependencies
|
||||
.join('\n');
|
||||
|
||||
return (
|
||||
<File name={`${model.modelName}.py`}>
|
||||
{imports}
|
||||
{'\n\n'}
|
||||
{ownIncludes}
|
||||
{[...imports, ...ownIncludes, ...anonymousModels].join('\n')}
|
||||
{'\n\n'}
|
||||
{model.result}
|
||||
</File>
|
||||
|
||||
Reference in New Issue
Block a user