river_tao/OneForAll-mirror
1
0
Fork 0
mirror of https://github.com/shmilylty/OneForAll.git synced 2026-08-26 21:07:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

修复传入包含批量主域的文件路径无效的问题

Browse Source
This commit is contained in:
Jing Ling
2020-04-11 00:18:35 +08:00
parent 3b3e5551e2
commit 4a7fab3822
1 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
oneforall/common/utils.py
+6 -6
View File
@@ -124,19 +124,19 @@ def get_domains(target):
elif isinstance(target, list): elif isinstance(target, list):
domains = target domains = target
elif isinstance(target, str): elif isinstance(target, str):
target = target.lower().strip()
domain = Domain(target).match()
if domain:
domains.append(domain)
else:
path = Path(target) path = Path(target)
if path.is_file() and path.exists(): if path.exists() and path.is_file():
with open(target, encoding='utf-8', errors='ignore') as file: with open(target, encoding='utf-8', errors='ignore') as file:
for line in file: for line in file:
line = line.lower().strip() line = line.lower().strip()
domain = Domain(line).match() domain = Domain(line).match()
if domain: if domain:
domains.append(domain) domains.append(domain)
else:
target = target.lower().strip()
domain = Domain(target).match()
if domain:
domains.append(domain)
count = len(domains) count = len(domains)
if count == 0: if count == 0:
logger.log('FATAL', f'获取到{count}个域名') logger.log('FATAL', f'获取到{count}个域名')