from pathlib import Path
import sys
import re
import json

ROOT = Path(__file__).resolve().parents[2]
sys.stdout.reconfigure(encoding='utf-8')
sys.path.insert(0, str(ROOT / 'scripts/iqc-import-20261012/deps'))
import pymysql

def connect():
    config = (ROOT / 'yudao-server/src/main/resources/application-trt.yaml').read_text(encoding='utf-8')
    match = re.search(r'url: jdbc:mysql:[\s\S]*?\n\s+username:\s*(\S+)[\s\S]*?\n\s+password:\s*([^\r\n#]+)', config)
    return pymysql.connect(host='36.213.150.237', port=10000, user=match.group(1),
                           password=match.group(2).strip().strip('\"\x27'), database='mom-hm',
                           charset='utf8mb4', connect_timeout=15, read_timeout=120,
                           cursorclass=pymysql.cursors.DictCursor, autocommit=False)

if __name__ == '__main__':
    with connect() as connection:
        with connection.cursor() as cursor:
            for sql in [
                'SELECT DATABASE() db, NOW() db_now, @@session.time_zone tz',
                'SHOW COLUMNS FROM system_operate_log',
                'SELECT deleted,tenant_id,COUNT(*) count,MIN(create_time) earliest,MAX(create_time) latest FROM system_operate_log GROUP BY deleted,tenant_id',
                'SELECT DATE(create_time) day,COUNT(*) count FROM system_operate_log WHERE deleted=0 GROUP BY DATE(create_time) ORDER BY day',
            ]:
                cursor.execute(sql)
                print(json.dumps({'sql': sql, 'rows': cursor.fetchall()}, ensure_ascii=False, default=str))