mirror of
https://github.com/usnistgov/macos_security.git
synced 2026-02-03 14:03:24 +00:00
add function for get_macos_version
This commit is contained in:
@@ -3,6 +3,7 @@
|
||||
|
||||
import argparse
|
||||
import sys
|
||||
import platform
|
||||
from mscp.common_utils import set_logger
|
||||
from mscp.generate import (
|
||||
generate_scap,
|
||||
@@ -10,6 +11,11 @@ from mscp.generate import (
|
||||
)
|
||||
from mscp.cli import Customparser, validate_file
|
||||
|
||||
def get_macos_version() -> float:
|
||||
version_str, _, _ = platform.mac_ver()
|
||||
if version_str:
|
||||
major = int(version_str.split(".")[0])
|
||||
return float(major)
|
||||
|
||||
def main() -> None:
|
||||
logger = set_logger()
|
||||
@@ -67,7 +73,7 @@ def main() -> None:
|
||||
parser.add_argument(
|
||||
"-o",
|
||||
"--os_version",
|
||||
default=15.0,
|
||||
default=get_macos_version(),
|
||||
type=float,
|
||||
help="Operating system version (eg: 14.0, 15.0).",
|
||||
)
|
||||
|
||||
@@ -3,6 +3,7 @@
|
||||
# Standard python modules
|
||||
import argparse
|
||||
import sys
|
||||
import platform
|
||||
from pathlib import Path
|
||||
|
||||
# Local python modules
|
||||
@@ -17,6 +18,11 @@ from .generate import (
|
||||
generate_scap,
|
||||
)
|
||||
|
||||
def get_macos_version() -> float:
|
||||
version_str, _, _ = platform.mac_ver()
|
||||
if version_str:
|
||||
major = int(version_str.split(".")[0])
|
||||
return float(major)
|
||||
|
||||
class Customparser(argparse.ArgumentParser):
|
||||
"""
|
||||
@@ -68,7 +74,7 @@ def parse_cli() -> None:
|
||||
|
||||
parser.add_argument(
|
||||
"--os_version",
|
||||
default=15.0,
|
||||
default=get_macos_version(),
|
||||
type=float,
|
||||
help="Operating system version (eg: 14.0, 15.0).",
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user