runme

#!/usr/bin/env python

import ql
import os
import ql.prepr
import ql.perf
import ql.migrate
from ql.pg import psql
import sys
import glob
import re

if 'DB' in os.environ:
    db = os.environ['DB']
else:
    print "Please set env DB=?"
    exit(1)

def each_file(fls,cb):
    for fl in sorted(fls):
        if os.path.isfile(fl):
            cb(fl)
        else:
            print 'Could not fild file %s' % fl

def run_tests(args):
    load(['./src/tests.sql'])
    if not args:
        args = glob.glob('./test/*_spec.sql')

    psql(db, 'SELECT tests.start()')
    def run(fl):
        ql.reload_test(db, fl)

    each_file(args,run)
    res = psql(db, 'SELECT tests.fail()')
    if res['returncode'] != 0:
        print ql.hl('red','FAILED TESTS')
        report = psql(db, 'SELECT * from tests.report()')['stdout']
        print ql.hl('red', report)
        sys.exit(1)
    else:
        print ql.hl('green','All tests passed')

def compile(args):
    def run(fl):
        f = open(fl)
        print '-- FILE: %s' % fl
        print ql.prepr.process(fl, f.read())
        f.close
    each_file(args,run)

def reload(args):
    if not args: args = glob.glob('./src/**/*.sql')
    ql.reload_files(db,args)

def load(args):
    if not args: args = glob.glob('./src/**/*.sql')
    def run(fl): ql.reload(db, fl)
    each_file(args,run)

def integrate(args):
    print psql('postgres','drop database if exists %s' % db)['stdout']
    print psql('postgres','create database %s' % db)['stdout']
    migrate(args)
    run_tests([])

def build(args):
    integrate(args)
    ql.pgdump(db)

def install(args):
    print 'TODO'


def migrate(args):
    ql.migrate.migrate(db)

def seed(args):
    if not 'PATIENTS_COUNT' in os.environ:
        print "Please set required PATIENTS_COUNT environment variable"
        exit(1)

    if not 'RAND_SEED' in os.environ:
        os.environ['RAND_SEED'] = '0.21'

    r = psql(db, """\\conninfo
                    SELECT fhir.generate_tables('{Patient,Organization,Encounter,Practitioner}');
""")

    perf_files = glob.glob('./perf/*.sql')
    reload(perf_files)

    r = psql(db, """\\timing
                 SELECT generator.generate(%(patients_count)s, %(rand_seed)s)
                 """ \
                 % {'patients_count': os.environ['PATIENTS_COUNT'],
                    'rand_seed': os.environ['RAND_SEED']})

    print r["stdout"]

def perf(args):
    ql.perf.perf(db, args)

def rename(args):
    mod_nm = args[0]
    new_nm = args[1]
    print '%s -> %s' % (mod_nm, new_nm)

commands = dict(test=run_tests, compile=compile, load=load, reload=reload, integrate=integrate, build=build, migrate=migrate, seed=seed, perf=perf, rename=rename)

if len(sys.argv) < 2:
    print "Please use command:"
    print commands.keys()
    exit(1)

cmd = sys.argv[1]


print cmd
if cmd in commands:
    print 'run command %s' % cmd
    commands[cmd](sys.argv[2:])
else:
    print 'Unknown command %s' % cmd
    print commands