File: minitj.py 1 #!/usr/bin/python3 2 3 # The MIT License (MIT) 4 # 5 # Copyright © 2024 pacman64 6 # 7 # Permission is hereby granted, free of charge, to any person obtaining a copy 8 # of this software and associated documentation files (the “Software”), to deal 9 # in the Software without restriction, including without limitation the rights 10 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 11 # copies of the Software, and to permit persons to whom the Software is 12 # furnished to do so, subject to the following conditions: 13 # 14 # The above copyright notice and this permission notice shall be included in 15 # all copies or substantial portions of the Software. 16 # 17 # THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 18 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 19 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 20 # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 21 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 22 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE 23 # SOFTWARE. 24 25 26 info = ''' 27 minitj [options...] [python expression] [files/URIs...] 28 29 This is the MINImal version of the Transform Json tool. 30 ''' 31 32 33 from json import dump, load 34 from sys import argv, exit, stderr, stdin, stdout 35 from typing import Iterable 36 37 38 if len(argv) < 2: 39 print(info.strip(), file=stderr) 40 exit(0) 41 if len(argv) == 2 and argv[1] in ('-h', '--h', '-help', '--help'): 42 print(info.strip()) 43 exit(0) 44 45 46 class Skip: 47 pass 48 49 50 skip = Skip() 51 52 53 def rescue(attempt, fallback = None): 54 try: 55 return attempt() 56 except Exception as e: 57 if callable(fallback): 58 return fallback(e) 59 return fallback 60 61 catch = rescue 62 catched = rescue 63 caught = rescue 64 recover = rescue 65 recovered = rescue 66 rescued = rescue 67 68 69 def result_needs_fixing(x): 70 if x is None or isinstance(x, (bool, int, float, str)): 71 return False 72 rec = result_needs_fixing 73 if isinstance(x, dict): 74 return any(rec(k) or rec(v) for k, v in x.items()) 75 if isinstance(x, (list, tuple)): 76 return any(rec(e) for e in x) 77 return True 78 79 80 def fix_result(x, default): 81 if x is type: 82 return type(default).__name__ 83 84 # if expression results in a func, auto-call it with the original data 85 if callable(x): 86 x = x(default) 87 88 if x is None or isinstance(x, (bool, int, float, str)): 89 return x 90 91 rec = fix_result 92 93 if isinstance(x, dict): 94 return { 95 rec(k, default): rec(v, default) for k, v in x.items() if not 96 (isinstance(k, Skip) or isinstance(v, Skip)) 97 } 98 if isinstance(x, Iterable): 99 return tuple(rec(e, default) for e in x if not isinstance(e, Skip)) 100 101 if isinstance(x, Exception): 102 raise x 103 104 return None if isinstance(x, Skip) else str(x) 105 106 107 def seems_url(path): 108 protocols = ('https://', 'http://', 'file://', 'ftp://', 'data:') 109 return any(path.startswith(p) for p in protocols) 110 111 112 cr = '\r' 113 crlf = '\r\n' 114 dquo = '"' 115 dquote = '"' 116 empty = '' 117 lcurly = '{' 118 lf = '\n' 119 rcurly = '}' 120 s = '' 121 squo = '\'' 122 squote = '\'' 123 utf8bom = '\xef\xbb\xbf' 124 125 nil = None 126 none = None 127 null = None 128 129 130 no_input_opts = ( 131 '=', '-n', '--n', '-nil', '--nil', '-none', '--none', '-null', '--null', 132 ) 133 compact_output_opts = ( 134 '-c', '--c', '-compact', '--compact', '-j0', '--j0', '-json0', '--json0', 135 ) 136 more_modules_opts = ('-mm', '--mm', '-more', '--more') 137 138 args = argv[1:] 139 no_input = False 140 compact_output = False 141 142 while len(args) > 0: 143 if args[0] in no_input_opts: 144 no_input = True 145 args = args[1:] 146 continue 147 148 if args[0] in compact_output_opts: 149 compact_output = True 150 args = args[1:] 151 continue 152 153 if args[0] in more_modules_opts: 154 import functools 155 import itertools 156 import math 157 import random 158 import statistics 159 import string 160 import time 161 args = args[1:] 162 continue 163 164 break 165 166 167 try: 168 expr = 'data' 169 if len(args) > 0: 170 expr = args[0] 171 args = args[1:] 172 173 if expr == '.': 174 expr = 'data' 175 176 if len(args) > 1: 177 raise Exception('can\'t use more than 1 input') 178 path = '-' if len(args) == 0 else args[0] 179 180 if no_input: 181 data = None 182 elif path == '-': 183 data = load(stdin) 184 elif seems_url(path): 185 from io import TextIOWrapper 186 from urllib.request import urlopen 187 with urlopen(path) as inp: 188 with TextIOWrapper(inp, encoding='utf-8') as txt: 189 data = load(txt) 190 else: 191 with open(path, encoding='utf-8') as inp: 192 data = load(inp) 193 194 exec = None 195 open = None 196 v = value = d = data 197 v = eval(expr) 198 if result_needs_fixing(v): 199 v = fix_result(v, data) 200 201 if compact_output: 202 dump(v, stdout, indent=None, separators=(',', ':'), allow_nan=False) 203 else: 204 dump(v, stdout, indent=2, separators=(',', ': '), allow_nan=False) 205 print() 206 except BrokenPipeError: 207 # quit quietly, instead of showing a confusing error message 208 stderr.close() 209 exit(0) 210 except KeyboardInterrupt: 211 # stderr.close() 212 exit(2) 213 except Exception as e: 214 print(f'\x1b[31m{str(e)}\x1b[0m', file=stderr) 215 exit(1)