Programmation PYTHON - Zenk - Security - Repository

362
La bibliothèque standard
TROISIÈME PARTIE
def add_to_zip(zip, path):
"""Ajoute un fichier dans une archive zip."""
zip = ZipFile(zip, mode="a")
try:
if path not in zip.namelist():
zip.write(path, path)
return True
return False
finally:
zip.close()
def print_zip(zip):
"""Affiche le contenu d'un fichier zip."""
zip = ZipFile(zip)
try:
print 'Contenu de %s:\n' % zip.filename
zip.printdir()
print '%d fichier(s)' % len(zip.filelist)
finally:
zip.close()
if __name__ == '__main__':
if len(sys.argv) < 2:
print(__doc__ % {'prog': sys.argv[0]})
sys.exit(0)
zip = sys.argv[1]
if not is_zipfile(zip):
print('"%s" n\'est pas un fichier zip' % zip)
sys.exit(0)
if len(sys.argv) > 2:
if not add_to_zip(zip, sys.argv[2]):
print('Fichier avec le même nom déjà existant')
else:
print('Fichier ajouté')
else:
print_zip(zip)
Discussion
Lorsqu’aucun fichier à ajouter n’est fourni, le programme se contente de fournir la
liste des fichiers de l’archive.
Le module zipfile fournit, outre la classe ZipFile, une petite fonction utilitaire
is_zipfile() qui permet de tester un fichier pour savoir si c’est une archive zip.
L’information est lue dans les premiers octets du fichier.