zserio.extension.python.SqlDatabaseEmitterTemplateData Maven / Gradle / Ivy

Go to download
package zserio.extension.python;

import java.util.ArrayList;
import java.util.List;

import zserio.ast.Field;
import zserio.ast.SqlDatabaseType;
import zserio.ast.SqlTableType;
import zserio.ast.TypeInstantiation;
import zserio.ast.ZserioType;
import zserio.extension.common.ZserioExtensionException;
import zserio.extension.python.types.PythonNativeType;

/**
 * FreeMarker template data for SqlDatabaseEmitter.
 */
public final class SqlDatabaseEmitterTemplateData extends UserTypeTemplateData
{
    public SqlDatabaseEmitterTemplateData(TemplateDataContext context, SqlDatabaseType databaseType)
            throws ZserioExtensionException
    {
        super(context, databaseType, databaseType);

        importPackage("typing");
        importPackage("apsw");

        fields = new ArrayList();
        for (Field field : databaseType.getFields())
            fields.add(new DatabaseFieldData(context, field, this));
    }

    public Iterable getFields()
    {
        return fields;
    }

    public static final class DatabaseFieldData
    {
        public DatabaseFieldData(TemplateDataContext context, Field field, ImportCollector importCollector)
                throws ZserioExtensionException
        {
            final TypeInstantiation fieldTypeInstantiation = field.getTypeInstantiation();
            final ZserioType fieldBaseType = fieldTypeInstantiation.getBaseType();
            final PythonNativeType nativeType =
                    context.getPythonNativeMapper().getPythonType(fieldTypeInstantiation);
            importCollector.importType(nativeType);

            name = field.getName();
            snakeCaseName = PythonSymbolConverter.toLowerSnakeCase(name);
            typeInfo = new NativeTypeInfoTemplateData(nativeType, fieldTypeInstantiation);
            propertyName = AccessorNameFormatter.getPropertyName(field);
            isWithoutRowIdTable = (fieldBaseType instanceof SqlTableType)
                    ? ((SqlTableType)fieldBaseType).isWithoutRowId()
                    : false;

            docComments = DocCommentsDataCreator.createData(context, field);
        }

        public String getName()
        {
            return name;
        }

        public String getSnakeCaseName()
        {
            return snakeCaseName;
        }

        public NativeTypeInfoTemplateData getTypeInfo()
        {
            return typeInfo;
        }

        public String getPropertyName()
        {
            return propertyName;
        }

        public boolean getIsWithoutRowIdTable()
        {
            return isWithoutRowIdTable;
        }

        public DocCommentsTemplateData getDocComments()
        {
            return docComments;
        }

        private final String name;
        private final String snakeCaseName;
        private final NativeTypeInfoTemplateData typeInfo;
        private final String propertyName;
        private final boolean isWithoutRowIdTable;
        private final DocCommentsTemplateData docComments;
    }

    private final List fields;
}