From fc85da6355bd58b0013d77f702d1440f27764643 Mon Sep 17 00:00:00 2001 From: Chris Lamb Date: Tue, 27 Oct 2009 11:11:05 +0000 Subject: [PATCH] Avoid if/elif/elif.. with lookup table. Signed-off-by: Chris Lamb --- daklib/utils.py | 40 ++++++++++++++++++++++------------------ 1 file changed, 22 insertions(+), 18 deletions(-) diff --git a/daklib/utils.py b/daklib/utils.py index f0df5984..cfee2c7e 100755 --- a/daklib/utils.py +++ b/daklib/utils.py @@ -365,30 +365,34 @@ def check_dsc_files(dsc_filename, dsc=None, dsc_files=None): # announced has = defaultdict(lambda: 0) + ftype_lookup = ( + (r'orig.tar.gz', ('orig_tar_gz', 'orig_tar')), + (r'diff.gz', ('debian_diff',)), + (r'tar.gz', ('native_tar_gz', 'native_tar')), + (r'debian\.tar\.(gz|bz2|lzma)', ('debian_tar',)), + (r'orig\.tar\.(gz|bz2|lzma)', ('orig_tar',)), + (r'tar\.(gz|bz2|lzma)', ('native_tar',)), + (r'orig-.+\.tar\.(gz|bz2|lzma)', ('more_orig_tar',)), + ) + for f in dsc_files.keys(): m = re_issource.match(f) if not m: rejmsg.append("%s: %s in Files field not recognised as source." % (dsc_filename, f)) continue - ftype = m.group(3) - if ftype == "orig.tar.gz": - has['orig_tar_gz'] += 1 - has['orig_tar'] += 1 - elif ftype == "diff.gz": - has['debian_diff'] += 1 - elif ftype == "tar.gz": - has['native_tar_gz'] += 1 - has['native_tar'] += 1 - elif re.match(r"debian\.tar\.(gz|bz2|lzma)", ftype): - has['debian_tar'] += 1 - elif re.match(r"orig\.tar\.(gz|bz2|lzma)", ftype): - has['orig_tar'] += 1 - elif re.match(r"tar\.(gz|bz2|lzma)", ftype): - has['native_tar'] += 1 - elif re.match(r"orig-.+\.tar\.(gz|bz2|lzma)", ftype): - has['more_orig_tar'] += 1 - else: + + # Populate 'has' dictionary by resolving keys in lookup table + matched = False + for regex, keys in ftype_lookup: + if re.match(regex, m.group(3)): + matched = True + for key in keys: + has[key] += 1 + break + + # File does not match anything in lookup table; reject + if not matched: reject("%s: unexpected source file '%s'" % (dsc_filename, f)) # Check for multiple files -- 2.39.2