Correção no build danfe

pytrustnfe/pdf/__init__.py

@@ -2,7 +2,7 @@
 # © 2016 Danimar Ribeiro, Trustcode
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
 
-from super_table import PdfTable, PdfCell
+# from super_table import PdfTable, PdfCell
 from pytrustnfe import xml
 
 inch = 28.34
@@ -13,6 +13,7 @@ class Danfe(object):
     objeto = None
 
     def __init__(self, objetoNFe):
+        if objetoNFe:
             self.objeto = xml.sanitize_response(objetoNFe)[1]
             self.NFe = self.objeto.getchildren()[2]
             self.infNFe = self.NFe.getchildren()[0]
@@ -21,9 +22,12 @@ class Danfe(object):
             self.destinatario = self.infNFe.getchildren()[2]
 
     def nfe(self):
-        danfe = PdfTable(columns=12)
+        # danfe = PdfTable(columns=12)
         pass
 
-
     def gerar(self):
         pass
+
+if __name__ == '__main__':
+    pass
+    # import ipdb; ipdb.set_trace()