river_tao/OneForAll-mirror
1
0
Fork 0
mirror of https://github.com/shmilylty/OneForAll.git synced 2026-10-25 00:56:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化域名获取

Browse Source
This commit is contained in:
Jing Ling
2020-04-08 21:58:30 +08:00
parent 74f7077c51
commit dd7935b69c
1 changed files with 13 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
oneforall/common/utils.py
+8 -4
View File
@@ -124,15 +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(): if path.is_file() and path.exists():
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:
domain = Domain(line.strip()).match() line = line.lower().strip()
domain = Domain(line).match()
if domain: if domain:
domains.append(domain) domains.append(domain)
elif Domain(target).match():
domains = [target]
count = len(domains) count = len(domains)
if count == 0: if count == 0:
logger.log('FATAL', f'获取到{count}个域名') logger.log('FATAL', f'获取到{count}个域名')