Elep

from rest_framework import serializers
from .models import SnowflakeAuthentication

class SnowflakeAuthenticationSerializer(serializers.Serializer):
    id = serializers.IntegerField(read_only=True)
    account = serializers.CharField(max_length=255)
    user = serializers.CharField(max_length=255, style={'input_type': 'password'})
    password = serializers.CharField(max_length=255)
    role = serializers.ChoiceField(choices=[('accountadmin', 'accountadmin'), ('orgadmin', 'orgadmin')])
    sf_user_id = serializers.CharField(max_length=255)

def create(self, validated_data):
        print(validated_data)
        return SnowflakeAuthentication.objects.create(**validated_data)
    #
    # def update(self, instance, validated_data):
    #     instance.account = validated_data.get('account', instance.account)
    #     instance.user = validated_data.get('user', instance.user)
    #     instance.password = validated_data.get('password', instance.password)
    #     instance.role = validated_data.get('role', instance.role)
    #     instance.user_id = validated_data.get('user_id', instance.user_id)
    #     instance.save()
    #     return instance

class SnowflakeDatabaseList(serializers.Serializer):
    snowflake_id = serializers.IntegerField()

class SnowflakeDatabaseListReturn(serializers.Serializer):
    snowflake_id = serializers.IntegerField()
    warehouses = serializers.ListField()
    databases = serializers.ListField()

class SnowflakeSchemaList(serializers.Serializer):
    snowflake_id = serializers.IntegerField()
    warehouse = serializers.CharField(max_length=255)
    database = serializers.CharField(max_length=255)

class SnowflakeSchemaListReturn(serializers.Serializer):
    snowflake_id = serializers.IntegerField()
    warehouse = serializers.CharField(max_length=255)
    database = serializers.CharField(max_length=255)
    schemas = serializers.ListField()

class SnowflakeTableList(serializers.Serializer):
    snowflake_id = serializers.IntegerField()
    warehouse = serializers.CharField(max_length=255)
    database = serializers.CharField(max_length=255)
    schema = serializers.CharField(max_length=255)

class SnowflakeTableListReturn(serializers.Serializer):
    snowflake_id = serializers.IntegerField()
    warehouse = serializers.CharField(max_length=255)
    database = serializers.CharField(max_length=255)
    schema = serializers.CharField(max_length=255)
    tables = serializers.ListField()

class SnowflakeColumnList(serializers.Serializer):
    snowflake_id = serializers.IntegerField()
    warehouse = serializers.CharField(max_length=255)
    database = serializers.CharField(max_length=255)
    schema = serializers.CharField(max_length=255)
    table = serializers.CharField(max_length=255)

class SnowflakeColumnListReturn(serializers.Serializer):
    snowflake_id = serializers.IntegerField()
    warehouse = serializers.CharField(max_length=255)
    database = serializers.CharField(max_length=255)
    schema = serializers.CharField(max_length=255)
    table = serializers.CharField(max_length=255)
    columns = serializers.ListField()

class SnowflakeMapping(serializers.Serializer):
    warehouse = serializers.CharField(max_length=255)
    database = serializers.CharField(max_length=255)
    schema = serializers.CharField(max_length=255)
    table = serializers.CharField(max_length=255)

retention_database = serializers.CharField(max_length=255)

snowflake_id = serializers.IntegerField()
    timestamp = serializers.CharField(max_length=255)
    e = serializers.CharField(max_length=255)
    customer = serializers.CharField(max_length=255)
    metadata = serializers.CharField(max_length=255)

client_id = serializers.CharField(max_length=255)
    subdomain = serializers.CharField(max_length=255)

class SnowflakeError(serializers.Serializer):
    error = serializers.CharField()