新增配时方案

Signed-off-by: yinzijian <yinzijian@haomozhixing.onaliyun.com>

proto/phase_grpc.py

@@ -0,0 +1,124 @@
+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()