mirror of
https://github.com/IRBorisov/ConceptPortal.git
synced 2025-06-26 13:00:39 +03:00
218 lines
9.0 KiB
Python
218 lines
9.0 KiB
Python
import json
|
|
from django.http import HttpResponse
|
|
from django_filters.rest_framework import DjangoFilterBackend
|
|
from rest_framework import views, viewsets, filters, generics, permissions
|
|
from rest_framework.decorators import action
|
|
from rest_framework.response import Response
|
|
from rest_framework.decorators import api_view
|
|
|
|
import pyconcept
|
|
from . import models
|
|
from . import serializers
|
|
from . import utils
|
|
|
|
|
|
class ConstituentAPIView(generics.RetrieveUpdateAPIView):
|
|
queryset = models.Constituenta.objects.all()
|
|
serializer_class = serializers.ConstituentaSerializer
|
|
|
|
def get_permissions(self):
|
|
result = super().get_permissions()
|
|
if self.request.method.lower() == 'get':
|
|
result.append(permissions.AllowAny())
|
|
else:
|
|
result.append(utils.SchemaOwnerOrAdmin())
|
|
return result
|
|
|
|
|
|
class RSFormViewSet(viewsets.ModelViewSet):
|
|
queryset = models.RSForm.objects.all()
|
|
serializer_class = serializers.RSFormSerializer
|
|
|
|
filter_backends = (DjangoFilterBackend, filters.OrderingFilter)
|
|
filterset_fields = ['owner', 'is_common']
|
|
ordering_fields = ('owner', 'title', 'time_update')
|
|
ordering = ('-time_update')
|
|
|
|
def perform_create(self, serializer):
|
|
if not self.request.user.is_anonymous and 'owner' not in self.request.POST:
|
|
return serializer.save(owner=self.request.user)
|
|
else:
|
|
return serializer.save()
|
|
|
|
def get_permissions(self):
|
|
if self.action in ['update', 'destroy', 'partial_update',
|
|
'cst_create', 'cst_multidelete']:
|
|
permission_classes = [utils.ObjectOwnerOrAdmin]
|
|
elif self.action in ['create', 'claim']:
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
else:
|
|
permission_classes = [permissions.AllowAny]
|
|
return [permission() for permission in permission_classes]
|
|
|
|
@action(detail=True, methods=['post'], url_path='cst-create')
|
|
def cst_create(self, request, pk):
|
|
''' Create new constituenta '''
|
|
schema: models.RSForm = self.get_object()
|
|
serializer = serializers.CstCreateSerializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
if ('insert_after' in serializer.validated_data):
|
|
cstafter = models.Constituenta.objects.get(pk=serializer.validated_data['insert_after'])
|
|
constituenta = schema.insert_at(cstafter.order + 1,
|
|
serializer.validated_data['alias'],
|
|
serializer.validated_data['csttype'])
|
|
else:
|
|
constituenta = schema.insert_last(serializer.validated_data['alias'], serializer.validated_data['csttype'])
|
|
schema.refresh_from_db()
|
|
outSerializer = serializers.RSFormDetailsSerlializer(schema)
|
|
response = Response(status=201, data={'new_cst': constituenta.to_json(), 'schema': outSerializer.data})
|
|
response['Location'] = constituenta.get_absolute_url()
|
|
return response
|
|
|
|
@action(detail=True, methods=['patch'], url_path='cst-multidelete')
|
|
def cst_multidelete(self, request, pk):
|
|
''' Delete multiple constituents '''
|
|
schema: models.RSForm = self.get_object()
|
|
serializer = serializers.CstListSerlializer(data=request.data, context={'schema': schema})
|
|
serializer.is_valid(raise_exception=True)
|
|
schema.delete_cst(serializer.validated_data['constituents'])
|
|
schema.refresh_from_db()
|
|
outSerializer = serializers.RSFormDetailsSerlializer(schema)
|
|
return Response(status=202, data=outSerializer.data)
|
|
|
|
@action(detail=True, methods=['patch'], url_path='cst-moveto')
|
|
def cst_moveto(self, request, pk):
|
|
''' Delete multiple constituents '''
|
|
schema: models.RSForm = self.get_object()
|
|
serializer = serializers.CstMoveSerlializer(data=request.data, context={'schema': schema})
|
|
serializer.is_valid(raise_exception=True)
|
|
schema.move_cst(serializer.validated_data['constituents'], serializer.validated_data['move_to'])
|
|
schema.refresh_from_db()
|
|
outSerializer = serializers.RSFormDetailsSerlializer(schema)
|
|
return Response(status=200, data=outSerializer.data)
|
|
|
|
@action(detail=True, methods=['post'])
|
|
def claim(self, request, pk=None):
|
|
schema: models.RSForm = self.get_object()
|
|
if schema.owner == self.request.user:
|
|
return Response(status=304)
|
|
else:
|
|
schema.owner = self.request.user
|
|
schema.save()
|
|
return Response(status=200, data=serializers.RSFormSerializer(schema).data)
|
|
|
|
@action(detail=True, methods=['get'])
|
|
def contents(self, request, pk):
|
|
''' View schema contents (including constituents) '''
|
|
schema = self.get_object().to_json()
|
|
return Response(schema)
|
|
|
|
@action(detail=True, methods=['get'])
|
|
def details(self, request, pk):
|
|
''' Detailed schema view including statuses '''
|
|
schema: models.RSForm = self.get_object()
|
|
serializer = serializers.RSFormDetailsSerlializer(schema)
|
|
return Response(serializer.data)
|
|
|
|
@action(detail=True, methods=['post'])
|
|
def check(self, request, pk):
|
|
''' Check RS expression against schema context '''
|
|
schema = self.get_object().to_json()
|
|
serializer = serializers.ExpressionSerializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
expression = serializer.validated_data['expression']
|
|
result = pyconcept.check_expression(json.dumps(schema), expression)
|
|
return Response(json.loads(result))
|
|
|
|
@action(detail=True, methods=['get'], url_path='export-trs')
|
|
def export_trs(self, request, pk):
|
|
''' Download Exteor compatible file '''
|
|
schema = self.get_object().to_json()
|
|
trs = utils.write_trs(schema)
|
|
filename = self.get_object().alias
|
|
if filename == '' or not filename.isascii():
|
|
# Note: non-ascii symbols in Content-Disposition
|
|
# are not supported by some browsers
|
|
filename = 'Schema'
|
|
filename += '.trs'
|
|
|
|
response = HttpResponse(trs, content_type='application/zip')
|
|
response['Content-Disposition'] = f'attachment; filename={filename}'
|
|
return response
|
|
|
|
|
|
class TrsImportView(views.APIView):
|
|
''' Upload RS form in Exteor format '''
|
|
serializer_class = serializers.FileSerializer
|
|
|
|
def post(self, request, format=None):
|
|
data = utils.read_trs(request.FILES['file'].file)
|
|
owner = self.request.user
|
|
if owner.is_anonymous:
|
|
owner = None
|
|
schema = models.RSForm.import_json(owner, data)
|
|
result = serializers.RSFormSerializer(schema)
|
|
return Response(status=201, data=result.data)
|
|
|
|
|
|
@api_view(['POST'])
|
|
def create_rsform(request):
|
|
''' Create RSForm from user input and/or trs file '''
|
|
owner = request.user
|
|
if owner.is_anonymous:
|
|
owner = None
|
|
if ('file' not in request.FILES):
|
|
serializer = serializers.RSFormSerializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
schema = models.RSForm.objects.create(
|
|
title=request.data['title'],
|
|
owner=owner,
|
|
alias=request.data.get('alias', ''),
|
|
comment=request.data.get('comment', ''),
|
|
is_common=request.data.get('is_common', False),
|
|
)
|
|
else:
|
|
data = utils.read_trs(request.FILES['file'].file)
|
|
if ('title' in request.data and request.data['title'] != ''):
|
|
data['title'] = request.data['title']
|
|
if ('alias' in request.data and request.data['alias'] != ''):
|
|
data['alias'] = request.data['alias']
|
|
if ('comment' in request.data and request.data['comment'] != ''):
|
|
data['comment'] = request.data['comment']
|
|
is_common = True
|
|
if ('is_common' in request.data):
|
|
is_common = request.data['is_common']
|
|
schema = models.RSForm.import_json(owner, data, is_common)
|
|
result = serializers.RSFormSerializer(schema)
|
|
return Response(status=201, data=result.data)
|
|
|
|
|
|
@api_view(['POST'])
|
|
def parse_expression(request):
|
|
'''Parse RS expression '''
|
|
serializer = serializers.ExpressionSerializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
expression = serializer.validated_data['expression']
|
|
result = pyconcept.parse_expression(expression)
|
|
return Response(json.loads(result))
|
|
|
|
|
|
@api_view(['POST'])
|
|
def convert_to_ascii(request):
|
|
''' Convert to ASCII syntax '''
|
|
serializer = serializers.ExpressionSerializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
expression = serializer.validated_data['expression']
|
|
result = pyconcept.convert_to_ascii(expression)
|
|
return Response({'result': result})
|
|
|
|
|
|
@api_view(['POST'])
|
|
def convert_to_math(request):
|
|
'''Convert to MATH syntax '''
|
|
serializer = serializers.ExpressionSerializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
expression = serializer.validated_data['expression']
|
|
result = pyconcept.convert_to_math(expression)
|
|
return Response({'result': result})
|