概要

本文將介紹如何使用Python語言，結合RPM包格式，構建一款名為“Nest Info”的數據庫查詢工具。該工具旨在模擬查詢MySQL數據庫，獲取虛擬機和VPC信息，同時具備強大的配置化能力，允許用戶動態設定數據庫連接參數。

使用效果

動態配置數據庫連接信息

為了增強工具的通用性，我們通過命令行選項nest_info init，允許用戶動態配置數據庫服務器地址、端口號、用戶名、密碼和數據庫名。這樣，即使數據庫環境發生變化，也不需要修改源代碼，大大提升了工具的適應能力。

命令行接口設計

為了提供直觀且強大的用戶體驗，我們設計了命令行接口，支持以下子命令：

• nest_info vm list：查詢并列出所有虛擬機信息。
• nest_info vpc list：查詢并列出所有VPC信息。
• nest_info vm/vpc list -n <namespace>：查詢指定命名空間內的虛擬機信息。

實現方法

使用配置文件管理SQL查詢語句和字段名

通過創建config.json配置文件，我們能夠以JSON格式存儲SQL查詢語句和字段名稱。這種方式不僅便于管理，還極大地提高了工具的可維護性和可擴展性。

json

{
  "queries": {
    "vms": "SELECT id, vm_id, name, namespace FROM vms WHERE namespace = '%s'",    "vpcs": "SELECT id, name, cidr FROM vpcs WHERE namespace = '%s'"
  },
  "fields": {
    "vms": ["id", "name", "namespace"],
    "vpcs": ["id", "vm_ids", "name", "cidr", "namespace"]
  }
}

編寫主體實現代碼

python

import json
import sys
import argparse
import mysql.connector

def load_config():
    # 加載配置文件
    with open('config.json', 'r') as file:
        config = json.load(file)
    return config

def get_db_connection(host='localhost', port=3306, user='root', password='password', database='mydb'):
    # 連接數據庫
    conn = mysql.connector.connect(
        host=host,
        port=port,
        user=user,
        password=password,
        database=database
    )
    return conn

def query_data(query, *args):
    # 執行查詢并返回結果
    conn = get_db_connection()
    cursor = conn.cursor()
    query = query % args if args else query
    cursor.execute(query)
    columns = [desc[0] for desc in cursor.description]
    results = [dict(zip(columns, row)) for row in cursor.fetchall()]
    cursor.close()
    conn.close()
    return results

def main():
    config = load_config()
    queries = config['queries']
    fields = config['fields']

    parser = argparse.ArgumentParser(description='Nest Info 工具')
    subparsers = parser.add_subparsers(dest='command')

    # vm list 命令
    vm_parser = subparsers.add_parser('vm', help='列出虛擬機')
    vm_parser.add_argument('-n', '--namespace', help='指定命名空間')

    # vpc list 命令
    vpc_parser = subparsers.add_parser('vpc', help='列出 VPC')
    vpc_parser .add_argument('-n', '--namespace', help='指定命名空間')

    # 初始化數據庫連接信息
    init_parser = subparsers.add_parser('init', help='初始化數據庫連接')
    init_parser.add_argument('--host', default='localhost', help='數據庫服務器地址')
    init_parser.add_argument('--port', type=int, default=3306, help='數據庫端口')
    init_parser.add_argument('--user', default='root', help='數據庫用戶名')
    init_parser.add_argument('--password', help='數據庫密碼')
    init_parser.add_argument('--database', default='mydb', help='數據庫名稱')

    args = parser.parse_args()

    if args.command == 'init':
        # 更新數據庫連接信息
        global get_db_connection
        get_db_connection = lambda: get_db_connection(
            host=args.host,
            port=args.port,
            user=args.user,
            password=args.password,
            database=args.database
        )
        print("數據庫連接信息已更新。")
    elif args.command == 'vm':
        if args.namespace:
            data = query_data(queries['vms_namespace'] % args.namespace)
        else:
            data = query_data(queries['vms'])
        print(json.dumps(data, ensure_ascii=False, indent=2))
    elif args.command == 'vpc':
        data = query_data(queries['vpcs'])
        print(json.dumps(data, ensure_ascii=False, indent=2))

if __name__ == "__main__":
    main()

構建RPM包

確保 config.json 文件被正確包含在 RPM 包中。更新 nest_info.spec 文件中的 %files 部分：

spec

%files
%defattr(644,root,root,755)
%doc README
%config(noreplace) /etc/nest_info/config.json
%exec /usr/bin/nest_info

在包含 nest_info.spec 文件的目錄中，運行以下命令來構建 RPM 包：

bash

rpmbuild -ba nest_info.spec

這將在 ~/rpmbuild/RPMS 目錄下生成你的 RPM 包。
最后，我們使用rpmbuild工具，根據nest_info.spec規格文件，將工具封裝成RPM包，方便在基于RPM的Linux發行版上安裝和使用。

結語

本文詳細介紹了如何使用Python和RPM包構建數據庫查詢工具的全過程，從設計思路到具體實現，再到最終的RPM包構建，力求全面而深入。如果你對這一領域的探索感興趣，不妨嘗試按照本文的步驟動手實踐，相信會從中獲得不少收獲和樂趣。