#!/usr/bin/env python

# Makes a PostgreSQL CREATE TABLE statement from a CSV header

import csv

import os.path

import sys

sys.path.append(os.path.dirname(__file__)+"/../lib")

import csvs

import opts

import sql

def esc_name(name):

    return sql.esc_name_by_engine('PostgreSQL', name, preserve_case=True)

def main():

    # Usage

    env_names = []

    def usage_err():

        raise SystemExit('Usage: '+opts.env_usage(env_names)+' '+sys.argv[0]

            +' <header >ddl')

    # Get config from env vars

    schema = opts.get_env_var('schema', None, env_names)

    table = opts.get_env_var('table', None, env_names)

    if schema == None or table == None: usage_err()

    # Get col names

    reader, col_names = csvs.reader_and_header(sys.stdin)

    # Write CREATE TABLE statement

    out = sys.stdout

    out.write('CREATE TABLE '+esc_name(schema)+'.'+esc_name(table)+' (\n')

    out.write('row_num serial NOT NULL\n')

    for col_name in col_names: out.write(', '+esc_name(col_name)+' text\n')

    out.write(');\n')

main()