python - 有没有办法在 Python 中设置特定于脚本的默认参数，这样您就不必每次都指定它？

Question

我正在使用模块 sqlite3 使用 Pandas 的 .read_sql_query() 方法进行 SQL 查询。作为它的第二个参数，该方法需要与数据库的连接。这很容易存储在变量中:con = sqlite3.connect('PATH/TO/DATABASE')但是，在调用该方法时，必须继续输入“con”作为第二个参数(例如，df = pd.read_sql_query('SQL QUERY HERE', con))是非常乏味的。我可以在我的脚本顶部指定我自动希望第二个参数每次都是“con”吗？所以我可以只指定查询而不必每次都输入 con ？ Python有这个功能吗？ (为了记录，第二个参数被正式称为 con，根据 pandas 的文档。)
谢谢你的帮助!

Answer 1

您可以定义一个为您记住连接的类

class PdSqlReader:

    def __init__(self, path_to_db):
        self.con = sqlite3.connect(path_to_db)

    def query(self, sql_query):
        return pd.read_sql_query(sql_query, self.con)

reader = PdSqlReader('PATH/TO/DATABASE')
df = reader.query('SQL QUERY HERE')