import sys, shutil, os

args = sys.argv

if '--get-odb' in args:
    src = os.path.dirname(__file__) + '/omega_db.json'
    dest_dir = os.getcwd() + '/'
    fname = 'omega_db'
    suffix = '.json'
    v = 0
    while os.path.isfile(dest_dir + fname + str(v) + suffix):
        v += 1

    dest = dest_dir + fname + str(v) + suffix
    shutil.copy(src, dest)

if '-test' in args:
    '''
        Test suite for pykingas package
        Runs the tests placed in the list 'test_pkgs'
        Each element is a function that accepts the keyword arguments 'do_plot' and 'do_print'
        The function should return a single value, 0 for successfull tests, not 0 for failed tests.
    '''
    print('Testing from', __file__)
    try:
        import pyctp
    except:
        print('Missing module dependency pyctp (ThermoPack)')
        
    from pykingas import mie_unittests, collision_integral_unittests, py_unittest

    test_pkgs = [mie_unittests.run_tests, collision_integral_unittests.run_tests, py_unittest.run_test]

    r = 0
    for test in test_pkgs:
        if '-print' in args and '-plot' in args:
            r = test(do_print=True, do_plot=True)
        elif '-print' in args:
            r = test(do_print=True)
        elif '-plot' in args:
            r = test(do_plot=True)
        else:
            r = test()
        if r != 0:
            exit(r)
    exit(r)