CSV to pretty table

tablify.py

#! /usr/bin/env python

import prettytable
import sys

if len(sys.argv) != 2:
    print "usage: tablify FILE.csv"
    exit(1)

data = []
with open(sys.argv[1]) as f:
    s = f.readline()
    while s != '':
        data.append(s[:-1].split(';'))
        s = f.readline()

if len(data) <= 1:
    print "Error: csv file too short, should be at least two lines"
    exit(1)

for d in data[1:]:
    if len(d) != len(data[0]):
        print "Error: Wrong number of values, should be " + str(len(data[0]))
        print ";".join(d)
        exit(1)

table = prettytable.PrettyTable(data[0])
for h in data[0]:
    table.align[h] = "l"
for d in data[1:]:
    table.add_row(d)
print table