python/retrace: Support gziped traces.

This commit is contained in:
José Fonseca
2008-08-21 18:45:43 +01:00
parent 34d12c1787
commit 10624065b0
2 changed files with 14 additions and 3 deletions

View File

@@ -471,7 +471,12 @@ class Interpreter:
def main():
for arg in sys.argv[1:]:
parser = TraceParser(open(arg, 'rt'))
if arg.endswith('.gz'):
import gzip
stream = gzip.GzipFile(arg, 'rt')
else:
stream = open(arg, 'rt')
parser = TraceParser(stream)
trace = parser.parse()
interpreter = Interpreter()
interpreter.interpret(trace)

View File

@@ -322,9 +322,15 @@ class TraceParser(XmlParser):
def main():
for arg in sys.argv[1:]:
parser = TraceParser(open(arg, 'rt'))
if arg.endswith('.gz'):
import gzip
stream = gzip.GzipFile(arg, 'rt')
else:
stream = open(arg, 'rt')
parser = TraceParser(stream)
trace = parser.parse()
print trace
for call in trace.calls:
print call
if __name__ == '__main__':