优化域名获取

This commit is contained in:
Jing Ling
2020-04-08 21:58:30 +08:00
parent 74f7077c51
commit dd7935b69c
+11 -7
View File
@@ -124,15 +124,19 @@ def get_domains(target):
elif isinstance(target, list):
domains = target
elif isinstance(target, str):
path = Path(target)
if path.is_file():
with open(target, encoding='utf-8', errors='ignore') as file:
for line in file:
domain = Domain(line.strip()).match()
target = target.lower().strip()
domain = Domain(target).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)
elif Domain(target).match():
domains = [target]
count = len(domains)
if count == 0:
logger.log('FATAL', f'获取到{count}个域名')