From 5e2e0c8fb80eb15a8b394466726f00cb18d4324c Mon Sep 17 00:00:00 2001 From: Zambom Date: Wed, 7 Jun 2017 00:13:32 -0300 Subject: [PATCH] Start writing validationand save scripts for complete restore --- file_link/serializers.py | 92 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++- goals/serializers.py | 74 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++-- links/serializers.py | 67 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++-- pdf_file/serializers.py | 94 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++-- webpage/serializers.py | 67 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++-- youtube_video/serializers.py | 68 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++-- 6 files changed, 451 insertions(+), 11 deletions(-) diff --git a/file_link/serializers.py b/file_link/serializers.py index bea625d..02b6ca5 100644 --- a/file_link/serializers.py +++ b/file_link/serializers.py @@ -130,12 +130,102 @@ class SimpleFileLinkSerializer(serializers.ModelSerializer): class CompleteFileLinkSerializer(serializers.ModelSerializer): file_content = serializers.CharField(required = False, allow_blank = True) - topic = TopicSerializer() + topic = TopicSerializer('get_subject') tags = TagSerializer(many = True) pendencies_resource = PendenciesSerializer(many = True) groups = StudentsGroupSerializer(many = True) students = UserBackupSerializer(many = True) + 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: + 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 = "files/file_" + str(time.time()) + os.path.splitext(data["file_content"])[1] + + os.rename(os.path.join(dst_path, path), os.path.join(settings.MEDIA_ROOT, new_name)) + + data["file_content"] = new_name + else: + path = files.extract(data["file_content"], settings.MEDIA_ROOT) + + return data + class Meta: model = FileLink fields = '__all__' + + def create(self, data): + topic = data['topic'] + + file_link = 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']) + + data["topic"] = topic + else: + data["topic"] = get_object_or_404(Topic, id = topic["id"]) + + + f = open(os.path.join(settings.MEDIA_ROOT, data["file_content"]), encoding="latin-1") + file = File(f) + + data["file_content"] = file + + file_link_data = data + + pendencies = file_link_data["pendencies_resource"] + del file_link_data["pendencies_resource"] + + file_link = FileLink() + file_link.name = file_link_data["name"] + file_link.brief_description = file_link_data["brief_description"] + file_link.show_window = file_link_data["show_window"] + file_link.all_students = file_link_data["all_students"] + file_link.visible = file_link_data["visible"] + file_link.order = file_link_data["order"] + file_link.topic = file_link_data["topic"] + file_link.file_content = file_link_data["file_content"] + + file_link.save() + + tags = data["tags"] + + for tag in tags: + if tag["id"] == "": + tag = Tag.objects.create(name = tag["name"]) + else: + tag = get_object_or_404(Tag, id = tag["id"]) + + file_link.tags.add(tag) + + resource = get_object_or_404(Resource, id = file_link.id) + + for pend in pendencies: + Pendencies.objects.create(resource = resource, **pend) + + return file_link diff --git a/goals/serializers.py b/goals/serializers.py index 2b02029..11435ec 100644 --- a/goals/serializers.py +++ b/goals/serializers.py @@ -102,13 +102,83 @@ class SimpleGoalSerializer(serializers.ModelSerializer): return instance class CompleteGoalSerializer(serializers.ModelSerializer): - topic = TopicSerializer() + topic = TopicSerializer('get_subject') tags = TagSerializer(many = True) item_goal = GoalItemSerializer(many = True) pendencies_resource = PendenciesSerializer(many = True) groups = StudentsGroupSerializer(many = True) students = UserBackupSerializer(many = True) + def get_subject(self, obj): + subject = self.context.get("subject", None) + + return subject + class Meta: model = Goals - fields = '__all__' \ No newline at end of file + fields = '__all__' + + def create(self, data): + topic = data['topic'] + + goals = 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']) + + data["topic"] = topic + else: + data["topic"] = get_object_or_404(Topic, id = topic["id"]) + + goals_data = data + + pendencies = goals_data["pendencies_resource"] + del goals_data["pendencies_resource"] + + goal_items = goals_data["item_goal"] + del goals_data["item_goal"] + + goals = Goals() + goals.name = goals_data["name"] + goals.brief_description = goals_data["brief_description"] + goals.show_window = goals_data["show_window"] + goals.all_students = goals_data["all_students"] + goals.visible = goals_data["visible"] + goals.order = goals_data["order"] + goals.topic = goals_data["topic"] + goals.presentation = goals_data["presentation"] + goals.limit_submission_date = goals_data["limit_submission_date"] + + goals.save() + + tags = data["tags"] + + for tag in tags: + if tag["id"] == "": + tag = Tag.objects.create(name = tag["name"]) + else: + tag = get_object_or_404(Tag, id = tag["id"]) + + goals.tags.add(tag) + + resource = get_object_or_404(Resource, id = goals.id) + + for item in goal_items: + GoalItem.objects.create(goal = goals, **item) + + for pend in pendencies: + Pendencies.objects.create(resource = resource, **pend) + + return goals \ No newline at end of file diff --git a/links/serializers.py b/links/serializers.py index 5e44b13..e96711e 100644 --- a/links/serializers.py +++ b/links/serializers.py @@ -89,12 +89,75 @@ class SimpleLinkSerializer(serializers.ModelSerializer): return instance class CompleteLinkSerializer(serializers.ModelSerializer): - topic = TopicSerializer() + topic = TopicSerializer('get_subject') tags = TagSerializer(many = True) pendencies_resource = PendenciesSerializer(many = True) groups = StudentsGroupSerializer(many = True) students = UserBackupSerializer(many = True) + def get_subject(self, obj): + subject = self.context.get("subject", None) + + return subject + class Meta: model = Link - fields = '__all__' \ No newline at end of file + fields = '__all__' + + def create(self, data): + topic = data['topic'] + + link = 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']) + + data["topic"] = topic + else: + data["topic"] = get_object_or_404(Topic, id = topic["id"]) + + link_data = data + + pendencies = link_data["pendencies_resource"] + del link_data["pendencies_resource"] + + link = Link() + link.name = link_data["name"] + link.brief_description = link_data["brief_description"] + link.show_window = link_data["show_window"] + link.all_students = link_data["all_students"] + link.visible = link_data["visible"] + link.order = link_data["order"] + link.topic = link_data["topic"] + link.link_url = link_data["link_url"] + + link.save() + + tags = data["tags"] + + for tag in tags: + if tag["id"] == "": + tag = Tag.objects.create(name = tag["name"]) + else: + tag = get_object_or_404(Tag, id = tag["id"]) + + link.tags.add(tag) + + resource = get_object_or_404(Resource, id = link.id) + + for pend in pendencies: + Pendencies.objects.create(resource = resource, **pend) + + return link \ No newline at end of file diff --git a/pdf_file/serializers.py b/pdf_file/serializers.py index 63bfcbd..1af10b8 100644 --- a/pdf_file/serializers.py +++ b/pdf_file/serializers.py @@ -123,12 +123,102 @@ class SimplePDFFileSerializer(serializers.ModelSerializer): class CompletePDFFileSerializer(serializers.ModelSerializer): file = serializers.CharField(required = False, allow_blank = True) - topic = TopicSerializer() + topic = TopicSerializer('get_subject') tags = TagSerializer(many = True) pendencies_resource = PendenciesSerializer(many = True) groups = StudentsGroupSerializer(many = True) students = UserBackupSerializer(many = True) + 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: + 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 = "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) + + return data + class Meta: model = PDFFile - fields = '__all__' \ No newline at end of file + 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']) + + data["topic"] = topic + else: + data["topic"] = get_object_or_404(Topic, id = topic["id"]) + + + f = open(os.path.join(settings.MEDIA_ROOT, data["file"]), encoding="latin-1") + file = File(f) + + data["file"] = file + + 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 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 \ No newline at end of file diff --git a/webpage/serializers.py b/webpage/serializers.py index 48614af..d9870c2 100644 --- a/webpage/serializers.py +++ b/webpage/serializers.py @@ -89,12 +89,75 @@ class SimpleWebpageSerializer(serializers.ModelSerializer): return instance class CompleteWebpageSerializer(serializers.ModelSerializer): - topic = TopicSerializer() + topic = TopicSerializer('get_subject') tags = TagSerializer(many = True) pendencies_resource = PendenciesSerializer(many = True) groups = StudentsGroupSerializer(many = True) students = UserBackupSerializer(many = True) + def get_subject(self, obj): + subject = self.context.get("subject", None) + + return subject + class Meta: model = Webpage - fields = '__all__' \ No newline at end of file + fields = '__all__' + + def create(self, data): + topic = data['topic'] + + webpage = 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']) + + data["topic"] = topic + else: + data["topic"] = get_object_or_404(Topic, id = topic["id"]) + + webpage_data = data + + pendencies = webpage_data["pendencies_resource"] + del webpage_data["pendencies_resource"] + + webpage = Webpage() + webpage.name = webpage_data["name"] + webpage.brief_description = webpage_data["brief_description"] + webpage.show_window = webpage_data["show_window"] + webpage.all_students = webpage_data["all_students"] + webpage.visible = webpage_data["visible"] + webpage.order = webpage_data["order"] + webpage.topic = webpage_data["topic"] + webpage.content = webpage_data["content"] + + webpage.save() + + tags = data["tags"] + + for tag in tags: + if tag["id"] == "": + tag = Tag.objects.create(name = tag["name"]) + else: + tag = get_object_or_404(Tag, id = tag["id"]) + + webpage.tags.add(tag) + + resource = get_object_or_404(Resource, id = webpage.id) + + for pend in pendencies: + Pendencies.objects.create(resource = resource, **pend) + + return webpage \ No newline at end of file diff --git a/youtube_video/serializers.py b/youtube_video/serializers.py index 9d8b905..ff34e38 100644 --- a/youtube_video/serializers.py +++ b/youtube_video/serializers.py @@ -90,12 +90,76 @@ class SimpleYTVideoSerializer(serializers.ModelSerializer): return instance class CompleteYTVideoSerializer(serializers.ModelSerializer): - topic = TopicSerializer() + topic = TopicSerializer('get_subject') tags = TagSerializer(many = True) pendencies_resource = PendenciesSerializer(many = True) groups = StudentsGroupSerializer(many = True) students = UserBackupSerializer(many = True) + def get_subject(self, obj): + subject = self.context.get("subject", None) + + return subject + class Meta: model = YTVideo - fields = '__all__' \ No newline at end of file + fields = '__all__' + + def create(self, data): + topic = data['topic'] + + ytvideo = 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']) + + data["topic"] = topic + else: + data["topic"] = get_object_or_404(Topic, id = topic["id"]) + + + ytvideo_data = data + + pendencies = ytvideo_data["pendencies_resource"] + del ytvideo_data["pendencies_resource"] + + ytvideo = YTVideo() + ytvideo.name = ytvideo_data["name"] + ytvideo.brief_description = ytvideo_data["brief_description"] + ytvideo.show_window = ytvideo_data["show_window"] + ytvideo.all_students = ytvideo_data["all_students"] + ytvideo.visible = ytvideo_data["visible"] + ytvideo.order = ytvideo_data["order"] + ytvideo.topic = ytvideo_data["topic"] + ytvideo.url = ytvideo_data["url"] + + ytvideo.save() + + tags = data["tags"] + + for tag in tags: + if tag["id"] == "": + tag = Tag.objects.create(name = tag["name"]) + else: + tag = get_object_or_404(Tag, id = tag["id"]) + + ytvideo.tags.add(tag) + + resource = get_object_or_404(Resource, id = ytvideo.id) + + for pend in pendencies: + Pendencies.objects.create(resource = resource, **pend) + + return ytvideo \ No newline at end of file -- libgit2 0.21.2