125 lines
4.6 KiB
Python
125 lines
4.6 KiB
Python
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()
|