Python定時查詢starrocks數(shù)據(jù)庫并將結果保存在excel

更新時間：2025年03月07日 15:23:07 作者：運維仙人

這篇文章主要為大家詳細介紹了Python如何實現(xiàn)定時查詢starrocks數(shù)據(jù)庫并將結果保存在excel,文中的示例代碼講解詳細,感興趣的小伙伴可以參考一下

背景

在工作過程中要定期的更新excel表的信息，每個星期都要去查詢strarocks的數(shù)據(jù)導出結果到excel，俗話說：“不會偷懶的運維不是好運維”，于是寫了python小程序解決這個重復的工作，設置定時任務，直接去服務器下載導出的excel表格即可。

代碼

代碼的邏輯簡單介紹：將要執(zhí)行的SQL以名稱進行區(qū)分保存并放到目錄：SQLfileDir，設置結果存放路徑：./…/outputdir/。python先查詢數(shù)據(jù)，然后以SQL文件名前綴為excle名稱保存，最后移動到指定目錄。

本地調試界面：

# -*- coding: utf-8 -*-
# @Author : zjh
# @Time : 2023-12-27
# @Description: 定時跑數(shù)據(jù)保存到excel

import os
import shutil
import pandas as pd
import pymysql
import openpyxl
import datetime

class StarRocksExporter(object):
    def __init__(self, host, port, database, user, password, query,
                 srcdir, destdir,filename):
        self.host = host
        self.port = port
        self.database = database
        self.user = user
        self.password = password
        self.query = query
        self.srcdir = srcdir
        self.destdir = destdir
        self.filename = filename
        self.writer = pd.ExcelWriter(filename+str('.xlsx'))

    def export_to_excel(self):
        df = pd.read_sql(self.query, self.engine)
        # print(df)
        df.to_excel(self.writer, sheet_name=self.filename, index=False)
        self.writer.save()

    def move_to_dest(self):
        if not os.path.isdir(self.destdir):
            self.destdir = os.mkdir(self.destdir)
        file_list = os.listdir(self.srcdir)
        for file in file_list:
            #print(file)
            #print(file.split('.')[0])
            try:
                #print(file.split('.')[1])
                fiel_str = file.split('.')[1]
                if fiel_str == 'xlsx':
                    shutil.move(str(self.srcdir) + file, str(self.destdir) + file)
            except Exception:
                print("沒有后綴的文件:",file)
            #shutil.move(str(self.srcdir) + file, str(self.destdir) + file)


    def execute(self):
        with pymysql.connect(host=self.host,port=self.port,database=self.database,user=self.user,password=self.password) as engine:
            self.engine = engine
            self.query_star_rock(self.query)
            self.export_to_excel()
            self.move_to_dest()

    def query_star_rock(self, query):
        cursor = self.engine.cursor()
        cursor.execute(query)
        results = cursor.fetchall()
        return  results

    def get_user(self, user):
        pass

    def get_password(self, password):
        pass

if __name__ == '__main__':
    Destdir = './../outputdir/'
    Srcdir = './'
    folder_path = './../SQLfileDir/'
    file_list = os.listdir(folder_path)

    for sqlfile in file_list:
        file_path=str(folder_path)+str(sqlfile)
        with open(file_path, "r", encoding='utf-8') as f:
           sql = f.read()
        srfilename = sqlfile.split('.')[0]
        exporter = StarRocksExporter('192.168.10.11', 19030, 'manager', 'sys_ro','sdagfsdg!@#saf134',
                                sql,Srcdir,Destdir,srfilename)
        exporter.execute()

MySQL的導出也可以用該腳本，因為starrocks/Doris連接都是用的MySQL驅動，兼容MySQL語法。

方法擴展

下面小編為大家介紹一下Python如何基于starrocks庫連接查詢StarRocks數(shù)據(jù)庫，需要的可以參考一下

SQLAlchemy 用法

要使用 SQLAlchemy 連接到 StarRocks，連接字符串如下所示：

starrocks://<User>:<Password>@<Host>:<Port>/<Catalog>.<Database>

import pandas as pd
from sqlalchemy import create_engine, text

# 設置 pandas 顯示選項以顯示所有列
pd.set_option('display.max_columns', None)
pd.set_option('display.max_rows', None)
pd.set_option('display.max_colwidth', None)

"""
'starrocks://<User>:<Password>@<Host>:<Port>/<Catalog>.<Database>'
"""
def query_user_data(user_name):
    # 連接到StarRocks數(shù)據(jù)庫
    engine = create_engine('starrocks://test_user:test_user123@192.168.1.2:9030/sr_db')

    # 執(zhí)行查詢并獲取結果
    with engine.connect() as connection:
        sql_query = "select data from sr_db.user where user_name=" + user_name
        result = connection.execute(text(sql_query)).fetchall()

    # 將查詢結果轉換為 Pandas DataFrame
    ret_df = pd.DataFrame(result)
    return ret_df

# main function
if __name__ == '__main__':
    console = Console()
    user_name = "'tom'"
    df = query_user_data(user_name )
    # 如果 DataFrame 不為空，顯示
    if df is not None and not df.empty:
        print(df)
    else:
        print("數(shù)據(jù)為空")

到此這篇關于Python定時查詢starrocks數(shù)據(jù)庫并將結果保存在excel的文章就介紹到這了,更多相關Python定時查詢starrocks內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: