)
from dagster import Enum, EnumValue
from dagster.config.config_type import ConfigScalar
class BigQueryLoadSource(PyEnum):
DataFrame = 'DATA_FRAME'
GCS = 'GCS'
File = 'FILE'
BQCreateDisposition = Enum(
name='BQCreateDisposition',
enum_values=[
EnumValue(CreateDisposition.CREATE_IF_NEEDED),
EnumValue(CreateDisposition.CREATE_NEVER),
],
)
BQPriority = Enum(
name='BQPriority',
enum_values=[EnumValue(QueryPriority.BATCH), EnumValue(QueryPriority.INTERACTIVE)],
)
BQSchemaUpdateOption = Enum(
name='BQSchemaUpdateOption',
enum_values=[
EnumValue(
SchemaUpdateOption.ALLOW_FIELD_ADDITION,
description='Allow adding a nullable field to the schema.',
