utils.py 文件源码

def pg_load(file_location):
    env = os.environ.copy()
    env.update({  # requires having password set to test
        "PGPASSWORD": settings.DATABASES['default']['PASSWORD']
    })
    load = [
        'psql',
        '-h%s' % settings.DATABASES['default']['HOST'],
        '-U%s' % settings.DATABASES['default']['USER'],
        '-d%s' % settings.DATABASES['default']['NAME'],
        '-f%s' % file_location,
    ]
    task = Popen(load, env=env)
    return task.wait()