Created
November 9, 2019 14:21
-
-
Save benwu232/c5897dde94392c058f2ed3479c1a0d16 to your computer and use it in GitHub Desktop.
parse arbitrary commands to directory
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import ast | |
import sys | |
#test_string = '{"Nikhil" : 1, "Akshat" : 2, "Akash" : 3}' | |
in_str = "dict1=-ds ar=-ls 10 11 12 dict2=-ds x=1 y=2 seq=-ls 7 8 -le -de 13 -le a=1 c=2 dict2=-ds f=3 h=i -de j=k -de" | |
# dict1=-ds1 a=1 c=2 dict2=-ds2 f=3 h=i -de2 j=k -de2 | |
# dict1=-ds ar=-ls 10 11 12 dict2=-ds x=1 y=2 seq=-ls 7 8 seq2=-ls 9 -le -le -de 13 -le a=1 c=2 dict3=-ds f=3 h=i -de j=k -de | |
def parse_dict(str_in): | |
dic = {} | |
cur_dict = dic | |
layer = [dic] | |
for item in str_in: | |
slist = item.split('=') | |
if len(slist) == 2: #key-value | |
if '-ds' in slist[1]: | |
cur_dict[slist[0]] = {} | |
cur_dict = cur_dict[slist[0]] | |
layer.append(cur_dict) | |
else: | |
cur_dict[slist[0]] = slist[1] | |
elif len(slist) == 1 and '-de' in slist[0]: #end of key-value | |
layer.pop() | |
cur_dict = layer[-1] | |
return dic | |
def parse_cmd(str_in): | |
cmd_dict = {} | |
layers = [cmd_dict] | |
for s in str_in: | |
items = s.split('=') | |
if type(layers[-1]) == list: | |
if len(items) == 2: | |
if '-ds' in items[1]: # start of dict | |
d = {} | |
layers[-1].append(d) | |
layers.append(d) | |
elif '-ls' in items[1]: # start of list | |
l = [] | |
layers[-1].append(l) | |
layers.append(l) | |
else: | |
assert len(items) == 1 | |
if '-le' in items[0]: | |
layers.pop() | |
else: | |
layers[-1].append(items[0]) | |
elif type(layers[-1]) == dict: | |
if len(items) == 2: | |
if '-ds' in items[1]: # start of dict | |
layers[-1][items[0]] = {} | |
layers.append(layers[-1][items[0]]) | |
elif '-ls' in items[1]: # start of list | |
layers[-1][items[0]] = [] | |
layers.append(layers[-1][items[0]]) | |
else: | |
layers[-1][items[0]] = items[1] | |
else: | |
assert len(items) == 1 | |
if '-de' in items[0]: | |
layers.pop() | |
return cmd_dict | |
if __name__ in '__main__': | |
args = sys.argv[1:] | |
#args = in_str | |
print('input', args) | |
out_dict = parse_cmd(args) | |
print('out_dict', out_dict) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment