mirror of
				https://github.com/mfulz/qmk_firmware.git
				synced 2025-11-04 07:12:33 +01:00 
			
		
		
		
	Add support for passing files at the command line
This commit is contained in:
		
							parent
							
								
									9547774962
								
							
						
					
					
						commit
						1784d1bfac
					
				@ -6,22 +6,24 @@ import subprocess
 | 
				
			|||||||
from milc import cli
 | 
					from milc import cli
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					@cli.argument('files', nargs='*', help='Filename(s) to format.')
 | 
				
			||||||
@cli.entrypoint("Format C code according to QMK's style.")
 | 
					@cli.entrypoint("Format C code according to QMK's style.")
 | 
				
			||||||
def main(cli):
 | 
					def main(cli):
 | 
				
			||||||
    """Format C code according to QMK's style.
 | 
					    """Format C code according to QMK's style.
 | 
				
			||||||
    """
 | 
					    """
 | 
				
			||||||
    clang_format = ['clang-format', '-i']
 | 
					    clang_format = ['clang-format', '-i']
 | 
				
			||||||
    code_files = []
 | 
					    if not cli.args.files:
 | 
				
			||||||
    for dir in ['drivers', 'quantum', 'tests', 'tmk_core']:
 | 
					        for dir in ['drivers', 'quantum', 'tests', 'tmk_core']:
 | 
				
			||||||
        for dirpath, dirnames, filenames in os.walk(dir):
 | 
					            for dirpath, dirnames, filenames in os.walk(dir):
 | 
				
			||||||
            if 'tmk_core/protocol/usb_hid' in dirpath:
 | 
					                if 'tmk_core/protocol/usb_hid' in dirpath:
 | 
				
			||||||
                continue
 | 
					                    continue
 | 
				
			||||||
            for name in filenames:
 | 
					                for name in filenames:
 | 
				
			||||||
                if name.endswith('.c') or name.endswith('.h') or name.endswith('.cpp'):
 | 
					                    if name.endswith('.c') or name.endswith('.h') or name.endswith('.cpp'):
 | 
				
			||||||
                    code_files.append(os.path.join(dirpath, name))
 | 
					                        cli.args.files.append(os.path.join(dirpath, name))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    try:
 | 
					    try:
 | 
				
			||||||
        subprocess.run(clang_format + code_files, check=True)
 | 
					        subprocess.run(clang_format + cli.args.files, check=True)
 | 
				
			||||||
        cli.log.info('Successfully formatted the C code.')
 | 
					        cli.log.info('Successfully formatted the C code.')
 | 
				
			||||||
    except subprocess.CalledProcessError:
 | 
					    except subprocess.CalledProcessError:
 | 
				
			||||||
        cli.log.error('Error formatting C code!')
 | 
					        cli.log.error('Error formatting C code!')
 | 
				
			||||||
 | 
					        return False
 | 
				
			||||||
 | 
				
			|||||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user