prompt.py

from __future__ import unicode_literals

from prompt_toolkit.contrib.completers import WordCompleter
from util.GitCompleter import GitCompleter

from prompt_toolkit import prompt
from prompt_toolkit.interface import AbortAction
from prompt_toolkit.styles import style_from_dict
from prompt_toolkit.token import Token
from prompt_toolkit.history import FileHistory

import subprocess
import os
import sys
from os.path import expanduser
import json

cli_style = style_from_dict({
    Token.Comment:   '#888888 italic',
    Token.Keyword:   '#ff88ff bold',
	Token: '#ff0066'
})

scriptDir=os.path.dirname(os.path.abspath(sys.argv[0]))
with open(os.path.join(scriptDir,'git_command.json'), 'r') as f:
	git_command=json.load(f)
cli_completer = GitCompleter(git_command.keys(),meta_dict=git_command,ignore_case=True,sentence=True)


def main():
	home = expanduser("~")
	git_history = FileHistory(home+'/.git-inter-cli-history')
	try:
		while True:
	
			text = prompt('$ git ', completer=cli_completer,
						  style=cli_style,complete_while_typing=True,
						  mouse_support=True,history=git_history,
						  on_abort=AbortAction.RETRY)
			if text=='q':
				while True:
					text=prompt('$ ',style=cli_style)
					if text=='git':
						break
					status=subprocess.call(text, shell=True)
			elif text=='':
				currentBranchName=subprocess.check_output('git rev-parse --abbrev-ref HEAD', shell=True)
				currentPath=os.getcwd()				
				print(currentPath.replace('\n','')+' ('+currentBranchName.replace('\n','')+')')
				continue
			else:
				status=subprocess.call("git " + text, shell=True)
	except EOFError:
		print('exit')
		


if __name__ == '__main__':
    main()