diff --git a/common/records.py b/common/records.py index a6455b1..7f1ce95 100644 --- a/common/records.py +++ b/common/records.py @@ -1,12 +1,11 @@ import os -from sys import stdout from collections import OrderedDict -from contextlib import contextmanager from inspect import isclass -from .tablib import tablib from sqlalchemy import create_engine, exc, inspect, text +from .tablib import tablib + DATABASE_URL = os.environ.get('DATABASE_URL') diff --git a/common/utils.py b/common/utils.py index 323689a..74c03c4 100644 --- a/common/utils.py +++ b/common/utils.py @@ -832,4 +832,4 @@ def looks_like_ip(maybe_ip): if IP_RE.match(maybe_ip): return True except socket.error: - return False \ No newline at end of file + return False diff --git a/docs/collection_modules.md b/docs/collection_modules.md index d10aef6..18e1b2a 100644 --- a/docs/collection_modules.md +++ b/docs/collection_modules.md @@ -6,7 +6,7 @@ ```python enable_all_module = False # 不开启所有模块 -enable_partial_module = [('modules.search', 'ask'),('modules.search', 'baidu')] # 只使用ask和baidu搜索引擎收集子域 +enable_partial_module = [('modules.search', 'ask'), ('modules.search', 'baidu')] # 只使用ask和baidu搜索引擎收集子域 ``` 如果你指定使用某些模块使用代理可以在[api.py](../oneforall/config/api.py)中设置: diff --git a/modules/banner.py b/modules/banner.py index 83464bc..09c7195 100644 --- a/modules/banner.py +++ b/modules/banner.py @@ -332,7 +332,7 @@ class ParseException(Exception): class Token(object): - def __init__(self, type: TOKEN_TYPE, name: str = '', value: bool = False): + def __init__(self, type, name='', value=False): self.type = type self.name = name self.value = value