Dotaz: Jak na shellovou expanzi v Pythonu ?

[amnesiac]

Pro expanzi znaků * a ? jsem našel fci glob v modulu glob.

Pro expanzi znaku ~ a proměnných prostředí fce expanduser a expandvars v modulu os.path .

Jak ale na závorkovou {} expanzi ? Potřeboval bych něco jako:

for objednavka in glob('obj*.{edt,inh}'):
   zpracuj_objednavku(objednavka)

Rozdělovat to na tolik cyklů for, kolik je prvků v závorkách se mi fakt nechce.

Předem dík.

Odpovědi

[Jan Martinek]

Zkus to prohnat regulárním výrazem:

#!/usr/bin/env python
import os, re

filter = re.compile('obj.*(edt|inh)')
for file in os.listdir('.'):
  if not filter.match(file):
    continue
  print file

[Jan Martinek]

A asi by bylo lepší, kdyby to nerozlišovalo velká/malá, takže

filter = re.compile('obj.*(edt|inh)', re.IGNORECASE)

(mám dojem, že se tu nedávno něco podobného řešilo v bashi ...)

[amnesiac]

Dík za tu kombinaci os.listdir a re.match, to mě nenapadlo.

Podle zadání bude regulární výraz nakonec vypadat asi takto:
'obj.*\.(edt|inh)$'