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 Bulletin

class SimpleBulletinSerializer(serializers.ModelSerializer):
	topic = TopicSerializer('get_subject')
	tags = TagSerializer(many = True)
	pendencies_resource = PendenciesSerializer(many = True)
	indicators = serializers.CharField(required = False, allow_blank = True, max_length = 255)
	file_content = 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_content"] in files.namelist():
				file_path = os.path.join(settings.MEDIA_ROOT, data["file_content"])

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

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

					new_name = "goal_" + str(time.time()) + os.path.splitext(data["file_content"])[1]
					
					new_path = os.path.join("bulletin", os.path.join("goals", new_name))

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

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

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

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

					new_name = "ind_" + str(time.time()) + os.path.splitext(data["indicators"])[1]
					
					new_path = os.path.join("bulletin", os.path.join("indicators", new_name))

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

		return data

	class Meta:
		model = Bulletin
		extra_kwargs = {
			"tags": {
				"validators": [],
			},
		}
		exclude = ('students', 'groups',)
		validators = []

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

		bulletin = 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"])

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

				bulletin = Bulletin()
				bulletin.name = bulletin_data["name"]
				bulletin.brief_description = bulletin_data["brief_description"]
				bulletin.show_window = bulletin_data["show_window"]
				bulletin.all_students = bulletin_data["all_students"]
				bulletin.visible = bulletin_data["visible"]
				bulletin.order = bulletin_data["order"]
				bulletin.topic = bulletin_data["topic"]
				bulletin.content = bulletin_data["content"]
				bulletin.file_content = bulletin_data["file_content"]
				bulletin.indicators = bulletin_data["indicators"]

				bulletin.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"])

						bulletin.tags.add(tag)

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

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

		return bulletin

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

class CompleteBulletinSerializer(serializers.ModelSerializer):
	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)
	indicators = serializers.CharField(required = False, allow_blank = True, max_length = 255)
	file_content = serializers.CharField(required = False, allow_blank = True, max_length = 255)

	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_content"] in files.namelist():
				file_path = os.path.join(settings.MEDIA_ROOT, data["file_content"])

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

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

					new_name = "goal_" + str(time.time()) + os.path.splitext(data["file_content"])[1]
					
					new_path = os.path.join("bulletin", os.path.join("goals", new_name))

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

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

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

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

					new_name = "ind_" + str(time.time()) + os.path.splitext(data["indicators"])[1]
					
					new_path = os.path.join("bulletin", os.path.join("indicators", new_name))

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

		return data

	class Meta:
		model = Bulletin
		extra_kwargs = {
			"tags": {
				"validators": [],
			},
		}
		fields = '__all__'
		validators = []

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

		bulletin = 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"])

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

				bulletin = Bulletin()
				bulletin.name = bulletin_data["name"]
				bulletin.brief_description = bulletin_data["brief_description"]
				bulletin.show_window = bulletin_data["show_window"]
				bulletin.all_students = bulletin_data["all_students"]
				bulletin.visible = bulletin_data["visible"]
				bulletin.order = bulletin_data["order"]
				bulletin.topic = bulletin_data["topic"]
				bulletin.content = bulletin_data["content"]
				bulletin.file_content = bulletin_data["file_content"]
				bulletin.indicators = bulletin_data["indicators"]

				bulletin.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"])

						bulletin.tags.add(tag)

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

				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)

					bulletin.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)

					bulletin.groups.add(group)

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

		return bulletin

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