Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

chore: improved logging #14

Merged
merged 16 commits into from
Aug 2, 2023
Merged
Show file tree
Hide file tree
Changes from 6 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions action.yml
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,10 @@ inputs:
description: comma seperated list of tags to be excluded from appinspect scans
default: ""
required: false
log_level:
description: severity for python logging ("DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL")
default: "INFO"
required: false
runs:
using: "docker"
image: docker://ghcr.io/splunk/appinspect-api-action/appinspect-api-action:v3.0.0
4 changes: 2 additions & 2 deletions entrypoint.sh
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,6 @@
ADDON_NAME=$(ls $INPUT_APP_PATH)
ADDON_FULL_PATH="$INPUT_APP_PATH/$ADDON_NAME"

echo "$INPUT_USERNAME" "$INPUT_PASSWORD" "$ADDON_FULL_PATH" "$INPUT_INCLUDED_TAGS" "$INPUT_EXCLUDED_TAGS"
echo "$INPUT_USERNAME" "$INPUT_PASSWORD" "$ADDON_FULL_PATH" "$INPUT_INCLUDED_TAGS" "$INPUT_EXCLUDED_TAGS" "$INPUT_LOG_LEVEL"

python3 /main.py "$INPUT_USERNAME" "$INPUT_PASSWORD" "$ADDON_FULL_PATH" "$INPUT_INCLUDED_TAGS" "$INPUT_EXCLUDED_TAGS"
python3 /main.py "$INPUT_USERNAME" "$INPUT_PASSWORD" "$ADDON_FULL_PATH" "$INPUT_INCLUDED_TAGS" "$INPUT_EXCLUDED_TAGS" "$INPUT_LOG_LEVEL"
56 changes: 36 additions & 20 deletions main.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
import random
import json
import yaml
import logging

from pathlib import Path
from typing import Dict, Any, Tuple, Callable, Sequence, Optional, List
Expand Down Expand Up @@ -46,10 +47,12 @@ def _retry_request(
for retry_num in range(num_retries):
if retry_num > 0:
sleep_time = rand() + retry_num
print(
logging.info(
f"Sleeping {sleep_time} seconds before retry "
f"{retry_num} of {num_retries - 1} after {reason}"
f"{retry_num} of {num_retries - 1}"
)
if reason:
logging.info(reason)
sleep(sleep_time)
response = requests.request(
method,
Expand All @@ -70,7 +73,7 @@ def _retry_request(
reason = f"response status code: {response.status_code}, for message: {error_message}"
continue
if not validation_function(response):
print("Response did not pass the validation, retrying...")
logging.info("Response did not pass the validation, retrying...")
continue
return response
raise CouldNotRetryRequestException()
Expand All @@ -93,17 +96,18 @@ def _download_report(


def login(username: str, password: str) -> requests.Response:
logging.debug("Sending request to retrieve login token")
try:
return _retry_request(
"GET",
"https://api.splunk.com/2.0/rest/login/splunk",
auth=(username, password),
)
except CouldNotAuthenticateException:
print("Credentials are not correct, please check the configuration.")
logging.error("Credentials are not correct, please check the configuration.")
sys.exit(1)
except CouldNotRetryRequestException:
print("Could not get response after all retries, exiting...")
logging.error("Could not get response after all retries, exiting...")
sys.exit(1)


Expand All @@ -114,6 +118,7 @@ def validate(token: str, build: Path, payload: Dict[str, str]) -> requests.Respo
(build.name, open(build.as_posix(), "rb"), "application/octet-stream"),
)
]
logging.debug(f"Sending package {build.name} for validation")
try:
response = _retry_request(
"POST",
Expand All @@ -126,22 +131,26 @@ def validate(token: str, build: Path, payload: Dict[str, str]) -> requests.Respo
)
return response
except CouldNotAuthenticateException:
print("Credentials are not correct, please check the configuration.")
logging.error("Credentials are not correct, please check the configuration.")
sys.exit(1)
except CouldNotRetryRequestException:
print("Could not get response after all retries, exiting...")
logging.error("Could not get response after all retries, exiting...")
sys.exit(1)


def submit(token: str, request_id: str) -> requests.Response:
def _validate_validation_status(response: requests.Response) -> bool:
return response.json()["status"] == "SUCCESS"
is_successful = response.json()["status"] == "SUCCESS"
if is_successful:
logging.info('Response status is not "SUCCESS"')
return is_successful

# appinspect api needs some time to process the request
# if the response status will be "PROCESSING" wait 60s and make another call

# there is a problem with pycov marking this line as not covered - excluded from coverage
try:
logging.debug("Submitting package")
return _retry_request( # pragma: no cover
"GET",
f"https://appinspect.splunk.com/v1/app/validate/status/{request_id}",
Expand All @@ -153,22 +162,24 @@ def _validate_validation_status(response: requests.Response) -> bool:
validation_function=_validate_validation_status,
)
except CouldNotAuthenticateException:
print("Credentials are not correct, please check the configuration.")
logging.error("Credentials are not correct, please check the configuration.")
sys.exit(1)
except CouldNotRetryRequestException:
print("Could not get response after all retries, exiting...")
logging.error("Could not get response after all retries, exiting...")
sys.exit(1)


def download_json_report(
token: str, request_id: str, payload: Dict[str, Any]
) -> requests.Response:
logging.debug("Downloading response in json format")
return _download_report(
token=token, request_id=request_id, payload=payload, response_type="json"
)


def download_and_save_html_report(token: str, request_id: str, payload: Dict[str, Any]):
logging.debug("Downloading report in html format")
response = _download_report(
token=token, request_id=request_id, payload=payload, response_type="html"
)
Expand All @@ -178,6 +189,9 @@ def download_and_save_html_report(token: str, request_id: str, payload: Dict[str


def get_appinspect_failures_list(response_dict: Dict[str, Any]) -> List[str]:
logging.debug(
f"Parsing json respnose to find failed checks\n response: {response_dict}"
)
reports = response_dict["reports"]
groups = reports[0]["groups"]

Expand All @@ -187,7 +201,7 @@ def get_appinspect_failures_list(response_dict: Dict[str, Any]) -> List[str]:
for check in group["checks"]:
if check["result"] == "failure":
failed_tests_list.append(check["name"])
print(f"Failed appinspect check for name: {check['name']}\n")
logging.debug(f"Failed appinspect check for name: {check['name']}\n")
return failed_tests_list


Expand All @@ -196,26 +210,25 @@ def read_yaml_as_dict(filename_path: Path) -> Dict[str, str]:
try:
out_dict = yaml.safe_load(file)
except yaml.YAMLError as e:
print(f"Can not read yaml file named {filename_path}")
logging.error(f"Can not read yaml file named {filename_path}")
raise e
return out_dict if out_dict else {}


def compare_failures(failures: List[str], expected: List[str]):
if sorted(failures) != sorted(expected):
print(
logging.error(
"Appinspect failures doesn't match appinspect.expect file, check for exceptions file"
)
raise AppinspectFailures


def parse_results(results: Dict[str, Any]):
print(results)
print("\n======== AppInspect Api Results ========")
for metric, count in results["info"].items():
print(f"{metric:>15} : {count: <4}")
if results["info"]["error"] > 0 or results["info"]["failure"] > 0:
print("\nError or failures found in App Inspect\n")
logging.error("\nError or failures found in App Inspect\n")
raise AppinspectChecksFailuresException


Expand All @@ -236,8 +249,8 @@ def compare_against_known_failures(response_json: Dict[str, Any], exceptions_fil
expected_failures = list(read_yaml_as_dict(exceptions_file_path).keys())
compare_failures(failures, expected_failures)
else:
print(
f"ERROR: File `{exceptions_file_path.name}` not found, please create `{exceptions_file_path.name}` file with exceptions\n" # noqa: E501
logging.error(
f"File `{exceptions_file_path.name}` not found, please create `{exceptions_file_path.name}` file with exceptions\n" # noqa: E501
)
sys.exit(1)

Expand All @@ -251,24 +264,27 @@ def main(argv: Optional[Sequence[str]] = None):
parser.add_argument("app_path")
parser.add_argument("included_tags")
parser.add_argument("excluded_tags")
parser.add_argument("log_level")

appinspect_expect_filename = ".appinspect_api.expect.yaml"

args = parser.parse_args(argv)

print(
logging.basicConfig(level=args.log_level)

logging.info(
f"app_path={args.app_path}, included_tags={args.included_tags}, excluded_tags={args.excluded_tags}"
)
build = Path(args.app_path)

login_response = login(args.username, args.password)
token = login_response.json()["data"]["token"]
print("Successfully received token")
logging.info("Successfully received token")

payload = build_payload(args.included_tags, args.excluded_tags)

validate_response = validate(token, build, payload)
print(f"Successfully sent package for validation using {payload}")
logging.info(f"Successfully sent package for validation using {payload}")
request_id = validate_response.json()["request_id"]

submit_response = submit(token, request_id)
Expand Down
Loading