"""首次初始化部门规则，默认只检查，--apply 才写入。"""
import argparse
import json
import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parents[1]))


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('workbook')
    parser.add_argument('--apply', action='store_true')
    parser.add_argument('--local', action='store_true')
    args = parser.parse_args()
    if args.local:
        import config
        config.DB_CONFIG.update(user='dy_test', password=Path('.local-test-db-password').read_text().strip())
    from services.department_matching import workbook_records, import_workbook, check_samples
    rows = workbook_records(args.workbook)
    for i, row in enumerate(rows, 1):
        row['id'] = i
    print(json.dumps({'records': len(rows), 'validation': check_samples(rows)}, ensure_ascii=False))
    if args.apply:
        print(json.dumps({'imported': import_workbook(args.workbook)}, ensure_ascii=False))


if __name__ == '__main__':
    main()