doc: response time

- Document new feature for response times
- Refactor
pull/109/head
Tejasvi Nareddy 5 years ago
parent 4c3967fd6d
commit 038d531483

@ -37,7 +37,7 @@ optional arguments:
-h, --help show this help message and exit
--version Display version information and dependencies.
--verbose, -v, -d, --debug
Display extra debugging information.
Display extra debugging information and metrics.
--quiet, -q Disable debugging information (Default Option).
--tor, -t Make requests over TOR; increases runtime; requires
TOR to be installed and in system path.

@ -29,9 +29,12 @@ amount=0
# TODO: fix tumblr
# Extends FutureSession to add response time metric
# This is taken (almost) directly from here: https://github.com/ross/requests-futures#working-in-the-background
class ElapsedFuturesSession(FuturesSession):
"""
Extends FutureSession to add a response time metric to each request.
This is taken (almost) directly from here: https://github.com/ross/requests-futures#working-in-the-background
"""
def request(self, method, url, hooks={}, *args, **kwargs):
start = time()
@ -72,7 +75,7 @@ def print_error(err, errstr, var, verbose=False):
Fore.YELLOW + f" {err if verbose else var}")
def create_response_time(response_time, verbose):
def format_response_time(response_time, verbose):
return " [{} ms]".format(response_time) if verbose else ""
@ -80,7 +83,7 @@ def print_found(social_network, url, response_time, verbose=False):
print((Style.BRIGHT + Fore.WHITE + "[" +
Fore.GREEN + "+" +
Fore.WHITE + "]" +
create_response_time(response_time, verbose) +
format_response_time(response_time, verbose) +
Fore.GREEN + " {}:").format(social_network), url)
@ -88,7 +91,7 @@ def print_not_found(social_network, response_time, verbose=False):
print((Style.BRIGHT + Fore.WHITE + "[" +
Fore.RED + "-" +
Fore.WHITE + "]" +
create_response_time(response_time, verbose) +
format_response_time(response_time, verbose) +
Fore.GREEN + " {}:" +
Fore.YELLOW + " Not Found!").format(social_network))
@ -336,7 +339,7 @@ def main():
)
parser.add_argument("--verbose", "-v", "-d", "--debug",
action="store_true", dest="verbose", default=False,
help="Display extra debugging information."
help="Display extra debugging information and metrics."
)
parser.add_argument("--quiet", "-q",
action="store_false", dest="verbose",

Loading…
Cancel
Save