pixelarraylib 1.0.0__py3-none-any.whl

arraylib/aliyun/eci.py

@@ -0,0 +1,47 @@
+import json
+import traceback
+from typing import Optional, Dict, Any, List
+from alibabacloud_eci20180808.client import Client as EciClient
+from alibabacloud_eci20180808.models import (
+    CreateContainerGroupRequest,
+    DescribeContainerGroupsRequest,
+    DeleteContainerGroupRequest,
+    UpdateContainerGroupRequest,
+    RestartContainerGroupRequest,
+)
+from alibabacloud_tea_openapi import models as open_api_models
+from alibabacloud_tea_util import models as util_models
+from arraylib.monitor.feishu import Feishu
+
+feishu_alert = Feishu("devtoolkit服务报警")
+
+
+class ECIUtils:
+    def __init__(self, region_id: str, access_key_id: str, access_key_secret: str):
+        """
+        description:
+            初始化ECI工具类
+        parameters:
+            region_id(str): 地域ID
+            access_key_id(str): 访问密钥ID
+            access_key_secret(str): 访问密钥Secret
+        """
+        self.region_id = region_id
+        self.access_key_id = access_key_id
+        self.access_key_secret = access_key_secret
+        self.client = self._create_client()
+
+    def _create_client(self) -> EciClient:
+        """
+        description:
+            创建ECI客户端
+        return:
+            EciClient: ECI客户端实例
+        """
+        config = open_api_models.Config(
+            access_key_id=self.access_key_id,
+            access_key_secret=self.access_key_secret,
+            region_id=self.region_id,
+            endpoint=f"eci.{self.region_id}.aliyuncs.com",
+        )
+        return EciClient(config)

arraylib/aliyun/ecs.py

@@ -0,0 +1,68 @@
+from datetime import datetime, timedelta
+import json
+from alibabacloud_cms20190101.client import Client as Cms20190101Client
+from alibabacloud_tea_openapi import models as open_api_models
+from alibabacloud_cms20190101 import models as cms_20190101_models
+from alibabacloud_tea_util import models as util_models
+
+
+class ECSUtils:
+    def __init__(self, access_key_id: str, access_key_secret: str, region_id: str):
+        self.cms_client = Cms20190101Client(
+            open_api_models.Config(
+                type="access_key",
+                access_key_id=access_key_id,
+                access_key_secret=access_key_secret,
+                endpoint=f"metrics.{region_id}.aliyuncs.com",
+            )
+        )
+
+    def get_memory_utilization(self, instance_id: str, delta_minutes: int = 1) -> float:
+        """
+        description:
+            获取实例的内存使用率
+        parameters:
+            instance_id(str): 实例ID
+            delta_minutes(int): 时间差 默认1分钟
+        return:
+            memory_utilization(float): 内存使用率 取值范围0-1
+        """
+        describe_metric_list_request = cms_20190101_models.DescribeMetricListRequest(
+            namespace="acs_ecs_dashboard",
+            metric_name="memory_usedutilization",
+            dimensions=f'{{"instanceId": "{instance_id}"}}',
+            start_time=datetime.now() - timedelta(minutes=delta_minutes),
+            end_time=datetime.now(),
+        )
+        runtime = util_models.RuntimeOptions()
+        response = self.cms_client.describe_metric_list_with_options(
+            describe_metric_list_request, runtime
+        )
+        monitor_list = json.loads(response.body.datapoints)
+        max_value = max(monitor_list, key=lambda x: x["timestamp"])
+        return max_value["Average"] * 0.01
+
+    def get_cpu_utilization(self, instance_id: str, delta_minutes: int = 1) -> float:
+        """
+        description:
+            获取实例的CPU使用率
+        parameters:
+            instance_id(str): 实例ID
+            delta_minutes(int): 时间差 默认1分钟
+        return:
+            cpu_utilization(float): 内存使用率 取值范围0-1
+        """
+        describe_metric_list_request = cms_20190101_models.DescribeMetricListRequest(
+            namespace="acs_ecs_dashboard",
+            metric_name="CPUUtilization",
+            dimensions=f'{{"instanceId": "{instance_id}"}}',
+            start_time=datetime.now() - timedelta(minutes=delta_minutes),
+            end_time=datetime.now(),
+        )
+        runtime = util_models.RuntimeOptions()
+        response = self.cms_client.describe_metric_list_with_options(
+            describe_metric_list_request, runtime
+        )
+        monitor_list = json.loads(response.body.datapoints)
+        max_value = max(monitor_list, key=lambda x: x["timestamp"])
+        return max_value["Average"] * 0.01

arraylib/aliyun/fc.py

@@ -0,0 +1,142 @@
+import json
+import os
+from tempfile import TemporaryDirectory
+import traceback
+
+from alibabacloud_fc20230330.client import Client as FC20230330Client
+from alibabacloud_tea_openapi import models as open_api_models
+from alibabacloud_darabonba_stream.client import Client as StreamClient
+from alibabacloud_fc20230330 import models as fc20230330_models
+from alibabacloud_tea_util import models as util_models
+from arraylib.monitor.feishu import Feishu
+from arraylib.system.common import execute_command, file_to_base64
+
+feishu_alert = Feishu("devtoolkit服务报警")
+
+
+class FCUtils:
+    def __init__(self, access_key_id, access_key_secret, account_id, region_id):
+        self.client = FC20230330Client(
+            open_api_models.Config(
+                access_key_id=access_key_id,
+                access_key_secret=access_key_secret,
+                endpoint=f"{account_id}.{region_id}.fc.aliyuncs.com",
+            )
+        )
+
+    def invoke_function(
+        self,
+        function_name: str,
+        params: dict,
+        retry: int = 5,
+        timeout: int = 6000 * 1000,
+    ) -> tuple[str, bool]:
+        last_exception = None
+        for i in range(retry):
+            try:
+                body_stream = StreamClient.read_from_string(json.dumps(params))
+                invoke_function_headers = fc20230330_models.InvokeFunctionHeaders(
+                    x_fc_invocation_type="Sync", x_fc_log_type="None"
+                )
+                invoke_function_request = fc20230330_models.InvokeFunctionRequest(
+                    qualifier="LATEST", body=body_stream
+                )
+
+                runtime = util_models.RuntimeOptions()
+                runtime.connect_timeout = 60
+                runtime.read_timeout = timeout
+                response = self.client.invoke_function_with_options(
+                    function_name,
+                    invoke_function_request,
+                    invoke_function_headers,
+                    runtime,
+                )
+                if not response or response.status_code != 200:
+                    last_exception = f"aliyun fc invoke_function error: {i} {response}"
+                    # 如果不是最后一次重试，则继续重试
+                    if i < retry - 1:
+                        continue
+                    else:
+                        return None, False
+                return response.body.read().decode(), True
+            except Exception as e:
+                last_exception = (
+                    f"aliyun fc invoke_function error: {i} {traceback.format_exc()}"
+                )
+                # 如果不是最后一次重试，则继续重试
+                if i < retry - 1:
+                    continue
+                else:
+                    return None, False
+        # 如果所有重试都失败，返回最后一次的异常信息
+        feishu_alert.send(
+            f"aliyun fc invoke_function failed after {retry} retries. Last error: {last_exception}"
+        )
+        return None, False
+
+    def update_function(
+        self,
+        function_name: str,
+        dir_path: str,
+        retry: int = 5,
+    ) -> tuple[str, bool]:
+        last_exception = None
+        for i in range(retry):
+            try:
+                update_function_input_vpcconfig = fc20230330_models.VPCConfig(
+                    security_group_id="sg-2ze2gmwyq9wxt34xbbli",
+                    v_switch_ids=["vsw-2ze0wbhhcu9c0veyf2y1m"],
+                    vpc_id="vpc-2zewkxvljq1091qa8x1um",
+                )
+                with TemporaryDirectory() as temp_dir:
+                    code_zip_path = os.path.join(temp_dir, "code.zip")
+                    _, success = execute_command(["zip", "-r", code_zip_path, dir_path])
+                    if not success:
+                        continue
+                    update_function_input_input_code_location = (
+                        fc20230330_models.InputCodeLocation(
+                            zip_file=file_to_base64(code_zip_path)
+                        )
+                    )
+                update_function_input = fc20230330_models.UpdateFunctionInput(
+                    handler="index.handler",
+                    code=update_function_input_input_code_location,
+                    timeout=6000,
+                    disk_size=10240,
+                    internet_access=True,
+                    cpu=16,
+                    runtime="python3.12",
+                    instance_concurrency=1,
+                    memory_size=16384,
+                    vpc_config=update_function_input_vpcconfig,
+                )
+                update_function_request = fc20230330_models.UpdateFunctionRequest(
+                    body=update_function_input
+                )
+                runtime = util_models.RuntimeOptions()
+                headers = {}
+                response = self.client.update_function_with_options(
+                    function_name, update_function_request, headers, runtime
+                )
+                if not response or response.status_code != 200:
+                    last_exception = f"aliyun fc update_function error: {i} {response}"
+                    # 如果不是最后一次重试，则继续重试
+                    if i < retry - 1:
+                        continue
+                    else:
+                        return False
+                return True
+            except Exception as e:
+                last_exception = (
+                    f"aliyun fc update_function error: {i} {traceback.format_exc()}"
+                )
+                # 如果不是最后一次重试，则继续重试
+                if i < retry - 1:
+                    continue
+                else:
+                    return False
+        # 如果所有重试都失败，返回最后一次的异常信息
+        feishu_alert.send(
+            f"aliyun fc update_function failed after {retry} retries. Last error: {last_exception}"
+        )
+        return False