diff --git a/backend/webapp/prototype/filehandler/functions.py b/backend/webapp/prototype/filehandler/functions.py index 56301b7..6d39e2c 100644 --- a/backend/webapp/prototype/filehandler/functions.py +++ b/backend/webapp/prototype/filehandler/functions.py @@ -1,10 +1,32 @@ -from xml.etree.ElementTree import parse - -def isValidXML(file): - try: - parse(file) - except: - return(False) - return(True) +from prototype.filehandler.models import Forum, Discussion, Post, Paragraph +def addToDatabase(data, file_id): + out = [] + forum = Forum(forum_id = data['id'], name = data['name'], document_id = file_id) + forum.save() + for discussion_ in data['discussions']: + discussion = Discussion(discussion_id = discussion_['id'], title = discussion_['title'], first_post = discussion_['first_post'], forum_id = forum.pk) + discussion.save() + for post_ in discussion_['posts']: + post = Post(post_id = post_['id'], parent = post_['parent'], author = post_['author'], discussion_id = discussion.pk) + post.save() + for paragraph_ in post_['message']: + paragraph = Paragraph(message = paragraph_, label = '', post_id = post.pk) + paragraph.save() + out.append(paragraph.pk) + return(out) +def listDiscussionsFromFile(id): + out = {} + forum = Forum.objects.get(document_id = id) + out['id'] = id + out['name'] = forum.name + discussions = Discussion.objects.filter(forum_id = forum.pk) + discussions_ = [] + for elem in discussions: + obj = {} + obj['id'] = elem.pk + obj['title'] = elem.title + discussions_.append(obj) + out['discussions'] = discussions_ + return(out) diff --git a/backend/webapp/prototype/filehandler/models.py b/backend/webapp/prototype/filehandler/models.py index 84cc772..a7dc1a0 100644 --- a/backend/webapp/prototype/filehandler/models.py +++ b/backend/webapp/prototype/filehandler/models.py @@ -4,3 +4,25 @@ from django.db import models class Document(models.Model): file = models.FileField(upload_to='documents/') uploaded_at = models.DateTimeField(auto_now_add=True) + +class Forum(models.Model): + forum_id = models.IntegerField() + name = models.CharField(max_length=255) + document = models.ForeignKey(Document, on_delete=models.CASCADE) + +class Discussion(models.Model): + discussion_id = models.IntegerField() + title = models.CharField(max_length=255) + first_post = models.IntegerField() + forum = models.ForeignKey(Forum, on_delete=models.CASCADE) + +class Post(models.Model): + post_id = models.IntegerField() + parent = models.IntegerField() + author = models.IntegerField() + discussion = models.ForeignKey(Discussion, on_delete=models.CASCADE) + +class Paragraph(models.Model): + message = models.TextField() + label = models.CharField(max_length=255) + post = models.ForeignKey(Post, on_delete=models.CASCADE) diff --git a/backend/webapp/prototype/filehandler/views.py b/backend/webapp/prototype/filehandler/views.py index 42942dc..05a01f4 100644 --- a/backend/webapp/prototype/filehandler/views.py +++ b/backend/webapp/prototype/filehandler/views.py @@ -4,10 +4,11 @@ from django.core.files.storage import FileSystemStorage from django.views.decorators.csrf import csrf_exempt from django.http import JsonResponse, HttpResponse -from prototype.filehandler.models import Document +from prototype.filehandler.models import Document, Forum from prototype.filehandler.forms import DocumentForm - from prototype.filehandler.xmlParser import parseData +from prototype.filehandler.functions import addToDatabase, listDiscussionsFromFile + def home(request): documents = Document.objects.all() @@ -18,13 +19,13 @@ def model_form_upload(request): if request.method == 'POST': form = DocumentForm(request.POST, request.FILES) if form.is_valid(): - try: - data = parseData(request.FILES['file']) - print(data) - form.save() - return JsonResponse(data, safe=False) - except: - return HttpResponse('Niepoprawny format XML', status=406) + data = parseData(request.FILES['file']) + file_id = (form.save()).pk + addToDatabase(data, file_id) + output = listDiscussionsFromFile(file_id) + return JsonResponse(output, safe=False) + else: + return HttpResponse('Niepoprawny format XML', status=406) else: form = DocumentForm() return render(request, 'core/model_form_upload.html', {