From 088e4c92140339be5cd6ddbbceed65658bcb4ae4 Mon Sep 17 00:00:00 2001
From: Eduardo Santos <eduardo@gnumasters.com>
Date: Tue, 19 Aug 2014 12:47:15 -0300
Subject: [PATCH] Altera criação para métodos do objeto

---
 wscacicneo/model/orgao.py | 42 +++++++++++++++++++++++++++++++++++++++---
 wscacicneo/views.py       |  3 +--
 2 files changed, 40 insertions(+), 5 deletions(-)

diff --git a/wscacicneo/model/orgao.py b/wscacicneo/model/orgao.py
index 28e3253..beecb4f 100644
--- a/wscacicneo/model/orgao.py
+++ b/wscacicneo/model/orgao.py
@@ -2,12 +2,15 @@
 # -*- coding: utf-8 -*-
 __author__ = 'eduardo'
 
+from requests.exceptions import HTTPError
 from wscacicneo import WSCacicNeo
 from liblightbase.lbbase.struct import Base, BaseMetadata
 from liblightbase.lbbase.lbstruct.group import *
 from liblightbase.lbbase.lbstruct.field import *
 from liblightbase.lbbase.content import Content
 from liblightbase.lbrest.base import BaseREST
+from liblightbase.lbrest.document import DocumentREST
+from liblightbase.lbutils import conv
 
 class OrgaoBase(WSCacicNeo):
     """
@@ -19,6 +22,8 @@ class OrgaoBase(WSCacicNeo):
         """
         WSCacicNeo.__init__(self)
         self.baserest = BaseREST(rest_url=self.rest_url, response_object=True)
+        self.documentrest = DocumentREST(rest_url=self.rest_url,
+                base=self.lbbase, response_object=True)
 
     @property
     def lbbase(self):
@@ -145,12 +150,43 @@ class OrgaoBase(WSCacicNeo):
         else:
             raise IOError('Error excluding base from LB')
 
-OrgaoMetaClass = OrgaoBase().metaclass
+orgao_base = OrgaoBase()
 
-
-class Orgao(OrgaoMetaClass):
+class Orgao(orgao_base.metaclass):
     """
     Classe genérica de órgãos
     """
     def __init__(self, **args):
         super(Orgao, self).__init__(**args)
+
+    def orgao_to_dict(self):
+        """
+        Convert status object to Python dict
+        :return:
+        """
+
+        return conv.document2dict(orgao_base.lbbase, self)
+
+    def orgao_to_json(self):
+        """
+        Convert object to json
+        :return:
+        """
+
+        return conv.document2json(orgao_base.lbbase, self)
+
+    def create_orgao(self):
+        """
+        Insert document on base
+
+        :return: Document creation ID
+        """
+
+        document = self.orgao_to_json()
+        try:
+            result = orgao_base.documentrest.create(document)
+        except HTTPError, err:
+            log.error(err.strerror)
+            return None
+
+        return result
diff --git a/wscacicneo/views.py b/wscacicneo/views.py
index 1bb241b..4c57141 100644
--- a/wscacicneo/views.py
+++ b/wscacicneo/views.py
@@ -184,8 +184,7 @@ def post_orgao(request):
         telefone = doc['telefone']
     )
 
-    document = conv.document2json(orgaobase, orgao_obj)
-    id_doc = DocumentREST(rest_url, orgaobase).create(document)
+    id_doc = orgao_obj.create_orgao() 
     print(id_doc)
 
     return Response(str(id_doc))
--
libgit2 0.21.2