30 lines
1.0 KiB
Python
30 lines
1.0 KiB
Python
"""首次初始化部门规则，默认只检查，--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()
|