Software Público Brasileiro

import os
import zipfile
import time
from django.conf import settings
from django.core.files import File
from rest_framework import serializers
from django.shortcuts import get_object_or_404

from subjects.serializers import TagSerializer
from topics.serializers import TopicSerializer
from pendencies.serializers import PendenciesSerializer
from students_group.serializers import StudentsGroupSerializer
from users.serializers import UserBackupSerializer

from subjects.models import Tag, Subject
from topics.models import Topic, Resource
from pendencies.models import Pendencies
from students_group.models import StudentsGroup
from log.models import Log
from users.models import User

from .models import PDFFile

class SimplePDFFileSerializer(serializers.ModelSerializer):
	topic = TopicSerializer('get_subject')
	tags = TagSerializer(many = True)
	pendencies_resource = PendenciesSerializer(many = True)
	file = serializers.CharField(required = False, allow_blank = True, max_length = 255)

	def get_subject(self, obj):
		subject = self.context.get("subject", None)

		return subject

	def validate(self, data):
		files = self.context.get('files', None)

		if files:
			if data["file"] in files.namelist():
				file_path = os.path.join(settings.MEDIA_ROOT, data["file"])

				if os.path.isfile(file_path):
					dst_path = os.path.join(settings.MEDIA_ROOT, "tmp")

					path = files.extract(data["file"], dst_path)

					new_name = os.path.join("files","file_" + str(time.time()) + os.path.splitext(data["file"])[1])

					os.rename(os.path.join(dst_path, path), os.path.join(settings.MEDIA_ROOT, new_name))
					
					data["file"] = new_name
				else:
					path = files.extract(data["file"], settings.MEDIA_ROOT)
			else:
				data["file"] = None
		else:
			data["file"] = None

		return data

	class Meta:
		model = PDFFile
		exclude = ('students', 'groups',)

	def create(self, data):
		topic = data['topic']

		pdf = None

		if not topic["id"] is None:
			if "subject" in topic:
				r_exits = Resource.objects.filter(topic__subject = topic["subject"], name__unaccent__iexact = data["name"])
			else:
				r_exits = Resource.objects.filter(topic__subject__id = topic["subject_id"], name__unaccent__iexact = data["name"])

			if not r_exits.exists():
				if topic['id'] == "":
					topic_exist = Topic.objects.filter(subject = topic['subject'], name__unaccent__iexact = topic["name"])

					if topic_exist.exists():
						topic = topic_exist[0]
					else:
						topic = Topic.objects.create(name = topic['name'], subject = topic['subject'], repository = topic['repository'], visible = topic['visible'], order = topic['order'], description = topic['description'])
					
					data["topic"] = topic
				else:
					data["topic"] = get_object_or_404(Topic, id = topic["id"])

				pdf_data = data
				
				pendencies = pdf_data["pendencies_resource"]
				del pdf_data["pendencies_resource"]

				pdf = PDFFile()
				pdf.name = pdf_data["name"]
				pdf.brief_description = pdf_data["brief_description"]
				pdf.show_window = pdf_data["show_window"]
				pdf.all_students = pdf_data["all_students"]
				pdf.visible = pdf_data["visible"]
				pdf.order = pdf_data["order"]
				pdf.topic = pdf_data["topic"]
				pdf.file = pdf_data["file"]

				pdf.save()

				tags = data["tags"]

				for tag in tags:
					if not tag["name"] == "":
						if tag["id"] == "":
							tag = Tag.objects.create(name = tag["name"])
						else:
							tag = get_object_or_404(Tag, id = tag["id"])

						pdf.tags.add(tag)
				
				resource = get_object_or_404(Resource, id = pdf.id)

				for pend in pendencies:
					Pendencies.objects.create(resource = resource, **pend)

		return pdf

	def update(self, instance, data):
		return instance

class CompletePDFFileSerializer(serializers.ModelSerializer):
	file = serializers.CharField(required = False, allow_blank = True, max_length = 255)
	topic = TopicSerializer('get_subject')
	tags = TagSerializer(many = True)
	pendencies_resource = PendenciesSerializer(many = True)
	groups = StudentsGroupSerializer('get_files', many = True)
	students = UserBackupSerializer('get_files', many = True)

	def get_subject(self, obj):
		subject = self.context.get("subject", None)

		return subject

	def get_files(self, obj):
		files = self.context.get("files", None)

		return files

	def validate(self, data):
		files = self.context.get('files', None)

		if files:
			if data["file"] in files.namelist():
				file_path = os.path.join(settings.MEDIA_ROOT, data["file"])

				if os.path.isfile(file_path):
					dst_path = os.path.join(settings.MEDIA_ROOT, "tmp")

					path = files.extract(data["file"], dst_path)

					new_name = os.path.join("files","file_" + str(time.time()) + os.path.splitext(data["file"])[1])

					os.rename(os.path.join(dst_path, path), os.path.join(settings.MEDIA_ROOT, new_name))
					
					data["file"] = new_name
				else:
					path = files.extract(data["file"], settings.MEDIA_ROOT)
			else:
				data["file"] = None
		else:
			data["file"] = None

		return data

	class Meta:
		model = PDFFile
		fields = '__all__'

	def create(self, data):
		topic = data['topic']

		pdf = None

		if not topic["id"] is None:

			if "subject" in topic:
				r_exits = Resource.objects.filter(topic__subject = topic["subject"], name__unaccent__iexact = data["name"])
			else:
				r_exits = Resource.objects.filter(topic__subject__id = topic["subject_id"], name__unaccent__iexact = data["name"])

			if not r_exits.exists():
				if topic['id'] == "":
					topic_exist = Topic.objects.filter(subject = topic['subject'], name__unaccent__iexact = topic["name"])

					if topic_exist.exists():
						topic = topic_exist[0]
					else:
						topic = Topic.objects.create(name = topic['name'], subject = topic['subject'], repository = topic['repository'], visible = topic['visible'], order = topic['order'], description = topic['description'])
					
					data["topic"] = topic
				else:
					data["topic"] = get_object_or_404(Topic, id = topic["id"])

				pdf_data = data
				
				pendencies = pdf_data["pendencies_resource"]
				del pdf_data["pendencies_resource"]

				pdf = PDFFile()
				pdf.name = pdf_data["name"]
				pdf.brief_description = pdf_data["brief_description"]
				pdf.show_window = pdf_data["show_window"]
				pdf.all_students = pdf_data["all_students"]
				pdf.visible = pdf_data["visible"]
				pdf.order = pdf_data["order"]
				pdf.topic = pdf_data["topic"]
				pdf.file = pdf_data["file"]

				pdf.save()

				tags = data["tags"]

				for tag in tags:
					if not tag["name"] == "":
						if tag["id"] == "":
							tag = Tag.objects.create(name = tag["name"])
						else:
							tag = get_object_or_404(Tag, id = tag["id"])

						pdf.tags.add(tag)
				
				students = data["students"]
				subject = get_object_or_404(Subject, slug = self.context.get("subject", None))

				for student_data in students:
					logs = student_data["get_items"]
					
					if student_data["id"] == "":
						u_exist = User.objects.filter(email = student_data["email"])

						if not u_exist.exists():
							student = u_exist[0]

							for log in logs:
								log["user_id"] = student.id

								l_exists = Log.objects.filter(user_id = log["user_id"], user = log["user"], user_email = log["user_email"], action = log["action"], resource = log["resource"], component = log["component"], context = log["context"])

								if not l_exists.exists():
									Log.objects.create(**log)
						else:
							student = User()
							student.email = student_data["email"]
							student.username = student_data["username"]
							student.last_name = student_data["last_name"] 
							student.social_name = student_data["social_name"] 
							student.show_email = student_data["show_email"] 
							student.is_staff = student_data["is_staff"] 
							student.is_active = student_data["is_active"]
							student.image = student_data["image"]

							student.save()

							for log in logs:
								log["user_id"] = student.id

								Log.objects.create(**log)
					else:
						student = get_object_or_404(User, id = student_data["id"])

						for log in logs:
							l_exists = Log.objects.filter(user_id = log["user_id"], user = log["user"], user_email = log["user_email"], action = log["action"], resource = log["resource"], component = log["component"], context = log["context"])

							if not l_exists.exists():
								Log.objects.create(**log)

					pdf.students.add(student)
					subject.students.add(student)

				groups = data["groups"]


				for group_data in groups:
					g_exists = StudentsGroup.objects.filter(subject = subject, slug = group_data["slug"])

					if g_exists.exists():
						group = g_exists[0]
					else:
						group = StudentsGroup()
						group.name = group_data["name"]
						group.description = group_data["description"]
						group.subject = subject

						group.save()

						for participant in group_data["participants"]:
							p_user = get_object_or_404(User, email = participant["email"])

							group.participants.add(p_user)

					pdf.groups.add(group)

				resource = get_object_or_404(Resource, id = pdf.id)

				for pend in pendencies:
					Pendencies.objects.create(resource = resource, **pend)

		return pdf