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.

Sign up for GitHub

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

Diagnostic log argument #203

Merged
merged 2 commits into from
Jul 23, 2024
Merged

Diagnostic log argument #203

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
fa4fe25
Add cli argument for diagnostic log location
clint-lawrence Jul 23, 2024
9daa000
Either script path or zip file is required
clint-lawrence Jul 23, 2024
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
20 changes: 14 additions & 6 deletions src/fixate/__main__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
from importlib.machinery import SourceFileLoader
from zipimport import zipimporter
from pubsub import pub
from pathlib import Path
import fixate.config
from fixate.core.exceptions import SequenceAbort
from fixate.core.ui import user_info_important, user_serial, user_ok
Expand Down Expand Up @@ -38,14 +39,14 @@ def get_parser():
)

# Optional Arguments
mutex_group = parser.add_mutually_exclusive_group()
mutex_group.add_argument(
script_group = parser.add_mutually_exclusive_group(required=True)
script_group.add_argument(
"-p",
"--path",
help="""Path to the directory where the script file is located.
This is mutually exclusive with --zip""",
)
mutex_group.add_argument(
script_group.add_argument(
"-z",
"--zip",
help="""Path to zip file of test scripts. Mutually exclusive with --path""",
Expand Down Expand Up @@ -118,9 +119,16 @@ def get_parser():
action="store_true",
help="The sequencer will not prompt for retries.",
)
parser.add_argument(
diagnostic_group = parser.add_mutually_exclusive_group()
diagnostic_group.add_argument(
"--disable-logs", action="store_true", help="Turn off diagnostic logs"
)
diagnostic_group.add_argument(
"--diagnostic-log-dir",
default=fixate.config.LOG_DIRECTORY,
type=Path,
help="Directory to store the diagnostic log",
)

return parser

Expand Down Expand Up @@ -399,10 +407,10 @@ def run_main_program(test_script_path=None, main_args=None):
parser = get_parser()
args, unknown = parser.parse_known_args(args=main_args)
if not args.disable_logs:
fixate.config.LOG_DIRECTORY.mkdir(parents=True, exist_ok=True)
args.diagnostic_log_dir.mkdir(parents=True, exist_ok=True)

handler = RotateEachInstanceHandler(
fixate.config.LOG_DIRECTORY / "fixate.log", backupCount=10
args.diagnostic_log_dir / "fixate.log", backupCount=10
)
handler.setFormatter(
logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
Expand Down
Loading