entrega.py

import re
import shlex
from glob import glob
from subprocess import Popen, PIPE





r = []

for f in glob('questao*.sql'):
	n = int(f[len('questao'):-len('.sql')])
	r.append((n, f))

r = sorted(r)

for n, f in r:
	p = Popen(shlex.split('mysql -u root --database=IMDB'), stdin=PIPE, stdout=PIPE, stderr=PIPE)
	sql = open(f).readlines()
	sql = ' '.join([l.strip() for l in sql if l.strip() and not l.strip().startswith('--')])
	sql = re.sub('\s+', ' ', sql)
	sql = sql.replace('( ', '(')
	sql = sql.replace(' )', ')')

	out, _ = p.communicate(sql)
	header, body = out.split('\n', 1)
	columns = header.count('\t') + 1
	rows = body.count('\n')

	print('#%d | %d | %d | %s' % (n, rows, columns, sql))