mirror of
				https://github.com/mfulz/qmk_firmware.git
				synced 2025-11-03 23:02:34 +01:00 
			
		
		
		
	
		
			
				
	
	
		
			95 lines
		
	
	
		
			3.2 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			95 lines
		
	
	
		
			3.2 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
"""Functions for working with QMK's submodules.
 | 
						|
"""
 | 
						|
from milc import cli
 | 
						|
 | 
						|
 | 
						|
def status():
 | 
						|
    """Returns a dictionary of submodules.
 | 
						|
 | 
						|
    Each entry is a dict of the form:
 | 
						|
 | 
						|
        {
 | 
						|
            'name': 'submodule_name',
 | 
						|
            'status': None/False/True,
 | 
						|
            'githash': '<sha-1 hash for the submodule>'
 | 
						|
            'shorthash': '<short hash for the submodule>'
 | 
						|
            'describe': '<output of `git describe --tags`>'
 | 
						|
            'last_log_message': 'log message'
 | 
						|
            'last_log_timestamp': 'timestamp'
 | 
						|
        }
 | 
						|
 | 
						|
    status is None when the submodule doesn't exist, False when it's out of date, and True when it's current
 | 
						|
    """
 | 
						|
    submodules = {}
 | 
						|
    git_cmd = cli.run(['git', 'submodule', 'status'], timeout=30)
 | 
						|
 | 
						|
    for line in git_cmd.stdout.split('\n'):
 | 
						|
        if not line:
 | 
						|
            continue
 | 
						|
 | 
						|
        status = line[0]
 | 
						|
        githash, submodule = line[1:].split()[:2]
 | 
						|
        submodules[submodule] = {'name': submodule, 'githash': githash}
 | 
						|
 | 
						|
        if status == '-':
 | 
						|
            submodules[submodule]['status'] = None
 | 
						|
        elif status == '+':
 | 
						|
            submodules[submodule]['status'] = False
 | 
						|
        elif status == ' ':
 | 
						|
            submodules[submodule]['status'] = True
 | 
						|
        else:
 | 
						|
            raise ValueError('Unknown `git submodule status` sha-1 prefix character: "%s"' % status)
 | 
						|
 | 
						|
    submodule_logs = cli.run(['git', 'submodule', '-q', 'foreach', 'git --no-pager log --no-show-signature --pretty=format:"$sm_path%x01%h%x01%ad%x01%s%x0A" --date=iso -n1'])
 | 
						|
    for log_line in submodule_logs.stdout.split('\n'):
 | 
						|
        if not log_line:
 | 
						|
            continue
 | 
						|
 | 
						|
        r = log_line.split('\x01')
 | 
						|
        submodule = r[0]
 | 
						|
        submodules[submodule]['shorthash'] = r[1] if len(r) > 1 else ''
 | 
						|
        submodules[submodule]['last_log_timestamp'] = r[2] if len(r) > 2 else ''
 | 
						|
        submodules[submodule]['last_log_message'] = r[3] if len(r) > 3 else ''
 | 
						|
 | 
						|
    submodule_tags = cli.run(['git', 'submodule', '-q', 'foreach', '\'echo $sm_path `git describe --tags`\''])
 | 
						|
    for log_line in submodule_tags.stdout.split('\n'):
 | 
						|
        if not log_line:
 | 
						|
            continue
 | 
						|
 | 
						|
        r = log_line.split()
 | 
						|
        submodule = r[0]
 | 
						|
        submodules[submodule]['describe'] = r[1] if len(r) > 1 else ''
 | 
						|
 | 
						|
    return submodules
 | 
						|
 | 
						|
 | 
						|
def update(submodules=None):
 | 
						|
    """Update the submodules.
 | 
						|
 | 
						|
        submodules
 | 
						|
            A string containing a single submodule or a list of submodules.
 | 
						|
    """
 | 
						|
    git_sync_cmd = ['git', 'submodule', 'sync']
 | 
						|
    git_update_cmd = ['git', 'submodule', 'update', '--init']
 | 
						|
 | 
						|
    if submodules is None:
 | 
						|
        # Update everything
 | 
						|
        git_sync_cmd.append('--recursive')
 | 
						|
        git_update_cmd.append('--recursive')
 | 
						|
        cli.run(git_sync_cmd, check=True)
 | 
						|
        cli.run(git_update_cmd, check=True)
 | 
						|
 | 
						|
    else:
 | 
						|
        if isinstance(submodules, str):
 | 
						|
            # Update only a single submodule
 | 
						|
            git_sync_cmd.append(submodules)
 | 
						|
            git_update_cmd.append(submodules)
 | 
						|
            cli.run(git_sync_cmd, check=True)
 | 
						|
            cli.run(git_update_cmd, check=True)
 | 
						|
 | 
						|
        else:
 | 
						|
            # Update submodules in a list
 | 
						|
            for submodule in submodules:
 | 
						|
                cli.run([*git_sync_cmd, submodule], check=True)
 | 
						|
                cli.run([*git_update_cmd, submodule], check=True)
 |