Created
November 9, 2019 14:22
-
-
Save benwu232/6cd5a8ca514bec885ebc43d46a5328e7 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