常用 MaxKB 函数库(HTTP、MYSQL)

一、HTTP请求函数

1、代码示例

在 Python 中，执行简单的 GET 请求通常可以通过多种库来实现，但最常用和简便的是 requests 库。以下是一个使用 requests 库执行 GET 请求的基本示例：

import requests  

   

def get_weather_by_city_id(city_id):

    """  

    通过城市ID获取天气信息

    :param city_id: 城市ID，例如'101190101'  

    :return: 响应内容  

    """  

    # 构造请求URL  

    url = f"http://t.weather.sojson.com/api/weather/city/{city_id}"

       

    try:

        # 发送GET请求  

        response = requests.get(url)

           

        # 检查请求是否成功

        if response.status_code == 200:

            # 解析并打印JSON响应  

            weather_data = response.json()

            print(weather_data)

            return weather_data

        else:

            print(f"请求失败，状态码: {response.status_code}")

            return None

    except requests.RequestException as e:

        print(f"请求错误: {e}")

2、详细配置

添加输入变量，变量名和 Python 代码中的函数名称要一致，同时支持通过{变量名}进行传参。

输入不同传参值测试。

二、MySQL数据库连接函数

1、安装依赖

在 Python 中调用 MySQL 数据库的函数通常使用 Python 的数据库连接库如 mysql-connector-python、PyMySQL 等来实现。以下是一个使用 mysql-connector-python 库来调用MySQL数据库中函数的简单示例。

首先，确保你已经安装了 mysql-connector-python。如果没有安装，可以通过 pip 安装：

pip install mysql-connector-python

正常来说会很慢，可以使用阿里云的源：

pip install mysql-connector-python -i https://mirrors.aliyun.com/pypi/simple/

直接进入到 MaxKB 容器里面执行即可：

2、代码示例

使用以下 Python 脚本来连接 MySQL 数据库并执行 SQL 语句：

import mysql.connector

from mysql.connector import Error

def execute_sql_query(query):

    """

    :param query: 要执行的SQL查询

    :return: 查询结果

    """

    connection = None

    result = None

    try:

        connection = mysql.connector.connect(

            host= "10.1.14.175",

            user= "root",

            passwd= "YTkzZDNiNDItZDcyNC******",

            database= "jumpserver",

            port="3306"

        )

        if connection.is_connected():

            cursor = connection.cursor()

            cursor.execute(query)

            # 对于SELECT查询，我们使用fetchall()来获取所有结果

            # 对于INSERT、UPDATE、DELETE等，你可以通过cursor.rowcount来获取影响的行数

            if query.upper().startswith('SELECT'):

                result = cursor.fetchall()

            else:

                connection.commit()  # 确保更改被提交到数据库

                result = cursor.rowcount  # 获取影响的行数

    except Error as e:

        print(f"Error while connecting to MySQL {e}")

    finally:

        if connection.is_connected():

            cursor.close()

            connection.close()

            print("MySQL connection is closed")

    return result

3、详细配置

添加输入变量，变量名和Python代码中的函数名称要一致。

执行SQL语句，查看输出结果。