river_tao/OneForAll-mirror
1
0
Fork 0
mirror of https://github.com/shmilylty/OneForAll.git synced 2026-08-26 12:57:50 +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
+13 -9
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):
path = Path(target) target = target.lower().strip()
if path.is_file(): domain = Domain(target).match()
with open(target, encoding='utf-8', errors='ignore') as file: if domain:
for line in file: domains.append(domain)
domain = Domain(line.strip()).match() else:
if domain: path = Path(target)
domains.append(domain) if path.is_file() and path.exists():
elif Domain(target).match(): with open(target, encoding='utf-8', errors='ignore') as file:
domains = [target] for line in file:
line = line.lower().strip()
domain = Domain(line).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}个域名')