ep-sdk-4pd 0.1.0__tar.gz → 0.1.2__tar.gz

{ep_sdk_4pd-0.1.0 → ep_sdk_4pd-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ep_sdk_4pd
-Version: 0.1.0
+Version: 0.1.2
 Summary: 4paradigm Electricity Platform Service SDK Library for Python
 Home-page: https://gitlab.4pd.io/electricityproject/electricity-platform-sdk
 Author: 4paradigm Electricity Platform SDK

ep_sdk_4pd-0.1.2/ep_sdk_4pd/__init__.py

@@ -0,0 +1 @@
+__version__ = '0.1.2'

ep_sdk_4pd-0.1.2/ep_sdk_4pd/ep_data.py

@@ -0,0 +1,108 @@
+import json
+
+import requests
+
+from ep_sdk_4pd import models as ep_sdk_4pd_models
+from ep_sdk_4pd.models import HistoryDataRequest, PredictDataRequest
+
+# test 地址
+endpoint = 'http://172.27.88.56:6001'
+
+
+# prod 地址
+# endpoint = 'http://82.157.231.254:6001'
+
+class EpData:
+
+    @staticmethod
+    def get_history_data(
+            scope,
+            system_date,
+            days
+    ):
+        request = HistoryDataRequest(
+            scope=scope,
+            system_date=system_date,
+            days=days,
+        )
+        response = EpData.history_data(request=request)
+
+        if response.code == 200:
+            return response.data
+        else:
+            return None
+
+    @staticmethod
+    def history_data(
+            request: ep_sdk_4pd_models.HistoryDataRequest = None,
+    ) -> ep_sdk_4pd_models.HistoryDataResponse:
+
+        full_url = f'{endpoint}{request.api}'
+        headers = {
+            'content-type': request.content_type,
+        }
+
+        payload = {
+            'scope': request.scope,
+            'system_date': request.system_date,
+            'days': request.days
+        }
+
+        response = requests.request(
+            method=request.method,
+            url=full_url,
+            headers=headers,
+            data=json.dumps(payload),
+        )
+
+        base_resp = ep_sdk_4pd_models.BaseResponse(
+            code=response.json().get('code', None),
+            data=response.json().get('data', None),
+            message=response.json().get('message', None),
+        )
+        return ep_sdk_4pd_models.HistoryDataResponse(response=base_resp)
+
+    @staticmethod
+    def get_predict_data(
+            scope,
+            system_date
+    ):
+        request = PredictDataRequest(
+            scope=scope,
+            system_date=system_date,
+        )
+        response = EpData.predict_data(request=request)
+
+        if response.code == 200:
+            return response.data
+        else:
+            return None
+
+    @staticmethod
+    def predict_data(
+            request: ep_sdk_4pd_models.PredictDataRequest = None,
+    ) -> ep_sdk_4pd_models.PredictDataResponse:
+
+        full_url = f'{endpoint}{request.api}'
+        headers = {
+            'content-type': request.content_type,
+        }
+
+        payload = {
+            'scope': request.scope,
+            'system_date': request.system_date
+        }
+
+        response = requests.request(
+            method=request.method,
+            url=full_url,
+            headers=headers,
+            data=json.dumps(payload),
+        )
+
+        base_resp = ep_sdk_4pd_models.BaseResponse(
+            code=response.json().get('code', None),
+            data=response.json().get('data', None),
+            message=response.json().get('message', None),
+        )
+        return ep_sdk_4pd_models.PredictDataResponse(response=base_resp)

ep_sdk_4pd-0.1.2/ep_sdk_4pd/ep_system.py

@@ -0,0 +1,121 @@
+import json
+import logging
+import os
+
+import requests
+
+from ep_sdk_4pd import models as ep_sdk_4pd_models
+from datetime import datetime
+
+from ep_sdk_4pd.models import ModelOutputDirRequest, RunStrategyRequest
+
+# test 地址
+endpoint = 'http://172.27.88.56:6001'
+
+
+# prod 地址
+# endpoint = 'http://82.157.231.254:6001'
+
+class EpSystem:
+
+    @staticmethod
+    def model_output_dir(is_online: bool = False):
+        if is_online:
+            # 从环境变量中获取策略id
+            strategy_id = os.getenv('STRATEGY_ID')
+            logging.info(f'strategy_id: {strategy_id}')
+
+            if strategy_id is None:
+                raise Exception('STRATEGY_ID is not set')
+        else:
+            # 线下环境，给固定的策略id
+            strategy_id = 63
+        request = ModelOutputDirRequest(strategy_id=strategy_id)
+        full_url = f'{endpoint}{request.api}'
+        headers = {
+            'content-type': request.content_type,
+        }
+
+        payload = {
+            'strategy_id': request.strategy_id
+        }
+
+        response = requests.request(
+            method=request.method,
+            url=full_url,
+            headers=headers,
+            data=json.dumps(payload),
+        )
+
+        base_resp = ep_sdk_4pd_models.BaseResponse(
+            code=response.json().get('code', None),
+            data=response.json().get('data', None),
+            message=response.json().get('message', None),
+        )
+        response = ep_sdk_4pd_models.ModelOutputDirResponse(response=base_resp)
+
+        if response.code == 200:
+            return response.data
+        else:
+            return None
+
+    @staticmethod
+    def get_system_date(is_online: bool = False):
+        """
+        脚本每次预测的目标日，主要是拦截用户获取越界数据,用户不可修改
+        """
+        if is_online:
+            # 线上环境,随着真实调用运行时间变化
+            system_date = datetime.now().strftime('%Y-%m-%d')
+        else:
+            # 线下环境
+            system_date = "2024-12-31"
+
+        return system_date
+
+    @staticmethod
+    def get_run_strategy(is_online: bool = False):
+        """
+        获取此刻运行的策略模型基础信息
+        :param is_online:
+        :return:
+        """
+        if is_online:
+            # 从环境变量中获取策略id
+            strategy_id = os.getenv('STRATEGY_ID')
+            logging.info(f'strategy_id: {strategy_id}')
+
+            if strategy_id is None:
+                raise Exception('STRATEGY_ID is not set')
+        else:
+            # 线下环境，给固定的策略id
+            strategy_id = 56
+
+        request = RunStrategyRequest(strategy_id=strategy_id)
+        full_url = f'{endpoint}{request.api}'
+        headers = {
+            'content-type': request.content_type,
+        }
+
+        payload = {
+            'strategy_id': request.strategy_id
+        }
+
+        response = requests.request(
+            method=request.method,
+            url=full_url,
+            headers=headers,
+            data=json.dumps(payload),
+        )
+
+        base_resp = ep_sdk_4pd_models.BaseResponse(
+            code=response.json().get('code', None),
+            data=response.json().get('data', None),
+            message=response.json().get('message', None),
+        )
+        response = ep_sdk_4pd_models.RunStrategyResponse(response=base_resp)
+
+        if response.code == 200:
+            return response.data
+        else:
+            return None

ep_sdk_4pd-0.1.2/ep_sdk_4pd/models.py

@@ -0,0 +1,171 @@
+class BaseRequest:
+    """
+    Model for BaseRequest
+    """
+
+    def __init__(self):
+        self.api = None
+        self.method = None
+        self.content_type = None
+        self.payload = None
+
+
+class BaseResponse:
+    """
+    Model for BaseResponse
+    """
+
+    def __init__(
+            self, code: int = None, data: dict = None, message: str = None, **kwargs
+    ):
+        self.code = code
+        self.data = data
+        self.message = message
+
+
+class HistoryDataRequest(BaseRequest):
+    """
+    Model for HistoryDataRequest
+
+    获取具体系统时间往前x天的数据（最晚时间为系统时间 D-2）
+    特别注意：get_history_data 或 get_predict_data 会get_system_date进行检测，不可获取system_date以后得数据，避免数据穿越现象
+    """
+
+    def __init__(self, scope: str = None, system_date: str = None, days: int = None):
+        """
+        Args:
+            scope: "weather","plant","market"
+            system_date: 系统时间
+            days: 表示systemDate之前days的数据
+        """
+
+        self.scope = scope
+        self.system_date = system_date
+        self.days = days
+
+        super().__init__()
+        self.api = f'/ep/api/sdk/get_history_data'
+        self.method = 'POST'
+        self.content_type = 'application/json'
+
+
+class HistoryDataResponse(BaseResponse):
+    """
+    Model for HistoryDataResponse
+    """
+
+    def __init__(self, response: BaseResponse = None, **kwargs):
+        super().__init__(
+            code=response.code if response else None,
+            data=response.data if response else None,
+            message=response.message if response else None,
+            **kwargs,
+        )
+
+class PredictDataRequest(BaseRequest):
+    """
+    Model for PredictDataRequest
+
+    获取当前系统时间的预测数据（按照抓取时间，最晚时间为当前系统时间的d-1
+    """
+
+    def __init__(self, scope: str = None, system_date: str = None):
+        """
+        Args:
+            scope: "weather","market"
+            system_date: 系统时间
+        """
+
+        self.scope = scope
+        self.system_date = system_date
+
+        super().__init__()
+        self.api = f'/ep/api/sdk/get_predict_data'
+        self.method = 'POST'
+        self.content_type = 'application/json'
+
+
+class PredictDataResponse(BaseResponse):
+    """
+    Model for PredictDataResponse
+    """
+
+    def __init__(self, response: BaseResponse = None, **kwargs):
+        super().__init__(
+            code=response.code if response else None,
+            data=response.data if response else None,
+            message=response.message if response else None,
+            **kwargs,
+        )
+
+class ModelOutputDirRequest(BaseRequest):
+    """
+    Model for ModelOutputDirRequest
+
+    输出各策略的存放位置
+    输入：
+    线上环境：
+    ${user.home}/strategyId/
+
+    本地测试：
+    ${user.home}/test/
+    """
+
+    def __init__(self, strategy_id: int = None):
+        """
+        Args:
+            strategy_id: 是否线上环境
+        """
+        self.strategy_id = strategy_id
+
+        super().__init__()
+        self.api = f'/ep/api/sdk/model_output_dir'
+        self.method = 'POST'
+        self.content_type = 'application/json'
+
+
+class ModelOutputDirResponse(BaseResponse):
+    """
+    Model for ModelOutputDirResponse
+    """
+
+    def __init__(self, response: BaseResponse = None, **kwargs):
+        super().__init__(
+            code=response.code if response else None,
+            data=response.data if response else None,
+            message=response.message if response else None,
+            **kwargs,
+        )
+
+class RunStrategyRequest(BaseRequest):
+    """
+    Model for RunStrategyRequest
+
+    获取此刻运行的策略模型基础信息
+    """
+
+    def __init__(self, strategy_id: int = None):
+        """
+        Args:
+            strategy_id: 策略 id
+        """
+        self.strategy_id = strategy_id
+
+        super().__init__()
+        self.api = f'/ep/api/sdk/get_run_strategy'
+        self.method = 'POST'
+        self.content_type = 'application/json'
+
+
+class RunStrategyResponse(BaseResponse):
+    """
+    Model for RunStrategyResponse
+    """
+
+    def __init__(self, response: BaseResponse = None, **kwargs):
+        super().__init__(
+            code=response.code if response else None,
+            data=response.data if response else None,
+            message=response.message if response else None,
+            **kwargs,
+        )

{ep_sdk_4pd-0.1.0 → ep_sdk_4pd-0.1.2}/ep_sdk_4pd.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ep_sdk_4pd
-Version: 0.1.0
+Version: 0.1.2
 Summary: 4paradigm Electricity Platform Service SDK Library for Python
 Home-page: https://gitlab.4pd.io/electricityproject/electricity-platform-sdk
 Author: 4paradigm Electricity Platform SDK

{ep_sdk_4pd-0.1.0 → ep_sdk_4pd-0.1.2}/ep_sdk_4pd.egg-info/SOURCES.txt

@@ -1,11 +1,16 @@
 README.md
 setup.py
 ep_sdk_4pd/__init__.py
-ep_sdk_4pd/client.py
+ep_sdk_4pd/ep_data.py
+ep_sdk_4pd/ep_system.py
 ep_sdk_4pd/models.py
 ep_sdk_4pd.egg-info/PKG-INFO
 ep_sdk_4pd.egg-info/SOURCES.txt
 ep_sdk_4pd.egg-info/dependency_links.txt
 ep_sdk_4pd.egg-info/requires.txt
 ep_sdk_4pd.egg-info/top_level.txt
-tests/test_history_data.py
+tests/test_get_run_strategy.py
+tests/test_get_system_date.py
+tests/test_history_data.py
+tests/test_model_output_dir.py
+tests/test_predict_data.py

{ep_sdk_4pd-0.1.0 → ep_sdk_4pd-0.1.2}/setup.py

@@ -8,7 +8,7 @@ DESCRIPTION = '4paradigm Electricity Platform Service SDK Library for Python'
 AUTHOR = '4paradigm Electricity Platform SDK'
 AUTHOR_EMAIL = ''
 URL = 'https://gitlab.4pd.io/electricityproject/electricity-platform-sdk'
-VERSION = '0.1.0'
+VERSION = '0.1.2'
 REQUIRES = ['requests']
 
 LONG_DESCRIPTION = ''

ep_sdk_4pd-0.1.2/tests/test_get_run_strategy.py

@@ -0,0 +1,14 @@
+from ep_sdk_4pd.ep_data import EpData
+from ep_sdk_4pd.ep_system import EpSystem
+
+
+def test_get_run_strategy():
+    print('-------------test_get_run_strategy-------------')
+
+    data = EpSystem.get_run_strategy(is_online=False)
+    print(data)
+    print('-------------------------------------')
+
+
+if __name__ == '__main__':
+    test_get_run_strategy()

ep_sdk_4pd-0.1.2/tests/test_get_system_date.py

@@ -0,0 +1,13 @@
+from ep_sdk_4pd.ep_system import EpSystem
+
+
+def test_get_system_date():
+    print('-------------test_get_system_date-------------')
+
+    target_date = EpSystem.get_system_date(is_online=True)
+    print(target_date)
+    print('-------------------------------------')
+
+
+if __name__ == '__main__':
+    test_get_system_date()

ep_sdk_4pd-0.1.2/tests/test_history_data.py

@@ -0,0 +1,13 @@
+from ep_sdk_4pd.ep_data import EpData
+
+
+def test_history_data():
+    print('-------------test_history_data-------------')
+
+    data = EpData.get_history_data(scope="weather", system_date="2025-04-17", days=2)
+    print(data)
+    print('-------------------------------------')
+
+
+if __name__ == '__main__':
+    test_history_data()

ep_sdk_4pd-0.1.2/tests/test_model_output_dir.py

@@ -0,0 +1,13 @@
+from ep_sdk_4pd.ep_system import EpSystem
+
+
+def test_model_output_dir():
+    print('-------------test_model_output_dir-------------')
+
+    data = EpSystem.model_output_dir(is_online=False)
+    print(data)
+    print('-------------------------------------')
+
+
+if __name__ == '__main__':
+    test_model_output_dir()

ep_sdk_4pd-0.1.2/tests/test_predict_data.py

@@ -0,0 +1,13 @@
+from ep_sdk_4pd.ep_data import EpData
+
+
+def test_predict_data():
+    print('-------------test_predict_data-------------')
+
+    data = EpData.get_predict_data(scope="weather", system_date="2025-04-15")
+    print(data)
+    print('-------------------------------------')
+
+
+if __name__ == '__main__':
+    test_predict_data()

ep_sdk_4pd-0.1.0/ep_sdk_4pd/__init__.py

@@ -1 +0,0 @@
-__version__ = '0.1.0'

ep_sdk_4pd-0.1.0/ep_sdk_4pd/client.py

@@ -1,51 +0,0 @@
-import json
-
-import requests
-
-from ep_sdk_4pd import models as ep_sdk_4pd_models
-
-
-class Client:
-    def __init__(
-        self,
-        config: ep_sdk_4pd_models.Config,
-    ):
-        self._endpoint = config.endpoint
-
-
-    # def __get_endpoint(self) -> None:
-    #     # test 地址
-    #     self._endpoint = 'http://172.27.88.56:6001'
-    #     # prod 地址
-    #     # self._endpoint = 'http://82.157.231.254:6001'
-
-    def history_data(
-        self,
-        request: ep_sdk_4pd_models.HistoryDataRequest = None,
-    ) -> ep_sdk_4pd_models.HistoryDataResponse:
-
-
-        full_url = f'{self._endpoint}{request.api}'
-        headers = {
-            'content-type': request.content_type,
-        }
-
-        payload = {
-            'scope': request.scope,
-            'system_date': request.system_date,
-            'days': request.days
-        }
-
-        response = requests.request(
-            method=request.method,
-            url=full_url,
-            headers=headers,
-            data=json.dumps(payload),
-        )
-
-        base_resp = ep_sdk_4pd_models.BaseResponse(
-            code=response.json().get('code', None),
-            data=response.json().get('data', None),
-            message=response.json().get('message', None),
-        )
-        return ep_sdk_4pd_models.HistoryDataResponse(response=base_resp)

ep_sdk_4pd-0.1.0/ep_sdk_4pd/models.py

@@ -1,98 +0,0 @@
-class Config:
-    """
-    Model for initing client
-    """
-
-    def __init__(
-        self,
-    ):
-        # test 地址
-        self.endpoint = 'http://172.27.88.56:6001'
-        # prod 地址
-        # self.endpoint = 'http://82.157.231.254:6001'
-
-
-class BaseRequest:
-    """
-    Model for BaseRequest
-    """
-
-    def __init__(self):
-        self.api = None
-        self.method = None
-        self.content_type = None
-        self.payload = None
-
-
-class BaseResponse:
-    """
-    Model for BaseResponse
-    """
-
-    def __init__(
-        self, code: int = None, data: dict = None, message: str = None, **kwargs
-    ):
-        self.code = code
-        self.data = data
-        self.message = message
-
-
-class TestRequest(BaseRequest):
-
-    def __init__(self):
-        super().__init__()
-        self.api = '/ai/cpp/api/v1/audio-language-detection/test'
-        self.method = 'POST'
-        self.content_type = 'application/json'
-        self.payload = {}
-
-
-class TestResponse(BaseResponse):
-
-    def __init__(self, response: BaseResponse = None, **kwargs):
-        super().__init__(
-            code=response.code if response else None,
-            data=response.data if response else None,
-            message=response.message if response else None,
-            **kwargs,
-        )
-
-
-class HistoryDataRequest(BaseRequest):
-    """
-    Model for HistoryDataRequest
-
-    获取具体系统时间往前x天的数据（最晚时间为系统时间 D-2）
-    特别注意：get_history_data 或 get_predict_data 会get_system_date进行检测，不可获取system_date以后得数据，避免数据穿越现象
-    """
-
-    def __init__(self, scope: str = None, system_date: str = None, days: int = None):
-        """
-        Args:
-            scope: "weather","plant","market"
-            system_date: 系统时间
-            days: 表示systemDate之前days的数据
-        """
-
-        self.scope = scope
-        self.system_date = system_date
-        self.days = days
-
-        super().__init__()
-        self.api = f'/ep/api/sdk/get_history_data'
-        self.method = 'POST'
-        self.content_type = 'application/json'
-
-
-class HistoryDataResponse(BaseResponse):
-    """
-    Model for HistoryDataResponse
-    """
-
-    def __init__(self, response: BaseResponse = None, **kwargs):
-        super().__init__(
-            code=response.code if response else None,
-            data=response.data if response else None,
-            message=response.message if response else None,
-            **kwargs,
-        )

ep_sdk_4pd-0.1.0/tests/test_history_data.py

@@ -1,23 +0,0 @@
-import ep_sdk_4pd.client as ep_sdk_4pd_client
-import ep_sdk_4pd.models as ep_sdk_4pd_models
-
-
-def test_history_data():
-    print('-------------test client-------------')
-
-    config = ep_sdk_4pd_models.Config()
-    client = ep_sdk_4pd_client.Client(config=config)
-    request = ep_sdk_4pd_models.HistoryDataRequest(
-        scope="weather",
-        system_date="2025-04-17",
-        days=2,
-    )
-    response = client.history_data(request=request)
-    print(response.code)
-    print(response.data)
-    print(response.message)
-    print('-------------------------------------')
-
-
-if __name__ == '__main__':
-    test_history_data()