##############################################################################
# Parte do livro Introdução à Programação com Python
# Autor: Nilo Ney Coutinho Menezes
# Editora Novatec (c) 2010-2017
# Primeira edição - Novembro/2010 - ISBN 978-85-7522-250-8
# Primeira reimpressão - Outubro/2011
# Segunda reimpressão - Novembro/2012
# Terceira reimpressão - Agosto/2013
# Segunda edição - Junho/2014 - ISBN 978-85-7522-408-3
# Primeira reimpressão - Segunda edição - Maio/2015
# Segunda reimpressão - Segunda edição - Janeiro/2016
# Terceira reimpressão - Segunda edição - Junho/2016
# Quarta reimpressão - Segunda edição - Março/2017
# Terceira Edição - Janeiro/2019
#
# Site: http://python.nilo.pro.br/
#
# Arquivo: exercicios\capitulo 09\exercicio-09-33.py
##############################################################################
# Esta exercício pode ser realizado também com o módulo glob
# Consulte a documentação do Python para mais informações
import sys
import os
import os.path
# este módulo ajuda com a conversão de nomes de arquivos para links
# válidos em HTML
import urllib.request
if len(sys.argv) < 2:
print("Digite o nome do diretório para coletar os arquivos jpg e png!")
sys.exit(1)
diretório = sys.argv[1]
pagina = open("imagens.html", "w", encoding="utf-8")
pagina.write("""
<!DOCTYPE html>
<html lang="pt-BR">
<head>
<meta charset="utf-8">
<title>Imagens PNG e JPG</title>
</head>
<body>
""")
pagina.write("Imagens encontradas no diretório: %s" % diretório)
for entrada in os.listdir(diretório):
nome, extensão = os.path.splitext(entrada)
if extensão in [".jpg", ".png"]:
caminho_completo = os.path.join(diretório, entrada)
link = urllib.request.pathname2url(caminho_completo)
pagina.write("<p><a href='%s'>%s</a></p>" % (link, entrada))
pagina.write("""
</body>
</html>
""")
pagina.close()