|
|
|
@ -655,14 +655,6 @@ def main():
|
|
|
|
|
default=None,
|
|
|
|
|
help="Load data from a JSON file or an online, valid, JSON file.",
|
|
|
|
|
)
|
|
|
|
|
parser.add_argument(
|
|
|
|
|
"--pull-request",
|
|
|
|
|
"-pr",
|
|
|
|
|
action="store",
|
|
|
|
|
dest="pull_request",
|
|
|
|
|
default=None,
|
|
|
|
|
help="Load data.json file from a specific pull request.",
|
|
|
|
|
)
|
|
|
|
|
parser.add_argument(
|
|
|
|
|
"--timeout",
|
|
|
|
|
action="store",
|
|
|
|
@ -787,21 +779,25 @@ def main():
|
|
|
|
|
|
|
|
|
|
# Create object with all information about sites we are aware of.
|
|
|
|
|
try:
|
|
|
|
|
if args.local and args.pull_request is None:
|
|
|
|
|
if args.local:
|
|
|
|
|
sites = SitesInformation(
|
|
|
|
|
os.path.join(os.path.dirname(__file__), "resources/data.json")
|
|
|
|
|
)
|
|
|
|
|
else:
|
|
|
|
|
# Check for conflicting --json and --pull-request arguments
|
|
|
|
|
if args.json_file is not None and args.pull_request is not None:
|
|
|
|
|
print("You cannot use both --json and --pull-request at the same time.")
|
|
|
|
|
sys.exit(1)
|
|
|
|
|
|
|
|
|
|
json_file_location = args.json_file
|
|
|
|
|
if args.pull_request is not None:
|
|
|
|
|
pull_url = f"https://api.github.com/repos/sherlock-project/sherlock/pulls/{args.pull_request}"
|
|
|
|
|
if args.json_file:
|
|
|
|
|
# If --json parameter is a number, interpret it as a pull request number
|
|
|
|
|
if args.json_file.isnumeric():
|
|
|
|
|
pull_number = args.json_file
|
|
|
|
|
pull_url = f"https://api.github.com/repos/sherlock-project/sherlock/pulls/{pull_number}"
|
|
|
|
|
pull_request_raw = requests.get(pull_url).text
|
|
|
|
|
pull_request_json = json_loads(pull_request_raw)
|
|
|
|
|
|
|
|
|
|
# Check if it's a valid pull request
|
|
|
|
|
if "message" in pull_request_json:
|
|
|
|
|
print(f"ERROR: Pull request #{pull_number} not found.")
|
|
|
|
|
sys.exit(1)
|
|
|
|
|
|
|
|
|
|
head_commit_sha = pull_request_json["head"]["sha"]
|
|
|
|
|
json_file_location = f"https://raw.githubusercontent.com/sherlock-project/sherlock/{head_commit_sha}/sherlock_project/resources/data.json"
|
|
|
|
|
|
|
|
|
|