python/jc
1
0
Fork 0
mirror of https://github.com/kellyjonbrazil/jc.git synced 2025-08-06 22:32:54 +02:00
Code Issues Releases Activity

formatting

Browse Source
This commit is contained in:
Kelly Brazil
2022-02-07 16:31:43 -08:00
parent d392c7f901
commit f6c1463c15
1 changed files with 44 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
jc/cli.py
View File

@ -89,13 +89,15 @@ def set_env_colors(env_colors=None):
""" """
Return a dictionary to be used in Pygments custom style class. Return a dictionary to be used in Pygments custom style class.
Grab custom colors from JC_COLORS environment variable. JC_COLORS env variable takes 4 comma Grab custom colors from JC_COLORS environment variable. JC_COLORS env
separated string values and should be in the format of: variable takes 4 comma separated string values and should be in the
format of:
JC_COLORS=<keyname_color>,<keyword_color>,<number_color>,<string_color> JC_COLORS=<keyname_color>,<keyword_color>,<number_color>,<string_color>
Where colors are: black, red, green, yellow, blue, magenta, cyan, gray, brightblack, brightred, Where colors are: black, red, green, yellow, blue, magenta, cyan, gray,
brightgreen, brightyellow, brightblue, brightmagenta, brightcyan, white, default brightblack, brightred, brightgreen, brightyellow,
brightblue, brightmagenta, brightcyan, white, default
Default colors: Default colors:
@ -132,8 +134,10 @@ def set_env_colors(env_colors=None):
def piped_output(force_color): def piped_output(force_color):
"""Return False if stdout is a TTY. True if output is being piped to another program """
and foce_color is True. This allows forcing of ANSI color codes even when using pipes. Return False if stdout is a TTY. True if output is being piped to
another program and foce_color is True. This allows forcing of ANSI
color codes even when using pipes.
""" """
return not sys.stdout.isatty() and not force_color return not sys.stdout.isatty() and not force_color
@ -224,8 +228,8 @@ def helptext():
def help_doc(options): def help_doc(options):
""" """
Returns the parser documentation if a parser is found in the arguments, otherwise Returns the parser documentation if a parser is found in the arguments,
the general help text is returned. otherwise the general help text is returned.
""" """
for arg in options: for arg in options:
parser_name = parser_shortname(arg) parser_name = parser_shortname(arg)
@ -253,7 +257,10 @@ def versiontext():
def json_out(data, pretty=False, env_colors=None, mono=False, piped_out=False): def json_out(data, pretty=False, env_colors=None, mono=False, piped_out=False):
"""Return a JSON formatted string. String may include color codes or be pretty printed.""" """
Return a JSON formatted string. String may include color codes or be
pretty printed.
"""
separators = (',', ':') separators = (',', ':')
indent = None indent = None
@ -277,9 +284,9 @@ def magic_parser(args):
Parse command arguments for magic syntax: jc -p ls -al Parse command arguments for magic syntax: jc -p ls -al
Return a tuple: Return a tuple:
valid_command (bool) is this a valid command? (exists in magic dict) valid_command (bool) is this a valid cmd? (exists in magic dict)
run_command (list) list of the user's command to run. None if no command. run_command (list) list of the user's cmd to run. None if no cmd.
jc_parser (str) parser to use for this user's command. jc_parser (str) parser to use for this user's cmd.
jc_options (list) list of jc options jc_options (list) list of jc options
""" """
# bail immediately if there are no args or a parser is defined # bail immediately if there are no args or a parser is defined
@ -335,12 +342,15 @@ def magic_parser(args):
def run_user_command(command): def run_user_command(command):
"""Use subprocess to run the user's command. Returns the STDOUT, STDERR, and the Exit Code as a tuple.""" """
Use subprocess to run the user's command. Returns the STDOUT, STDERR,
and the Exit Code as a tuple.
"""
proc = subprocess.Popen(command, proc = subprocess.Popen(command,
stdout=subprocess.PIPE, stdout=subprocess.PIPE,
stderr=subprocess.PIPE, stderr=subprocess.PIPE,
close_fds=False, # Allows inheriting file descriptors. Useful for process substitution close_fds=False, # Allows inheriting file descriptors;
universal_newlines=True) universal_newlines=True) # useful for process substitution
stdout, stderr = proc.communicate() stdout, stderr = proc.communicate()
return ( return (
@ -441,14 +451,18 @@ def main():
raise raise
error_msg = os.strerror(e.errno) error_msg = os.strerror(e.errno)
utils.error_message([f'"{run_command_str}" command could not be run: {error_msg}. For details use the -d or -dd option.']) utils.error_message([
f'"{run_command_str}" command could not be run: {error_msg}. For details use the -d or -dd option.'
])
sys.exit(combined_exit_code(magic_exit_code, JC_ERROR_EXIT)) sys.exit(combined_exit_code(magic_exit_code, JC_ERROR_EXIT))
except Exception: except Exception:
if debug: if debug:
raise raise
utils.error_message([f'"{run_command_str}" command could not be run. For details use the -d or -dd option.']) utils.error_message([
f'"{run_command_str}" command could not be run. For details use the -d or -dd option.'
])
sys.exit(combined_exit_code(magic_exit_code, JC_ERROR_EXIT)) sys.exit(combined_exit_code(magic_exit_code, JC_ERROR_EXIT))
elif run_command is not None: elif run_command is not None:
@ -489,7 +503,10 @@ def main():
# streaming # streaming
if getattr(parser.info, 'streaming', None): if getattr(parser.info, 'streaming', None):
result = parser.parse(sys.stdin, raw=raw, quiet=quiet, ignore_exceptions=ignore_exceptions) result = parser.parse(sys.stdin,
raw=raw,
quiet=quiet,
ignore_exceptions=ignore_exceptions)
for line in result: for line in result:
print(json_out(line, print(json_out(line,
pretty=pretty, pretty=pretty,
@ -503,7 +520,9 @@ def main():
# regular # regular
else: else:
data = magic_stdout or sys.stdin.read() data = magic_stdout or sys.stdin.read()
result = parser.parse(data, raw=raw, quiet=quiet) result = parser.parse(data,
raw=raw,
quiet=quiet)
print(json_out(result, print(json_out(result,
pretty=pretty, pretty=pretty,
env_colors=jc_colors, env_colors=jc_colors,
@ -517,10 +536,11 @@ def main():
if debug: if debug:
raise raise
utils.error_message([f'Parser issue with {parser_name}:', utils.error_message([
f'{e.__class__.__name__}: {e}', f'Parser issue with {parser_name}:', f'{e.__class__.__name__}: {e}',
'If this is the correct parser, try setting the locale to C (LANG=C).', 'If this is the correct parser, try setting the locale to C (LANG=C).',
'For details use the -d or -dd option. Use "jc -h" for help.']) 'For details use the -d or -dd option. Use "jc -h" for help.'
])
sys.exit(combined_exit_code(magic_exit_code, JC_ERROR_EXIT)) sys.exit(combined_exit_code(magic_exit_code, JC_ERROR_EXIT))
except json.JSONDecodeError: except json.JSONDecodeError: