Linted code iwht black tool
This commit is contained in:
@@ -7,13 +7,13 @@ from wordlists import get_wordlists
|
||||
def generate_server_error() -> tuple[str, str]:
|
||||
wl = get_wordlists()
|
||||
server_errors = wl.server_errors
|
||||
|
||||
|
||||
if not server_errors:
|
||||
return ("500 Internal Server Error", "text/html")
|
||||
|
||||
|
||||
server_type = random.choice(list(server_errors.keys()))
|
||||
server_config = server_errors[server_type]
|
||||
|
||||
|
||||
error_codes = {
|
||||
400: "Bad Request",
|
||||
401: "Unauthorized",
|
||||
@@ -21,45 +21,45 @@ def generate_server_error() -> tuple[str, str]:
|
||||
404: "Not Found",
|
||||
500: "Internal Server Error",
|
||||
502: "Bad Gateway",
|
||||
503: "Service Unavailable"
|
||||
503: "Service Unavailable",
|
||||
}
|
||||
|
||||
|
||||
code = random.choice(list(error_codes.keys()))
|
||||
message = error_codes[code]
|
||||
|
||||
template = server_config.get('template', '')
|
||||
version = random.choice(server_config.get('versions', ['1.0']))
|
||||
|
||||
html = template.replace('{code}', str(code))
|
||||
html = html.replace('{message}', message)
|
||||
html = html.replace('{version}', version)
|
||||
|
||||
if server_type == 'apache':
|
||||
os = random.choice(server_config.get('os', ['Ubuntu']))
|
||||
html = html.replace('{os}', os)
|
||||
html = html.replace('{host}', 'localhost')
|
||||
|
||||
|
||||
template = server_config.get("template", "")
|
||||
version = random.choice(server_config.get("versions", ["1.0"]))
|
||||
|
||||
html = template.replace("{code}", str(code))
|
||||
html = html.replace("{message}", message)
|
||||
html = html.replace("{version}", version)
|
||||
|
||||
if server_type == "apache":
|
||||
os = random.choice(server_config.get("os", ["Ubuntu"]))
|
||||
html = html.replace("{os}", os)
|
||||
html = html.replace("{host}", "localhost")
|
||||
|
||||
return (html, "text/html")
|
||||
|
||||
|
||||
def get_server_header(server_type: str = None) -> str:
|
||||
wl = get_wordlists()
|
||||
server_errors = wl.server_errors
|
||||
|
||||
|
||||
if not server_errors:
|
||||
return "nginx/1.18.0"
|
||||
|
||||
|
||||
if not server_type:
|
||||
server_type = random.choice(list(server_errors.keys()))
|
||||
|
||||
|
||||
server_config = server_errors.get(server_type, {})
|
||||
version = random.choice(server_config.get('versions', ['1.0']))
|
||||
|
||||
version = random.choice(server_config.get("versions", ["1.0"]))
|
||||
|
||||
server_headers = {
|
||||
'nginx': f"nginx/{version}",
|
||||
'apache': f"Apache/{version}",
|
||||
'iis': f"Microsoft-IIS/{version}",
|
||||
'tomcat': f"Apache-Coyote/1.1"
|
||||
"nginx": f"nginx/{version}",
|
||||
"apache": f"Apache/{version}",
|
||||
"iis": f"Microsoft-IIS/{version}",
|
||||
"tomcat": f"Apache-Coyote/1.1",
|
||||
}
|
||||
|
||||
|
||||
return server_headers.get(server_type, "nginx/1.18.0")
|
||||
|
||||
Reference in New Issue
Block a user