add number of results in the end of the process

This commit is contained in:
Evelina Diamantopoulou 2022-06-08 17:25:17 +03:00
parent 548180749d
commit 62d77c3346

View file

@ -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)