diff --git a/oneforall/common/utils.py b/oneforall/common/utils.py index 1a142b6..1cb1c1b 100644 --- a/oneforall/common/utils.py +++ b/oneforall/common/utils.py @@ -124,19 +124,19 @@ def get_domains(target): elif isinstance(target, list): domains = target elif isinstance(target, str): - target = target.lower().strip() - domain = Domain(target).match() - if domain: - domains.append(domain) + path = Path(target) + if path.exists() and path.is_file(): + with open(target, encoding='utf-8', errors='ignore') as file: + for line in file: + line = line.lower().strip() + domain = Domain(line).match() + if domain: + domains.append(domain) else: - path = Path(target) - if path.is_file() and path.exists(): - with open(target, encoding='utf-8', errors='ignore') as file: - for line in file: - line = line.lower().strip() - domain = Domain(line).match() - if domain: - domains.append(domain) + target = target.lower().strip() + domain = Domain(target).match() + if domain: + domains.append(domain) count = len(domains) if count == 0: logger.log('FATAL', f'获取到{count}个域名')