Server-py/sri-server-bg01/api/v6/code3000.py

from hub import methods, Global

import os

save_dir = r'C:\logs'  # sri内网测试环境


async def code3001(**sources):
    """
    查询驾驶人员操作记录列表
    """
    # --- check ---
    if not sources.get('page'):
        return dict(code=1, details=f"something is wrong.")
    elif not sources.get('size'):
        return dict(code=2, details=f"something is wrong.")

    # --- get log_file_uuid_list ---
    log_file_list = methods.get_file_path_list(save_dir)
    log_file_name_list = [i.split('\\')[-1] for i in log_file_list]
    log_file_uuid_list = [i.split('.')[0] for i in log_file_name_list]

    # --- fill d2 ---
    d2 = dict()
    """
    VehicleInfo: 车辆信息表
    VehicleInfo.pid: 车牌号
    VehicleInfo.type: 车型号
    VehicleInfo.host_address: 工控机地址
    VehicleInfo.rtk_address: rtk地址
    VehicleInfo.cpe_address: cpe地址
    VehicleInfo.release_at: 出厂时期
    VehicleInfo.state: 当前状态 （1:离线 2:在线空闲 3: 现场驾驶中 4: 远程驾驶中）
    VehicleInfo.permit_state: 遥操状态 （0:允许 1:禁用）
    VehicleInfo.update_at: 更新时间
    """
    for item in Global.mdb.get_all('VehicleInfo'):
        ipv4 = int(f"{item.get('host_address').replace('.', '')}")
        d2[ipv4] = item.get('pid')

    # --- fill d1 ---
    """
    UserWorkRecordList: 用户操作记录
    UserWorkRecordList.uuid: 记录标识
    UserWorkRecordList.user_uuid: 用户标识
    UserWorkRecordList.start_at: 操作开始时间
    UserWorkRecordList.end_at: 操作结束时间
    UserWorkRecordList.cockpit_id: 舱端id
    UserWorkRecordList.vehicle_id: 车端id
    """
    d1 = list()
    page = sources.get('page', 1)
    size = sources.get('size', 10)
    for item in Global.mdb.get_all('UserWorkRecordList', sort_field=[('start_at', -1)]):
        # --- check ---
        uuid = str(item.get('_id'))
        if uuid not in log_file_uuid_list:
            continue

        # --- update ---
        item['uuid'] = uuid
        item['start_time_at'] = item.get('start_at')
        item['end_time_at'] = item.get('end_at')
        item['cockpit_name'] = item.get('cockpit_id')

        # --- check ---
        item['driver_name'] = item.get('user_uuid')
        if len('67174cf9cd36d7a6def99cae') == len(item.get('user_uuid')):
            item['driver_name'] = Global.mdb.get_one_by_id('UserInfo', item.get('user_uuid')).get('name')

        # --- check ---
        item['pid'] = item.get('vehicle_id')
        if int(item.get('vehicle_id')) in d2:
            item['pid'] = d2.get(int(item.get('vehicle_id')))

        # --- update --
        del item['_id']
        del item['user_uuid']
        del item['start_at']
        del item['end_at']
        del item['cockpit_id']
        del item['vehicle_id']
        d1.append(item)

    # d1 = [
    #     {
    #         'uuid': '65dbe96949fbe311a3a01d30',
    #         'pid': 'AA112233',  # 车牌号
    #         'start_time_at': methods.now_ts(),
    #         'end_time_at': methods.now_ts() + (3600 * 8),
    #         'driver_name': '张三',
    #         'cockpit_name': '1号舱',
    #     }, {
    #         'uuid': '65dbe96949fbe311a3a01d30',
    #         'pid': 'AA112233',
    #         'start_time_at': methods.now_ts(),
    #         'end_time_at': methods.now_ts() + (3600 * 8),
    #         'driver_name': '李四',
    #         'cockpit_name': '1号舱',
    #     }, {
    #         'uuid': '65dbe96949fbe311a3a01d30',
    #         'pid': 'AA112233',
    #         'start_time_at': methods.now_ts(),
    #         'end_time_at': methods.now_ts() + (3600 * 8),
    #         'driver_name': '李四',
    #         'cockpit_name': '1号舱',
    #     }, {
    #         'uuid': '65dbe96949fbe311a3a01d30',
    #         'pid': 'AA112233',
    #         'start_time_at': methods.now_ts(),
    #         'end_time_at': methods.now_ts() + (3600 * 8),
    #         'driver_name': '李四',
    #         'cockpit_name': '1号舱',
    #     }, {
    #         'uuid': '65dbe96949fbe311a3a01d30',
    #         'pid': 'AA112233',
    #         'start_time_at': methods.now_ts(),
    #         'end_time_at': methods.now_ts() + (3600 * 8),
    #         'driver_name': '李四',
    #         'cockpit_name': '1号舱',
    #     }, {
    #         'uuid': '65dbe96949fbe311a3a01d30',
    #         'pid': 'AA112233',
    #         'start_time_at': methods.now_ts(),
    #         'end_time_at': methods.now_ts() + (3600 * 8),
    #         'driver_name': '李四',
    #         'cockpit_name': '1号舱',
    #     }, {
    #         'uuid': '65dbe96949fbe311a3a01d30',
    #         'pid': 'AA112233',
    #         'start_time_at': methods.now_ts(),
    #         'end_time_at': methods.now_ts() + (3600 * 8),
    #         'driver_name': '李四',
    #         'cockpit_name': '1号舱',
    #     }, {
    #         'uuid': '65dbe96949fbe311a3a01d30',
    #         'pid': 'AA112233',
    #         'start_time_at': methods.now_ts(),
    #         'end_time_at': methods.now_ts() + (3600 * 8),
    #         'driver_name': '李四',
    #         'cockpit_name': '1号舱',
    #     }, {
    #         'uuid': '65dbe96949fbe311a3a01d30',
    #         'pid': 'AA112233',
    #         'start_time_at': methods.now_ts(),
    #         'end_time_at': methods.now_ts() + (3600 * 8),
    #         'driver_name': '李四',
    #         'cockpit_name': '1号舱',
    #     }, {
    #         'uuid': '65dbe96949fbe311a3a01d30',
    #         'pid': 'AA112233',
    #         'start_time_at': methods.now_ts(),
    #         'end_time_at': methods.now_ts() + (3600 * 8),
    #         'driver_name': '李四',
    #         'cockpit_name': '1号舱',
    #     }
    # ]
    return dict(code=0, data=d1[(page - 1) * size: page * size], total=len(d1), page=page, size=size)


async def code3002(**sources):
    """
    下载指定驾驶人员操作日志
    """
    if not sources.get('uuid'):
        return dict(code=1, details=f"not found uuid.")
    # log_uuid = '671754598302fc6fd344819f'
    # if sources.get('uuid'):
    #     log_uuid = sources.get('uuid')
    methods.debug_log('code3000.code3002|145:', f"#log_uuid: {sources.get('uuid')}")
    file_path = os.path.join(save_dir, f"{sources.get('uuid')}.log")

    # --- get file_name ---
    item = Global.mdb.get_one_by_id('UserWorkRecordList', sources.get('uuid'))
    start_at = methods.ts_to_string(item.get('start_at'), '%Y%m%d%H%M%S')
    end_at = methods.ts_to_string(item.get('end_at'), '%Y%m%d%H%M%S')
    file_name = f"{start_at}-{end_at}.log"
    return {'file_path': file_path, 'file_name': file_name}