From 4abacdb0b15b58816881b083f8fb86bbf2272bb1 Mon Sep 17 00:00:00 2001
From: yinzijian <yinzijian@haomozhixing.onaliyun.com>
Date: Sun, 26 Oct 2025 12:13:03 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E9=85=8D=E6=97=B6=E6=96=B9?=
 =?UTF-8?q?=E6=A1=88?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: yinzijian <yinzijian@haomozhixing.onaliyun.com>
---
 proto/phase_grpc.py | 124 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 124 insertions(+)
 create mode 100644 proto/phase_grpc.py

diff --git a/proto/phase_grpc.py b/proto/phase_grpc.py
new file mode 100644
index 0000000..3a29f39
--- /dev/null
+++ b/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()