Chinaunix

标题: python自动格式化json文件 [打印本页]

---

作者: mahaijiang    时间: 2015-07-08 14:05
标题: python自动格式化json文件
将代码混乱的json文件格式化。
还有一小堆python常用算法代码
项目地址：https://github.com/xsank/cabbird
[Python]代码

1. class JsonFormatter:
   
2.     def __init__(self,intend=4,name=""):
   
3.         self.name=name
   
4.         self.intend=intend
   
5.         self.stack=[]
   
6.         self.obj=None
   
7.         self.source=self.get_source(name)
   
8.         self.prepare()
   
9. 
   
10.     @staticmethod
    
11.     def json_str(s):
    
12.         return '"'+s+'"'
    
13. 
    
14.     @staticmethod
    
15.     def get_source(name):
    
16.         with open(name,'r') as f:
    
17.             return ''.join(f.read().split())
    
18. 
    
19.     def prepare(self):
    
20.         try:
    
21.             self.obj=eval(self.source)
    
22.         except:
    
23.             raise Exception('Invalid json string!')
    
24. 
    
25.     def line_intend(self,level=0):
    
26.         return '\n'+' '*self.intend*level
    
27. 
    
28.     def parse_dict(self,obj=None,intend_level=0):
    
29.         self.stack.append(self.line_intend(intend_level)+'{')
    
30.         intend_level+=1
    
31.         for key,value in obj.items():
    
32.             key=self.json_str(str(key))
    
33.             self.stack.append(self.line_intend(intend_level)+key+':')
    
34.             self.parse(value,intend_level)
    
35.             self.stack.append(',')
    
36.         self.stack.append(self.line_intend(intend_level-1)+'}')
    
37. 
    
38.     def parse_list(self,obj=None,intend_level=0):
    
39.         self.stack.append(self.line_intend(intend_level)+'[')
    
40.         intend_level+=1
    
41.         for item in obj:
    
42.             self.parse(item,intend_level)
    
43.             self.stack.append(',')
    
44.         self.stack.append(self.line_intend(intend_level-1)+']')
    
45. 
    
46.     def parse(self,obj,intend_level=0):
    
47.         if obj is None:
    
48.             self.stack.append('null')
    
49.         elif obj is True:
    
50.             self.stack.append('true')
    
51.         elif obj is False:
    
52.             self.stack.append('false')
    
53.         elif isinstance(obj,(int,long,float)):
    
54.             self.stack.append(str(obj))
    
55.         elif isinstance(obj,str):
    
56.             self.stack.append(self.json_str(obj))
    
57.         elif isinstance(obj,(list,tuple)):
    
58.             self.parse_list(obj,intend_level)
    
59.         elif isinstance(obj,dict):
    
60.             self.parse_dict(obj,intend_level)
    
61.         else:
    
62.             raise Exception('Invalid json type %s!' % obj)
    
63. 
    
64.     def render(self):
    
65.         self.parse(self.obj,0)
    
66.         res_file='good'+self.name
    
67.         res=''.join(self.stack)
    
68.         with open(res_file,'w') as f:
    
69.             f.write(res)
    
70.         print res
    
71. 
    
72. if __name__=="__main__":
    
73.     jf=JsonFormatter(name="json.txt")
    
74.     jf.render()
    

复制代码

---

欢迎光临 Chinaunix (http://bbs.chinaunix.net/)

Powered by Discuz! X3.2