Merge pull request #1650 from orionkendrick/master

Add --browse functionallity
pull/601/head
Siddharth Dushantha 2 years ago committed by GitHub
commit 9ea08b2c73
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -5,6 +5,7 @@ results of queries.
""" """
from result import QueryStatus from result import QueryStatus
from colorama import Fore, Style from colorama import Fore, Style
import webbrowser
globvar = 0 # global variable to count the number of results. globvar = 0 # global variable to count the number of results.
class QueryNotify: class QueryNotify:
@ -109,7 +110,7 @@ class QueryNotifyPrint(QueryNotify):
Query notify class that prints results. Query notify class that prints results.
""" """
def __init__(self, result=None, verbose=False, print_all=False): def __init__(self, result=None, verbose=False, print_all=False, browse=False):
"""Create Query Notify Print Object. """Create Query Notify Print Object.
Contains information about a specific method of notifying the results Contains information about a specific method of notifying the results
@ -121,6 +122,7 @@ class QueryNotifyPrint(QueryNotify):
results for this query. results for this query.
verbose -- Boolean indicating whether to give verbose output. verbose -- Boolean indicating whether to give verbose output.
print_all -- Boolean indicating whether to only print all sites, including not found. print_all -- Boolean indicating whether to only print all sites, including not found.
browse -- Boolean indicating whether to open found sites in a web browser.
Return Value: Return Value:
Nothing. Nothing.
@ -129,6 +131,7 @@ class QueryNotifyPrint(QueryNotify):
super().__init__(result) super().__init__(result)
self.verbose = verbose self.verbose = verbose
self.print_all = print_all self.print_all = print_all
self.browse = browse
return return
@ -227,7 +230,9 @@ class QueryNotifyPrint(QueryNotify):
Fore.GREEN + Fore.GREEN +
f" {self.result.site_name}: " + f" {self.result.site_name}: " +
Style.RESET_ALL + Style.RESET_ALL +
f"{self.result.site_url_user}") f"{self.result.site_url_user}")
if self.browse:
webbrowser.open(self.result.site_url_user,2)
elif result.status == QueryStatus.AVAILABLE: elif result.status == QueryStatus.AVAILABLE:
if self.print_all: if self.print_all:

@ -664,7 +664,8 @@ def main():
# Create notify object for query results. # Create notify object for query results.
query_notify = QueryNotifyPrint(result=None, query_notify = QueryNotifyPrint(result=None,
verbose=args.verbose, verbose=args.verbose,
print_all=args.print_all) print_all=args.print_all,
browse=args.browse)
# Run report on all specified users. # Run report on all specified users.

Loading…
Cancel
Save