comparison doc/check-seclevel.py @ 27770:1b8c7d59be43

check-seclevel: use a context manager for file I/O
author Bryan O'Sullivan <bryano@fb.com>
date Tue, 12 Jan 2016 14:28:16 -0800
parents 44a596a8bed1
children 98153441c8cc
comparison
equal deleted inserted replaced
27769:40bd01be5c25 27770:1b8c7d59be43
105 def checkfile(ui, filename, initlevel): 105 def checkfile(ui, filename, initlevel):
106 if filename == '-': 106 if filename == '-':
107 filename = 'stdin' 107 filename = 'stdin'
108 doc = sys.stdin.read() 108 doc = sys.stdin.read()
109 else: 109 else:
110 fp = open(filename) 110 with open(filename) as fp:
111 try:
112 doc = fp.read() 111 doc = fp.read()
113 finally:
114 fp.close()
115 112
116 ui.note(('checking input from %s with initlevel %d\n') % 113 ui.note(('checking input from %s with initlevel %d\n') %
117 (filename, initlevel)) 114 (filename, initlevel))
118 return checkseclevel(ui, doc, 'input from %s' % filename, initlevel) 115 return checkseclevel(ui, doc, 'input from %s' % filename, initlevel)
119 116