python适应不同场景的函数 - *args, **kwargs

在Python中，*args和**kwargs是用来处理函数中不定数量的参数的特殊语法。

*args和**kwargs介绍

*args

*args用于传递不定数量的位置参数，它会将传递给函数的位置参数收集到一个元组中。例如：

def my_func(*args):
    for arg in args:
        print(arg)

my_func(1, 2, 3)

在这个例子中，my_func函数接受不定数量的位置参数，并将它们打印出来。

**kwargs

**kwargs用于传递不定数量的关键字参数，它会将传递给函数的关键字参数收集到一个字典中。例如：

def my_func(**kwargs):
    for key, value in kwargs.items():
        print(f"{key}: {value}")

my_func(a=1, b=2, c=3)

在这个例子中，my_func函数接受不定数量的关键字参数，并将它们打印出来。

可以同时在函数定义中使用args和**kwargs，并且它们的顺序是固定的，即args必须在**kwargs之前。例如：

def my_func(*args, **kwargs):
    for arg in args:
        print(arg)
    for key, value in kwargs.items():
        print(f"{key}: {value}")

my_func(1, 2, 3, a=4, b=5)

在这个例子中，my_func函数接受不定数量的位置参数和关键字参数，并将它们打印出来。

应用

如果希望函数里参数有默认值

如果我希望my_func()函数里，a、b有默认值，比如a="AA", b=10

可以在函数定义中给**kwargs设置默认值，然后在函数内部检查是否传入了对应的关键字参数。如果没有传入，则使用默认值。例如：

def my_func(*args, **kwargs):
    a = kwargs.get('a', 'AA')
    b = kwargs.get('b', 10)

    for arg in args:
        print(arg)
    
    print(f"a: {a}")
    print(f"b: {b}")

my_func(1, 2, 3, b=20)

在这个例子中，如果没有传入a和b的值，a会使用默认值'AA'，b会使用默认值10。你可以调用my_func函数并传入其他位置参数和关键字参数，如果没有传入a和b，则会使用默认值。

实际应用

当使用args和**kwargs时，可以处理不定数量的参数，这在实际应用中非常有用。以下是一些使用args和**kwargs的实际应用示例：

日志记录器函数

可以编写一个通用的日志记录器函数，接受不定数量的参数，并将它们记录到日志文件中。

def log_message(*args, **kwargs):
    level = kwargs.get('level', 'INFO')
    message = ' '.join(args)
    print(f"[{level}] {message}")

log_message("User logged in", level='DEBUG')
log_message("Error occurred", level='ERROR')

配置文件解析器

可以编写一个函数，接受不定数量的键值对作为参数，然后将这些参数解析为配置字典。

def parse_config(**kwargs):
    config = {}
    for key, value in kwargs.items():
        config[key] = value
    return config

config = parse_config(username='admin', password='123456', host='localhost')
print(config)

HTML标签生成器

可以编写一个函数，接受标签名和不定数量的属性作为参数，然后生成相应的HTML标签。

def create_html_tag(tag_name, **kwargs):
    attributes = ' '.join([f'{key}="{value}"' for key, value in kwargs.items()])
    return f'<{tag_name} {attributes}></{tag_name}>'

input_tag = create_html_tag('input', type='text', name='username', placeholder='Enter your username')
print(input_tag)

这些示例展示了如何使用args和**kwargs处理不定数量的参数，使函数更加灵活和通用。您可以根据具体的需求和场景，灵活运用args和**kwargs来简化代码并提高可复用性。

操作mysql的类

当编写一个链接MySQL数据库的类时，你可以使用*args和**kwargs来处理不定数量的连接参数。以下是一个示例代码，展示了如何编写一个简单的MySQL连接类：

import mysql.connector

class MySQLConnection:
    def __init__(self, **kwargs):
        self.connection = mysql.connector.connect(**kwargs)
        self.cursor = self.connection.cursor()

    def execute_query(self, query):
        self.cursor.execute(query)
        result = self.cursor.fetchall()
        return result

    def close_connection(self):
        self.cursor.close()
        self.connection.close()

# 使用示例
connection_params = {
    'host': 'localhost',
    'user': 'root',
    'password': 'password',
    'database': 'my_database'
}

mysql_conn = MySQLConnection(**connection_params)
result = mysql_conn.execute_query("SELECT * FROM my_table")
print(result)

mysql_conn.close_connection()

在这个示例中，MySQLConnection类接受不定数量的连接参数作为**kwargs，然后使用这些参数来建立MySQL数据库连接。你可以调用execute_query方法来执行查询，并调用close_connection方法来关闭连接。

这个示例展示了如何结合*args和**kwargs来编写一个灵活的MySQL连接类，以处理不定数量的连接参数。你可以根据实际的需求进一步扩展和定制这个类。