"""新增采样申请组别；历史记录保留空白，可重复执行。测试环境使用 --local。"""
import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
if '--local' in sys.argv:
    import test_wsgi

from services.db import query


def migrate():
    exists = query(
        'SELECT COLUMN_NAME FROM information_schema.COLUMNS '
        'WHERE TABLE_SCHEMA=DATABASE() AND TABLE_NAME=%s AND COLUMN_NAME=%s',
        ('spot_applications', 'group_name'),
    )
    if not exists:
        query("ALTER TABLE spot_applications ADD COLUMN group_name VARCHAR(30) NOT NULL DEFAULT ''")
    print('spot_applications.group_name: ready')


if __name__ == '__main__':
    migrate()