From 137b2386a45ec48d00c543db56e03dcac6e668b3 Mon Sep 17 00:00:00 2001
From: Jamil Atta Junior <atta.jamil@gmail.com>
Date: Mon, 2 May 2016 18:06:35 -0300
Subject: [PATCH] =?UTF-8?q?Ajuste=20na=20entidade=20section=20e=20adiciona?=
 =?UTF-8?q?do=20m=C3=A9todos=20relacionados=20a=20tradu=C3=A7=C3=B5es=20em?=
 =?UTF-8?q?=20v=C3=A1rios=20atributos.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 opac_schema/v1/models.py | 57 +++++++++++++++++++++++++++++-----------
 1 file changed, 41 insertions(+), 16 deletions(-)

diff --git a/opac_schema/v1/models.py b/opac_schema/v1/models.py
index 3fd810b..9ce5e0f 100644
--- a/opac_schema/v1/models.py
+++ b/opac_schema/v1/models.py
@@ -47,18 +47,6 @@ def __unicode__(self):
         return self.code
 
 
-class Section(EmbeddedDocument):
-    order = IntField()
-    subjects = EmbeddedDocumentListField('Subject')
-
-    meta = {
-        'collection': 'sections'
-    }
-
-    def __unicode__(self):
-        return '<Section: %s>' % self.order
-
-
 class Timeline(EmbeddedDocument):
     since = DateTimeField()
     reason = StringField()
@@ -115,7 +103,7 @@ class LastIssue(EmbeddedDocument):
     label = StringField()
     start_month = IntField()
     end_month = IntField()
-    sections = EmbeddedDocumentListField(Section)
+    sections = EmbeddedDocumentListField('TranslatedSection')
     cover_url = StringField()
     iid = StringField()
     bibliographic_legend = StringField()
@@ -128,12 +116,12 @@ def __unicode__(self):
         return self.label
 
 
-class Subject(EmbeddedDocument):
+class TranslatedSection(EmbeddedDocument):
     name = StringField()
     language = StringField()
 
     meta = {
-        'collection': 'subjects'
+        'collection': 'translated_section'
     }
 
     def __unicode__(self):
@@ -241,6 +229,15 @@ class Journal(Document):
     def __unicode__(self):
         return self.acronym or 'undefined acronym'
 
+    def get_mission_by_lang(self, lang):
+        """
+        Retorna a missão por idioma, caso não encontra retorna ``None``.
+        """
+
+        for mission in self.mission:
+            if mission.language == lang:
+                return mission.description
+
 
 class Issue(Document):
 
@@ -281,6 +278,7 @@ def __unicode__(self):
 
 
 class Article(Document):
+
     _id = StringField(max_length=32, primary_key=True, required=True, unique=True)
     aid = StringField(max_length=32, required=True, unique=True)
 
@@ -290,7 +288,7 @@ class Article(Document):
     title = StringField()
     translated_titles = EmbeddedDocumentListField(TranslatedTitle)
     section = StringField()
-    sections = EmbeddedDocumentListField(Section)
+    sections = EmbeddedDocumentListField(TranslatedSection)
     is_aop = BooleanField()
     order = IntField()
     doi = StringField()
@@ -316,3 +314,30 @@ class Article(Document):
 
     def __unicode__(self):
         return self.title
+
+    def get_title_by_lang(self, lang):
+        """
+        Retorna o título do artigo por idioma, caso não encontre o idioma
+        retorna o título original, caso não tenha título original retorna ``None``.
+
+        O parâmetro ``lang`` é o acrônimo do idioma, ex.: en, es, pt.
+        """
+
+        for title in self.translated_titles:
+            if title.language == lang:
+                return title.name
+
+        return self.title
+
+    def get_section_by_lang(self, lang):
+        """
+        Retorna a seção por idioma, caso não encontre retorna o atributo:
+        ``article.section``, caso o artigo não tenha o atributo ``section``,
+        retorna ``None``.
+        """
+
+        for section in self.sections:
+            if section.language == lang:
+                return section.name
+
+        return self.section