Browse Source

Rename RelationType model

pull/1655/head
Hironsan 2 years ago
parent
commit
b3a20ca283
5 changed files with 27 additions and 8 deletions
  1. 19
      backend/label_types/migrations/0002_rename_relationtypes_relationtype.py
  2. 2
      backend/label_types/models.py
  3. 4
      backend/label_types/serializers.py
  4. 6
      backend/label_types/views.py
  5. 4
      backend/labels/models.py

19
backend/label_types/migrations/0002_rename_relationtypes_relationtype.py

@ -0,0 +1,19 @@
# Generated by Django 3.2.11 on 2022-01-27 07:27
from django.db import migrations
class Migration(migrations.Migration):
dependencies = [
('labels', '0003_auto_20220127_0654'),
('api', '0033_auto_20220127_0654'),
('label_types', '0001_initial'),
]
operations = [
migrations.RenameModel(
old_name='RelationTypes',
new_name='RelationType',
),
]

2
backend/label_types/models.py

@ -88,7 +88,7 @@ class SpanType(LabelType):
return SpanType.objects.filter(project=self.project)
class RelationTypes(models.Model):
class RelationType(models.Model):
color = models.TextField()
name = models.TextField()
project = models.ForeignKey(Project, related_name='relation_types', on_delete=models.CASCADE)

4
backend/label_types/serializers.py

@ -1,7 +1,7 @@
from rest_framework import serializers
from rest_framework.exceptions import ValidationError
from .models import LabelType, CategoryType, SpanType, RelationTypes
from .models import LabelType, CategoryType, SpanType, RelationType
class LabelSerializer(serializers.ModelSerializer):
@ -84,5 +84,5 @@ class RelationTypesSerializer(serializers.ModelSerializer):
return super().validate(attrs)
class Meta:
model = RelationTypes
model = RelationType
fields = ('id', 'color', 'name')

6
backend/label_types/views.py

@ -11,7 +11,7 @@ from rest_framework.response import Response
from rest_framework.views import APIView
from members.permissions import IsInProjectReadOnlyOrAdmin, IsProjectAdmin
from .models import LabelType, CategoryType, SpanType, RelationTypes
from .models import LabelType, CategoryType, SpanType, RelationType
from .exceptions import LabelValidationError
from .serializers import (CategoryTypeSerializer, LabelSerializer,
RelationTypesSerializer, SpanTypeSerializer)
@ -70,12 +70,12 @@ class SpanTypeDetail(generics.RetrieveUpdateDestroyAPIView):
class RelationTypeList(LabelList):
model = RelationTypes
model = RelationType
serializer_class = RelationTypesSerializer
class RelationTypeDetail(generics.RetrieveUpdateDestroyAPIView):
queryset = RelationTypes.objects.all()
queryset = RelationType.objects.all()
serializer_class = RelationTypesSerializer
lookup_url_kwarg = 'relation_type_id'
permission_classes = [IsAuthenticated & IsInProjectReadOnlyOrAdmin]

4
backend/labels/models.py

@ -4,7 +4,7 @@ from django.db import models
from .managers import LabelManager, CategoryManager, SpanManager, TextLabelManager
from api.models import Example, Project
from label_types.models import CategoryType, SpanType, RelationTypes
from label_types.models import CategoryType, SpanType, RelationType
class Label(models.Model):
@ -108,7 +108,7 @@ class TextLabel(Label):
class Relation(models.Model):
annotation_id_1 = models.IntegerField()
annotation_id_2 = models.IntegerField()
type = models.ForeignKey(RelationTypes, related_name='annotation_relations', on_delete=models.CASCADE)
type = models.ForeignKey(RelationType, related_name='annotation_relations', on_delete=models.CASCADE)
timestamp = models.DateTimeField()
user = models.ForeignKey(User, related_name='annotation_relations', on_delete=models.CASCADE)
project = models.ForeignKey(Project, related_name='annotation_relations', on_delete=models.CASCADE)

Loading…
Cancel
Save