diff options
Diffstat (limited to 'monitor_menu')
| -rw-r--r-- | monitor_menu/__main__.py | 11 | 
1 files changed, 8 insertions, 3 deletions
diff --git a/monitor_menu/__main__.py b/monitor_menu/__main__.py index e42c440..fce1a65 100644 --- a/monitor_menu/__main__.py +++ b/monitor_menu/__main__.py @@ -8,6 +8,7 @@ import logging  from os.path import expanduser  import subprocess  from typing import Any +import yaml  import dialog @@ -161,9 +162,13 @@ def main():      args = parser.parse_args()      logging.basicConfig(level=args.log_level.numerical_level) -    config_file = '~/.config/monitor-profiles.json' -    with open(expanduser(config_file)) as fh: -        data = json.load(fh) +    try: +        with open(expanduser("~/.config/monitor-profiles.yaml")) as fh: +            data = yaml.safe_load(fh) +    except FileNotFoundError: +        with open(expanduser("~/.config/monitor-profiles.json")) as fh: +            data = json.load(fh) +      profiles = [Profile.from_json_dict(item) for item in data]      match args.command:  | 
