csv_2_latex_longtable.py

#!/usr/bin/env python
# A little script that helps convert csv files into a latex longtable appendix file.

import sys
import csv

# Title, Letter, Column_count, Title, Title, Letter, Columns, Col Count, Columns, Col count, footer, Col count, footer
longtable_header = """%% !TEX root = paper_that_includes_this_appendix.tex

\\chapter{%s}
\\label{app:appendix%s}

\\begin{landscape}
\\begin{longtable}{%s}
    \\caption[%s]{%s}
    \\label{t:table%s} \\\\
    \\hline
    %s \\\\
    \\hline
\\endfirsthead
    \\multicolumn{%d}{c}{{\\bfseries \\tablename\\ \\thetable{} -- continued from previous page}} \\\\
    \\hline
    %s \\\\
    \\hline
\\endhead
    \\hline
    \\multicolumn{%d}{|r|}{{
        %s
    }} \\\\
    \\hline
\\endfoot
    \\hline
    \\multicolumn{%d}{|r|}{{
        %s
    }} \\\\
    \\hline
\\endlastfoot
"""

longtable_footer = """
\\end{longtable}
\\end{landscape}
"""

def gen_longtable_header(title, letter, header_columns, footer):
    column_types = "|"
    longtable_header_columns = ""
    for col in header_columns:
        column_types += "l|"
        longtable_header_columns += col + " & "

    longtable_header_columns = longtable_header_columns[:-2]
    return longtable_header % (
            title,
            letter,
            column_types,
            title,
            title,
            letter,
            longtable_header_columns,
            len(header_columns),
            longtable_header_columns,
            len(header_columns),
            footer,
            len(header_columns),
            footer)

def gen_longtable_footer():
    return longtable_footer


def gen_longtable(table_name, appendix_letter, csv_name, footer_text):
    with open(csv_name, 'r') as csvfile:
        stats = csv.reader(csvfile)
        header = next(stats)
        print(gen_longtable_header(table_name, appendix_letter, header, footer_text))
        for row in stats:
            print " & ".join(row) + " \\\\"
        print(gen_longtable_footer())

if __name__ == "__main__":
    gen_longtable(sys.argv[1], sys.argv[2], sys.argv[3], sys.argv[4])