diff --git a/sherlock/notify.py b/sherlock/notify.py index 17eafb8c..a1afb9c7 100644 --- a/sherlock/notify.py +++ b/sherlock/notify.py @@ -6,6 +6,7 @@ results of queries. from result import QueryStatus from colorama import Fore, Style +globvar = 0 class QueryNotify: """Query Notify Object. @@ -52,7 +53,7 @@ class QueryNotify: Nothing. """ - # return + # return def update(self, result): """Notify Update. @@ -101,8 +102,7 @@ class QueryNotify: Return Value: Nicely formatted string to get information about this object. """ - return str(self.result) - + return str(self.result) class QueryNotifyPrint(QueryNotify): """Query Notify Print Object. @@ -132,7 +132,7 @@ class QueryNotifyPrint(QueryNotify): self.print_all = print_all return - + def start(self, message): """Notify Start. @@ -185,6 +185,20 @@ class QueryNotifyPrint(QueryNotify): # return + def countResults(self): + """Count the number of results. Every time the fuction is called, + the number of results is increasing. + + Keyword Arguments: + self -- This object. + + Return Value: + The number of results by the time we call the function. + """ + global globvar + globvar += 1 + return globvar + def update(self, result): """Notify Update. @@ -206,6 +220,7 @@ class QueryNotifyPrint(QueryNotify): # Output to the terminal is desired. if result.status == QueryStatus.CLAIMED: + self.countResults() print(Style.BRIGHT + Fore.WHITE + "[" + Fore.GREEN + "+" + Fore.WHITE + "]" + @@ -213,7 +228,7 @@ class QueryNotifyPrint(QueryNotify): Fore.GREEN + f" {self.result.site_name}: " + Style.RESET_ALL + - f"{self.result.site_url_user}") + f"{self.result.site_url_user}") elif result.status == QueryStatus.AVAILABLE: if self.print_all: @@ -255,11 +270,19 @@ class QueryNotifyPrint(QueryNotify): Will print the last line to the standard output. Keyword Arguments: self -- This object. - message -- The last phrase. + message -- The 2 last phrases. Return Value: Nothing. """ + numberOfResults = self.countResults() - 1 + + title = "Results:" + print(Style.BRIGHT + Fore.GREEN + "[" + + Fore.YELLOW + "*" + + Fore.GREEN + f"] {title}" + + Fore.WHITE + f" {numberOfResults}" ) + title = "End" print('\r') # An empty line between last line of main output and last line(more clear output) @@ -282,4 +305,4 @@ class QueryNotifyPrint(QueryNotify): Return Value: Nicely formatted string to get information about this object. """ - return str(self.result) + return str(self.result) \ No newline at end of file