pgmigrate.py 文件源码

def _set_baseline(baseline_v, user, cursor):
    """
    Cleanup schema_version and set baseline
    """
    cursor.execute('SELECT EXISTS(SELECT 1 FROM public'
                   '.schema_version WHERE version >= %s::bigint)',
                   (baseline_v,))
    check_failed = cursor.fetchone()[0]

    if check_failed:
        raise BaselineError(
            'Unable to baseline, version '
            '{version} already applied'.format(version=text(baseline_v)))

    LOG.info('cleaning up table schema_version')
    cursor.execute('DELETE FROM public.schema_version')
    LOG.info(cursor.statusmessage)

    LOG.info('setting baseline')
    cursor.execute('INSERT INTO public.schema_version '
                   '(version, type, description, installed_by) '
                   'VALUES (%s::bigint, %s, %s, %s)',
                   (text(baseline_v), 'manual',
                    'Forced baseline', user))
    LOG.info(cursor.statusmessage)