空のシーケンスで関数が失敗する

Question

このサイトで恥知らずに見つけたPythonコードに基づいて、ディレクトリ内の最新ファイルの日付を検索しようとしています。

def newest_file_in_tree(rootfnewer, extension=".avi"): 
    return max(
     (os.path.join(dirname, filename) 
     for dirname, dirnames, filenames in os.walk(rootfnewer) 
     for filename in filenames 
     if filename.endswith(extension)), 
     key=lambda fn: os.stat(fn).st_mtime) 

これは機能しますが、空のディレクトリでは失敗します。

key=lambda fn: os.stat(fn).st_mtime) 
ValueError: max() arg is an empty sequence 

は、私はPythonの初心者だということを認めて、そして最後の行は、私の給与等級を超えた方法です。

どのようにエラーをトラップして機能がクラッシュするのを防ぐことができますか？

Answer 1

あなたは、あなたが最新のものを選んで、そこからファイルの非空のリストを持っているときにのみmaxが呼び出されたことを確認するために、二つの別々の行にそのものすごいニシキヘビ1つのライナーを分離することができます

def newest_file_in_tree(rootfnewer, extension=".avi"): 
    files = [os.path.join(dirname, filename) 
     for dirname, dirnames, filenames in os.walk(rootfnewer) 
     for filename in filenames 
     if filename.endswith(extension)] 

    if files: 
     return max(files, key=lambda fn: os.stat(fn).st_mtime) 

    return None 

@ hiroprotagonistのソリューションほど美しいものではありませんが、読みやすくなっています。

Answer 2

maxはdefaultキーワードがあります。

m = max([], key=lambda x: abs(x), default='emtpy...') 
print(m) # 'emtpy...' 

反復可能が空の場合はdefaultが返されます。 defaultがなければ、ValueErrorとなります。

---

あなたが行くための別の方法は、ファイルを一覧表示するためにglobを使用することです：

import os 
from glob import glob 

def newest_file_in_tree(rootdir, ext='.avi'): 
    glob_wildcard = os.path.join(rootdir, '**/*{}'.format(ext)) 
    glob_gen = glob(glob_wildcard, recursive=True) 
    return max(glob_gen, key=lambda fn: os.stat(fn).st_mtime, default='?') 

あるいはより簡潔なpathlibと：

from pathlib import Path 

def newest_file_in_tree(rootpath, ext='.avi'): 
    return max(Path(rootpath).glob('**/*{}'.format(ext)), 
       key=lambda pth: pth.stat().st_mtime, default='?')