From 52823380412ae8f2b28b059fa1f4871d4ce1147a Mon Sep 17 00:00:00 2001 From: Elliott Sales de Andrade Date: May 07 2019 03:56:23 +0000 Subject: Make known forge URL check more extensible. --- diff --git a/go2rpm/__main__.py b/go2rpm/__main__.py index fe0754d..debda41 100644 --- a/go2rpm/__main__.py +++ b/go2rpm/__main__.py @@ -422,10 +422,13 @@ def main(): goipath = re.sub(r"^http(s?)://", r"", args.goipath) goipath = goipath.strip('/') - if (not re.search(r"^(github.com|gitlab.com|bitbucket.org)", goipath) - and args.forge is None): + known_forge = ('github.com', 'gitlab.com', 'bitbucket.org') + known_forge_re = (r'^(' + + r'|'.join(re.escape(url) for url in known_forge) + + r')') + if not re.search(known_forge_re, goipath) and args.forge is None: parser.error( - "forge URL is required for import path other than github, gitlab and bitbucket" + f"forge URL is required for import path other than {known_forge}" ) if args.forge is None: