Enum PE section names for large collection of malware

pe-section-xtract.py

import pefile
import sys
import os

dir = '/tmp/mlwr'

for dirpath, dirnames, filenames in os.walk(dir):
  for filename in filenames:
    with open(os.path.join(dir,dirpath,filename), 'rb') as current:
      xtract = current.read(2)
      conv = xtract.decode('ascii',errors='ignore')
      if conv == 'MZ':
        try:
          pe = pefile.PE(os.path.join(dir,dirpath,filename))
          for section in pe.sections:
            asciisection = section.Name.decode('ascii',errors='ignore')
            print(asciisection)
        except pefile.PEFormatError as err:
          print("{} in file '{}'".format(err, filename))
          continue