Skip to content

Commit 4591228

Browse files
Simplify the serializers
1 parent 1010344 commit 4591228

1 file changed

Lines changed: 0 additions & 28 deletions

File tree

core/serializers.py

Lines changed: 0 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -1,32 +1,4 @@
11
from rest_framework import serializers
2-
from .models import DatabaseConnection, SchemaDocument, QueryHistory
3-
4-
5-
class DatabaseConnectionSerializer(serializers.ModelSerializer):
6-
class Meta:
7-
model = DatabaseConnection
8-
fields = ['id', 'name', 'host', 'port', 'database', 'username', 'password', 'created_at', 'updated_at', 'is_active']
9-
extra_kwargs = {'password': {'write_only': True}}
10-
11-
def create(self, validated_data):
12-
password = validated_data.pop('password', None)
13-
instance = super().create(validated_data)
14-
if password:
15-
instance.password = password
16-
instance.save()
17-
return instance
18-
19-
20-
class SchemaDocumentSerializer(serializers.ModelSerializer):
21-
class Meta:
22-
model = SchemaDocument
23-
fields = ['id', 'connection', 'name', 'content', 'uploaded_at']
24-
25-
26-
class QueryHistorySerializer(serializers.ModelSerializer):
27-
class Meta:
28-
model = QueryHistory
29-
fields = ['id', 'connection', 'natural_language', 'generated_sql', 'executed_sql', 'result', 'error', 'created_at']
302

313

324
class QueryRequestSerializer(serializers.Serializer):

0 commit comments

Comments
 (0)