Python re.search Anomalie

Question

Ich habe eine Routine, die durch ein Verzeichnis von Dateien sucht und extrahiert eine Kundennummer aus dem Dateinamen:

import os 
import re 

suffix= '.csv' 

# For each file in input folder, extract customer number 
input_list = os.listdir(path_in) 
for input_file in input_list: 
     fileInput = os.path.join(path_in,input_file) 
     customer_ID = re.search('custID_(.+?)'+suffix,fileInput).group(1) 
     print(customer_ID) 

Mit suffix='.csv' und einem Ordner voller csv-Dateien:

avg_hrly_custID_8147611.csv, avg_hrly_custID_8147612.csv, avg_hrly_custID_8147613.csv ...

ich die erwartete Ausgabe:

8147611, 8147612, 8147613...

ABER, mit suffix = '.png' und eine Falte er von .png Bilddateien ,:

yearly_average_plot_custID_8147611.png, yearly_average_plot_custID_8147612.png, yearly_average_plot_custID_8147613.png ...

ich diesen Fehler:

AttributeError: 'NoneType' object has no attribute 'group'

Warum wird es für die Bilddateien nicht?

Answer 1

@BrenBarn erkannte die Ursache des Problems. Der Regex ist fehlgeschlagen, weil im Verzeichnis ein Unterverzeichnis vorhanden war, dessen Name nicht übereinstimmt. Ich habe es von try....except

import os 
import re 

suffix= '.png' 

# For each file in input folder, extract customer number 
input_list = os.listdir(path_in) 
for input_file in input_list: 
     fileInput = os.path.join(path_in,input_file) 
     try: 
      customer_ID = re.search('custID_(.+?)'+suffix,fileInput).group(1) 
      print(customer_ID) 
     except: 
      pass 

Einführung gelöst