def tests_quiet_colored_logo(self):
"""
Test if the logo is not printed when quiet is activated.
"""
PyFunceble.CONFIGURATION["quiet"] = True
Core.colorify_logo()
expected = ""
actual = sys.stdout.getvalue()
self.assertEqual(expected, actual)
del PyFunceble.CONFIGURATION["quiet"]
def tests_colored_logo_green(self):
"""
Test if the logo is green colored.
"""
PyFunceble.CONFIGURATION["counter"]["percentage"]["up"] = 51
Core.colorify_logo()
expected = self.logo_green + "\n"
actual = sys.stdout.getvalue()
self.assertEqual(expected, actual)
PyFunceble.CONFIGURATION["counter"]["percentage"]["up"] = 0
def tests_colored_logo_red(self):
"""
Test if the logo is red colored.
"""
PyFunceble.INTERN["counter"]["percentage"]["up"] = 1
Core.colorify_logo()
expected = self.logo_red + "\n"
actual = sys.stdout.getvalue()
self.assertEqual(expected, actual)
PyFunceble.INTERN["counter"]["percentage"]["up"] = 0
def _command_line(): # pragma: no cover pylint: disable=too-many-branches,too-many-statements
"""
Provide the command line interface.
"""
if __name__ == "PyFunceble":
# We initiate the end of the coloration at the end of each line.
initiate(autoreset=True)
# We load the configuration and the directory structure.
load_config(True)
try:
# The following handle the command line argument.
try:
PARSER = argparse.ArgumentParser(
epilog="Crafted with %s by %s" % (
Fore.RED + "♥" + Fore.RESET,
Style.BRIGHT + Fore.CYAN +
"Nissar Chababy (Funilrys) " + Style.RESET_ALL +
"with the help of " + Style.BRIGHT + Fore.GREEN +
"https://pyfunceble.rtfd.io/en/master/contributors.html "
+ Style.RESET_ALL + "&& " + Style.BRIGHT + Fore.GREEN +
"https://pyfunceble.rtfd.io/en/master/special-thanks.html",
),
add_help=False,
)
CURRENT_VALUE_FORMAT = (Fore.YELLOW + Style.BRIGHT +
"Configured value: " + Fore.BLUE)
PARSER.add_argument(
"-ad",
"--adblock",
action="store_true",
help="Switch the decoding of the adblock format. %s" %
(CURRENT_VALUE_FORMAT + repr(CONFIGURATION["adblock"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"-a",
"--all",
action="store_false",
help="Output all available information on the screen. %s" %
(CURRENT_VALUE_FORMAT + repr(CONFIGURATION["less"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
""
"-c",
"--auto-continue",
"--continue",
action="store_true",
help="Switch the value of the auto continue mode. %s" %
(CURRENT_VALUE_FORMAT +
repr(CONFIGURATION["auto_continue"]) + Style.RESET_ALL),
)
PARSER.add_argument(
"--autosave-minutes",
type=int,
help="Update the minimum of minutes before we start "
"committing to upstream under Travis CI. %s" %
(CURRENT_VALUE_FORMAT +
repr(CONFIGURATION["travis_autosave_minutes"]) +
Style.RESET_ALL),
)
PARSER.add_argument("--clean",
action="store_true",
help="Clean all files under output.")
PARSER.add_argument(
"--clean-all",
action="store_true",
help=
"Clean all files under output and all file generated by PyFunceble.",
)
PARSER.add_argument(
"--cmd",
type=str,
help="Pass a command to run before each commit "
"(except the final one) under the Travis mode. %s" %
(CURRENT_VALUE_FORMAT +
repr(CONFIGURATION["command_before_end"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"--cmd-before-end",
type=str,
help="Pass a command to run before the results "
"(final) commit under the Travis mode. %s" %
(CURRENT_VALUE_FORMAT +
repr(CONFIGURATION["command_before_end"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"--commit-autosave-message",
type=str,
help="Replace the default autosave commit message. %s" %
(CURRENT_VALUE_FORMAT +
repr(CONFIGURATION["travis_autosave_commit"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"--commit-results-message",
type=str,
help=
"Replace the default results (final) commit message. %s" %
(CURRENT_VALUE_FORMAT +
repr(CONFIGURATION["travis_autosave_final_commit"]) +
Style.RESET_ALL),
)
PARSER.add_argument("-d",
"--domain",
type=str,
help="Set and test the given domain.")
PARSER.add_argument(
"-db",
"--database",
action="store_true",
help="Switch the value of the usage of a database to store "
"inactive domains of the currently tested list. %s" %
(CURRENT_VALUE_FORMAT + repr(
CONFIGURATION["inactive_database"]) + Style.RESET_ALL),
)
PARSER.add_argument(
"-dbr",
"--days-between-db-retest",
type=int,
help=
"Set the numbers of days between each retest of domains present "
"into inactive-db.json. %s" %
(CURRENT_VALUE_FORMAT +
repr(CONFIGURATION["days_between_db_retest"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"--debug",
action="store_true",
help="Switch the value of the debug mode. %s" %
(CURRENT_VALUE_FORMAT + repr(CONFIGURATION["debug"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"--directory-structure",
action="store_true",
help=
"Generate the directory and files that are needed and which does "
"not exist in the current directory.",
)
PARSER.add_argument(
"-ex",
"--execution",
action="store_true",
help=
"Switch the default value of the execution time showing. %s"
% (CURRENT_VALUE_FORMAT +
repr(CONFIGURATION["show_execution_time"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"-f",
"--file",
type=str,
help="Read the given file and test all domains inside it. "
"If a URL is given we download and test the content of the given URL.", # pylint: disable=line-too-long
)
PARSER.add_argument("--filter",
type=str,
help="Domain to filter (regex).")
PARSER.add_argument(
"--help",
action="help",
default=argparse.SUPPRESS,
help="Show this help message and exit.",
)
PARSER.add_argument(
"--hierarchical",
action="store_true",
help=
"Switch the value of the hierarchical sorting of the tested file. %s"
% (CURRENT_VALUE_FORMAT +
repr(CONFIGURATION["hierarchical_sorting"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"-h",
"--host",
action="store_true",
help="Switch the value of the generation of hosts file. %s"
%
(CURRENT_VALUE_FORMAT +
repr(CONFIGURATION["generate_hosts"]) + Style.RESET_ALL),
)
PARSER.add_argument(
"--http",
action="store_true",
help="Switch the value of the usage of HTTP code. %s" %
(CURRENT_VALUE_FORMAT + repr(HTTP_CODE["active"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"--iana",
action="store_true",
help="Update/Generate `iana-domains-db.json`.",
)
PARSER.add_argument(
"--idna",
action="store_true",
help="Switch the value of the IDNA conversion. %s" %
(CURRENT_VALUE_FORMAT +
repr(CONFIGURATION["idna_conversion"]) + Style.RESET_ALL),
)
PARSER.add_argument(
"-ip",
type=str,
help=
"Change the IP to print in the hosts files with the given one. %s"
% (CURRENT_VALUE_FORMAT +
repr(CONFIGURATION["custom_ip"]) + Style.RESET_ALL),
)
PARSER.add_argument(
"--json",
action="store_true",
help="Switch the value of the generation "
"of the JSON formatted list of domains. %s" %
(CURRENT_VALUE_FORMAT +
repr(CONFIGURATION["generate_json"]) + Style.RESET_ALL),
)
PARSER.add_argument(
"--less",
action="store_true",
help="Output less informations on screen. %s" %
(CURRENT_VALUE_FORMAT + repr(Core.switch("less")) +
Style.RESET_ALL),
)
PARSER.add_argument(
"--local",
action="store_true",
help="Switch the value of the local network testing. %s" %
(CURRENT_VALUE_FORMAT + repr(Core.switch("local")) +
Style.RESET_ALL),
)
PARSER.add_argument("--link",
type=str,
help="Download and test the given file.")
PARSER.add_argument(
"-m",
"--mining",
action="store_true",
help="Switch the value of the mining subsystem usage. %s" %
(CURRENT_VALUE_FORMAT + repr(CONFIGURATION["mining"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"-n",
"--no-files",
action="store_true",
help=
"Switch the value of the production of output files. %s" %
(CURRENT_VALUE_FORMAT + repr(CONFIGURATION["no_files"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"-nl",
"--no-logs",
action="store_true",
help="Switch the value of the production of logs files "
"in the case we encounter some errors. %s" %
(CURRENT_VALUE_FORMAT + repr(not CONFIGURATION["logs"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"-nu",
"--no-unified",
action="store_true",
help="Switch the value of the production unified logs "
"under the output directory. %s" %
(CURRENT_VALUE_FORMAT +
repr(not CONFIGURATION["unified"]) + Style.RESET_ALL),
)
PARSER.add_argument(
"-nw",
"--no-whois",
action="store_true",
help=
"Switch the value the usage of whois to test domain's status. %s"
% (CURRENT_VALUE_FORMAT + repr(CONFIGURATION["no_whois"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"-p",
"--percentage",
action="store_true",
help="Switch the value of the percentage output mode. %s" %
(CURRENT_VALUE_FORMAT +
repr(CONFIGURATION["show_percentage"]) + Style.RESET_ALL),
)
PARSER.add_argument(
"--plain",
action="store_true",
help="Switch the value of the generation "
"of the plain list of domains. %s" %
(CURRENT_VALUE_FORMAT + repr(
CONFIGURATION["plain_list_domain"]) + Style.RESET_ALL),
)
PARSER.add_argument(
"--production",
action="store_true",
help="Prepare the repository for production.",
)
PARSER.add_argument(
"-psl",
"--public-suffix",
action="store_true",
help="Update/Generate `public-suffix.json`.",
)
PARSER.add_argument(
"-q",
"--quiet",
action="store_true",
help="Run the script in quiet mode. %s" %
(CURRENT_VALUE_FORMAT + repr(CONFIGURATION["quiet"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"--share-logs",
action="store_true",
help="Switch the value of the sharing of logs. %s" %
(CURRENT_VALUE_FORMAT + repr(CONFIGURATION["share_logs"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"-s",
"--simple",
action="store_true",
help="Switch the value of the simple output mode. %s" %
(CURRENT_VALUE_FORMAT + repr(CONFIGURATION["simple"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"--split",
action="store_true",
help=
"Switch the value of the split of the generated output files. %s"
% (CURRENT_VALUE_FORMAT + repr(
CONFIGURATION["inactive_database"]) + Style.RESET_ALL),
)
PARSER.add_argument(
"--syntax",
action="store_true",
help="Switch the value of the syntax test mode. %s" %
(CURRENT_VALUE_FORMAT + repr(CONFIGURATION["syntax"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"-t",
"--timeout",
type=int,
default=3,
help="Switch the value of the timeout. %s" %
(CURRENT_VALUE_FORMAT +
repr(CONFIGURATION["seconds_before_http_timeout"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"--travis",
action="store_true",
help="Switch the value of the Travis mode. %s" %
(CURRENT_VALUE_FORMAT + repr(CONFIGURATION["travis"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"--travis-branch",
type=str,
default="master",
help="Switch the branch name where we are going to push. %s"
% (CURRENT_VALUE_FORMAT +
repr(CONFIGURATION["travis_branch"]) + Style.RESET_ALL),
)
PARSER.add_argument("-u",
"--url",
type=str,
help="Analyze the given URL.")
PARSER.add_argument(
"-uf",
"--url-file",
type=str,
help="Read and test the list of URL of the given file. "
"If a URL is given we download and test the content of the given URL.", # pylint: disable=line-too-long
)
PARSER.add_argument(
"-ua",
"--user-agent",
type=str,
help="Set the user-agent to use and set every time we "
"interact with everything which is not our logs sharing system.", # pylint: disable=line-too-long
)
PARSER.add_argument(
"-v",
"--version",
help="Show the version of PyFunceble and exit.",
action="version",
version="%(prog)s " + VERSION,
)
PARSER.add_argument(
"-vsc",
"--verify-ssl-certificate",
action="store_true",
help="Switch the value of the verification of the "
"SSL/TLS certificate when testing for URL. %s" %
(CURRENT_VALUE_FORMAT +
repr(CONFIGURATION["verify_ssl_certificate"]) +
Style.RESET_ALL),
)
PARSER.add_argument(
"-wdb",
"--whois-database",
action="store_true",
help="Switch the value of the usage of a database to store "
"whois data in order to avoid whois servers rate limit. %s"
%
(CURRENT_VALUE_FORMAT +
repr(CONFIGURATION["whois_database"]) + Style.RESET_ALL),
)
ARGS = PARSER.parse_args()
if ARGS.less:
CONFIGURATION.update({"less": ARGS.less})
elif not ARGS.all:
CONFIGURATION.update({"less": ARGS.all})
if ARGS.adblock:
CONFIGURATION.update({"adblock": Core.switch("adblock")})
if ARGS.auto_continue:
CONFIGURATION.update(
{"auto_continue": Core.switch("auto_continue")})
if ARGS.autosave_minutes:
CONFIGURATION.update(
{"travis_autosave_minutes": ARGS.autosave_minutes})
if ARGS.clean:
Clean(None)
if ARGS.clean_all:
Clean(None, ARGS.clean_all)
if ARGS.cmd:
CONFIGURATION.update({"command": ARGS.cmd})
if ARGS.cmd_before_end:
CONFIGURATION.update(
{"command_before_end": ARGS.cmd_before_end})
if ARGS.commit_autosave_message:
CONFIGURATION.update({
"travis_autosave_commit":
ARGS.commit_autosave_message
})
if ARGS.commit_results_message:
CONFIGURATION.update({
"travis_autosave_final_commit":
ARGS.commit_results_message
})
if ARGS.database:
CONFIGURATION.update({
"inactive_database":
Core.switch("inactive_database")
})
if ARGS.days_between_db_retest:
CONFIGURATION.update({
"days_between_db_retest":
ARGS.days_between_db_retest
})
if ARGS.debug:
CONFIGURATION.update({"debug": Core.switch("debug")})
if ARGS.directory_structure:
DirectoryStructure()
if ARGS.execution:
CONFIGURATION.update({
"show_execution_time":
Core.switch("show_execution_time")
})
if ARGS.filter:
CONFIGURATION.update({"filter": ARGS.filter})
if ARGS.hierarchical:
CONFIGURATION.update({
"hierarchical_sorting":
Core.switch("hierarchical_sorting")
})
if ARGS.host:
CONFIGURATION.update(
{"generate_hosts": Core.switch("generate_hosts")})
if ARGS.http:
HTTP_CODE.update(
{"active": Core.switch(HTTP_CODE["active"], True)})
if ARGS.iana:
IANA().update()
if ARGS.idna:
CONFIGURATION.update(
{"idna_conversion": Core.switch("idna_conversion")})
if ARGS.ip:
CONFIGURATION.update({"custom_ip": ARGS.ip})
if ARGS.json:
CONFIGURATION.update(
{"generate_json": Core.switch("generate_json")})
if ARGS.local:
CONFIGURATION.update({"local": Core.switch("local")})
if ARGS.mining:
CONFIGURATION.update({"mining": Core.switch("mining")})
if ARGS.no_files:
CONFIGURATION.update({"no_files": Core.switch("no_files")})
if ARGS.no_logs:
CONFIGURATION.update({"logs": Core.switch("logs")})
if ARGS.no_unified:
CONFIGURATION.update({"unified": Core.switch("unified")})
if ARGS.no_whois:
CONFIGURATION.update({"no_whois": Core.switch("no_whois")})
if ARGS.percentage:
CONFIGURATION.update(
{"show_percentage": Core.switch("show_percentage")})
if ARGS.plain:
CONFIGURATION.update({
"plain_list_domain":
Core.switch("plain_list_domain")
})
if ARGS.production:
Production()
if ARGS.public_suffix:
PublicSuffix().update()
if ARGS.quiet:
CONFIGURATION.update({"quiet": Core.switch("quiet")})
if ARGS.share_logs:
CONFIGURATION.update(
{"share_logs": Core.switch("share_logs")})
if ARGS.simple:
CONFIGURATION.update({
"simple": Core.switch("simple"),
"quiet": Core.switch("quiet")
})
if ARGS.split:
CONFIGURATION.update({"split": Core.switch("split")})
if ARGS.syntax:
CONFIGURATION.update({"syntax": Core.switch("syntax")})
if ARGS.timeout and ARGS.timeout % 3 == 0:
CONFIGURATION.update(
{"seconds_before_http_timeout": ARGS.timeout})
if ARGS.travis:
CONFIGURATION.update({"travis": Core.switch("travis")})
if ARGS.travis_branch:
CONFIGURATION.update({"travis_branch": ARGS.travis_branch})
if ARGS.user_agent:
CONFIGURATION.update({"user_agent": ARGS.user_agent})
if ARGS.verify_ssl_certificate:
CONFIGURATION.update({
"verify_ssl_certificate":
ARGS.verify_ssl_certificate
})
if ARGS.whois_database:
CONFIGURATION.update(
{"whois_database": Core.switch("whois_database")})
if not CONFIGURATION["quiet"]:
Core.colorify_logo(home=True)
# We compare the versions (upstream and local) and in between.
Version().compare()
# We call our Core which will handle all case depending of the configuration or
# the used command line arguments.
Core(
domain_or_ip_to_test=ARGS.domain,
file_path=ARGS.file,
url_to_test=ARGS.url,
url_file=ARGS.url_file,
link_to_test=ARGS.link,
)
except KeyError as e:
if not Version(True).is_cloned():
# We are not into the cloned version.
# We merge the local with the upstream configuration.
Merge(CURRENT_DIRECTORY)
else:
# We are in the cloned version.
# We raise the exception.
#
# Note: The purpose of this is to avoid having
# to search for a mistake while developing.
raise e
except KeyboardInterrupt:
stay_safe()
