Correção na verificação do HEADER retorno

caputchinefrobles · Aug 6, 2019 · 1067fc8 · 1067fc8
1 parent 1ed87af
commit 1067fc8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -8,8 +8,9 @@ Script que pesquisa no Diário Oficial da União termos de interesse e notifica
 - smtplib
 - urllib3
 - mailer
+- chardet (pdfminer)
 
-`pip install pdfminer.six urllib3 mailer`
+`pip install pdfminer.six urllib3 mailer chardet`
 
 ## Utilização
 

diff --git a/classes.py b/classes.py
@@ -57,7 +57,7 @@ def processar(self, jornal, pagina_inicio=1, pagina_fim=1100, extra=False):
            print(full_url)
            response = http.request('GET', full_url, headers=header)
 
-           if 'text/html' not in response.headers['Content-Type'] and response.headers['Content-Encoding'] == 'gzip':
+           if response.headers['Content-Type'] == 'application/pdf':
                buff = response.data
                arquivo = io.BytesIO(buff)
                texto = extrair_texto(arquivo).upper()