cross_doctor/proto/phase_grpc.py

import logging
import os
import sys
from typing import List

import grpc

from app.global_source import g_config

sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../proto')))
from proto import phase_server_pb2_grpc, phase_server_pb2


def channel_stub():
    channel = grpc.insecure_channel(f"{g_config['rpc_host']}:{g_config['rpc_port']}")
    stub = phase_server_pb2_grpc.PhaseServiceStub(channel)
    return stub, channel


def get_exception_phase_cross():
    stub, channel = channel_stub()
    try:
        return stub.ExceptionPhaseCross(phase_server_pb2.EmptyRequest(), timeout=30)
    except Exception as e:
        logging.error(e)
        return phase_server_pb2.ExceptionPhaseCrossResponse()
    finally:
        channel.close()


def get_exception_phase_cross_info(citycode):
    stub, channel = channel_stub()
    try:
        request_params = phase_server_pb2.ExceptionPhaseCrossInfoRequest(citycode=citycode)
        return stub.ExceptionPhaseCrossInfo(request_params, timeout=30), None
    except Exception as e:
        return None, e
    finally:
        channel.close()


def get_green_cross_phase(citycode: int = 0, crossids: List[str] = []):
    stub, channel = channel_stub()
    try:
        request_params = phase_server_pb2.ExceptionPhaseByCrossIDsRequest(citycode=citycode, crossid=crossids)
        return stub.ExceptionPhaseByCrossIDs(request_params, timeout=30), None
    except Exception as e:
        return None, e
    finally:
        channel.close()


def GetCrossPhaseDetailByCrossIDs(citycode: int = 0, crossids: List[str] = [], gw_src_dir='', tp_start='', tp_end='',
                                  week='', wave_id=''):
    stub, channel = channel_stub()
    try:
        request_params = phase_server_pb2.GetCrossPhaseDetailByCrossIDsRequest(citycode=citycode, crossids=crossids,
                                                                               gw_src_dir=gw_src_dir, tp_start=tp_start,
                                                                               tp_end=tp_end, week=week,
                                                                               wave_id=wave_id)
        return stub.GetCrossPhaseDetailByCrossIDs(request_params, timeout=30), None
    except Exception as e:
        return None, e
    finally:
        channel.close()


#获取有配时方案路口ID
def GetPhaseCrossID():
    stub, channel = channel_stub()
    try:
        return stub.GetPhaseCrossID(phase_server_pb2.EmptyRequest(), timeout=30), None
    except Exception as e:
        return None, e
    finally:
        channel.close()


def GetPhaseGreenRatio(citycode: int = 0, crossids: List[str] = [], gw_src_dir='', tp_start='', tp_end='', week='',
                       wave_id=''):
    stub, channel = channel_stub()
    try:
        request_params = phase_server_pb2.PhaseGreenRatioRequest(citycode=citycode, crossids=crossids,
                                                                 gw_src_dir=gw_src_dir, tp_start=tp_start,
                                                                 tp_end=tp_end, week=week, wave_id=wave_id)
        return stub.PhaseGreenRatio(request_params, timeout=30), None
    except Exception as e:
        return None, e
    finally:
        channel.close()


def GetQueryCrossPhaseRelativeOffset(citycode: int = 0, crossids: List[str] = [], tp: str = '', day: str = ''):
    stub, channel = channel_stub()
    try:
        request_params = phase_server_pb2.QueryCrossPhaseRelativeOffsetRequest(citycode=citycode, crossids=crossids,
                                                                               tp=tp, day=day)
        return stub.QueryCrossPhaseRelativeOffset(request_params, timeout=30), None
    except Exception as e:
        return None, e
    finally:
        channel.close()

#更新单绿波配时方案时段与路口配时方案时段对比异常状态
def QueryGreenWaveCrossPhaseTpCheck(waveid: str, citycode: int):
    stub, channel = channel_stub()
    try:
        request_params = phase_server_pb2.GreenWaveCrossPhaseTpCheckRequest(waveid=waveid, citycode=citycode)
        return stub.GreenWaveCrossPhaseTpCheck(request_params, timeout=30), None
    except Exception as e:
        return None, e
    finally:
        channel.close()


def QueryCrossRunningPhase(citycode: int, crossids: []):
    stub, channel = channel_stub()
    try:
        request_params = phase_server_pb2.CrossRunningPhaseRequest(citycode=citycode, crossids=crossids)
        return stub.CrossRunningPhase(request_params, timeout=30), None
    except Exception as e:
        return None, e
    finally:
        channel.close()