add number of results in the end of the process

pull/1355/head
Evelina Diamantopoulou 3 years ago
parent 548180749d
commit 62d77c3346

@ -6,6 +6,7 @@ results of queries.
from result import QueryStatus from result import QueryStatus
from colorama import Fore, Style from colorama import Fore, Style
globvar = 0
class QueryNotify: class QueryNotify:
"""Query Notify Object. """Query Notify Object.
@ -103,7 +104,6 @@ class QueryNotify:
""" """
return str(self.result) return str(self.result)
class QueryNotifyPrint(QueryNotify): class QueryNotifyPrint(QueryNotify):
"""Query Notify Print Object. """Query Notify Print Object.
@ -185,6 +185,20 @@ class QueryNotifyPrint(QueryNotify):
# return # 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): def update(self, result):
"""Notify Update. """Notify Update.
@ -206,6 +220,7 @@ class QueryNotifyPrint(QueryNotify):
# Output to the terminal is desired. # Output to the terminal is desired.
if result.status == QueryStatus.CLAIMED: if result.status == QueryStatus.CLAIMED:
self.countResults()
print(Style.BRIGHT + Fore.WHITE + "[" + print(Style.BRIGHT + Fore.WHITE + "[" +
Fore.GREEN + "+" + Fore.GREEN + "+" +
Fore.WHITE + "]" + Fore.WHITE + "]" +
@ -255,10 +270,18 @@ class QueryNotifyPrint(QueryNotify):
Will print the last line to the standard output. Will print the last line to the standard output.
Keyword Arguments: Keyword Arguments:
self -- This object. self -- This object.
message -- The last phrase. message -- The 2 last phrases.
Return Value: Return Value:
Nothing. Nothing.
""" """
numberOfResults = self.countResults() - 1
title = "Results:"
print(Style.BRIGHT + Fore.GREEN + "[" +
Fore.YELLOW + "*" +
Fore.GREEN + f"] {title}" +
Fore.WHITE + f" {numberOfResults}" )
title = "End" title = "End"

Loading…
Cancel
Save