* Implement stdin for json2c command * Refactor * Handle json decode error * Add stdin support for c2json cli command * Refactor to prevent code duplication * Change exit(1) to return False in c2json command * Remove unused import
		
			
				
	
	
		
			60 lines
		
	
	
		
			2.0 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			60 lines
		
	
	
		
			2.0 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
| """Generate a keymap.c from a configurator export.
 | |
| """
 | |
| import json
 | |
| import sys
 | |
| 
 | |
| from milc import cli
 | |
| 
 | |
| import qmk.keymap
 | |
| import qmk.path
 | |
| 
 | |
| 
 | |
| @cli.argument('-o', '--output', arg_only=True, type=qmk.path.normpath, help='File to write to')
 | |
| @cli.argument('-q', '--quiet', arg_only=True, action='store_true', help="Quiet mode, only output error messages")
 | |
| @cli.argument('filename', type=qmk.path.normpath, arg_only=True, help='Configurator JSON file')
 | |
| @cli.subcommand('Creates a keymap.c from a QMK Configurator export.')
 | |
| def json2c(cli):
 | |
|     """Generate a keymap.c from a configurator export.
 | |
| 
 | |
|     This command uses the `qmk.keymap` module to generate a keymap.c from a configurator export. The generated keymap is written to stdout, or to a file if -o is provided.
 | |
|     """
 | |
| 
 | |
|     try:
 | |
|         # Parse the configurator from stdin
 | |
|         if cli.args.filename and cli.args.filename.name == '-':
 | |
|             user_keymap = json.load(sys.stdin)
 | |
| 
 | |
|         else:
 | |
|             # Error checking
 | |
|             if not cli.args.filename.exists():
 | |
|                 cli.log.error('JSON file does not exist!')
 | |
|                 return False
 | |
| 
 | |
|             # Parse the configurator json file
 | |
|             else:
 | |
|                 user_keymap = json.loads(cli.args.filename.read_text())
 | |
| 
 | |
|     except json.decoder.JSONDecodeError as ex:
 | |
|         cli.log.error('The JSON input does not appear to be valid.')
 | |
|         cli.log.error(ex)
 | |
|         return False
 | |
| 
 | |
|     # Environment processing
 | |
|     if cli.args.output and cli.args.output.name == '-':
 | |
|         cli.args.output = None
 | |
| 
 | |
|     # Generate the keymap
 | |
|     keymap_c = qmk.keymap.generate_c(user_keymap['keyboard'], user_keymap['layout'], user_keymap['layers'])
 | |
| 
 | |
|     if cli.args.output:
 | |
|         cli.args.output.parent.mkdir(parents=True, exist_ok=True)
 | |
|         if cli.args.output.exists():
 | |
|             cli.args.output.replace(cli.args.output.name + '.bak')
 | |
|         cli.args.output.write_text(keymap_c)
 | |
| 
 | |
|         if not cli.args.quiet:
 | |
|             cli.log.info('Wrote keymap to %s.', cli.args.output)
 | |
| 
 | |
|     else:
 | |
|         print(keymap_c)
 |