Python腳本開發實戰

隨著技術的迅猛發展，Python 作為一種高級語言，被越來越多的開發者所熟知和使用。Python 不僅可以用來進行 web 開發，在數據分析、人工智慧、自動化測試和腳本開發等領域都有著廣泛的應用。

腳本語言顧名思義是用來編寫腳本的語言。在計算機領域中，腳本指的是一系列指令的集合，主要用於無需用戶干預或只需要很少干預的情況下，完成某一任務的自動化流程。

Python 作為一種腳本語言，具有可讀性強、語法簡潔易懂、功能強大、速度快等優點，在腳本開發領域有著得天獨厚的優勢。Python 腳本開發，可以幫助開發者快速實現自動化流程，並且能夠處理海量數據，提升效率。

一、快速生成目錄列表

在實際工作中，我們經常需要將某個目錄下的文件以某種規則進行分類，比如按照文件後綴名分組，或者按照文件大小進行排序。這時，我們可以使用 Python 腳本快速生成對應的目錄列表。

import os

def get_file_list(path):
    file_list = os.listdir(path)
    return file_list

if __name__ == "__main__":
    path = './test/'
    file_list = get_file_list(path)
    print(file_list)

代碼功能解釋：

• 使用 Python 內置的 os 模塊，獲取某個路徑下的所有文件名。
• 定義函數 get_file_list，傳入一個路徑參數 path。
• 函數內部調用 os.listdir()，返回一個包含文件名的列表。
• 在主程序中調用 get_file_list() 函數，返回包含目錄下文件名的列表，並將其列印輸出。

二、自動生成 Excel 報表

在進行數據分析的過程中，我們經常需要將數據結果整理成報表，並上傳到公司內部的管理平台。這時，我們可以使用 Python 腳本，自動生成 Excel 報表，並自動發送到指定郵箱。

import pandas as pd
import smtplib
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
from email.mime.application import MIMEApplication

# 數據初始化
data = {'Name':['Tom', 'John', 'Mary'], 'Math':[80, 90, 85], 'English':[85, 95, 90]}
df = pd.DataFrame(data)

# 生成 Excel 文件
writer = pd.ExcelWriter('report.xlsx')
df.to_excel(writer, index=False)
writer.save()

# 發送郵件
msg = MIMEMultipart()
msg['Subject'] = 'Monthly Report'
msg['From'] = 'sender@example.com'
msg['To'] = 'receiver@example.com'

body = MIMEText('Please find attached the monthly report.')
msg.attach(body)

with open('report.xlsx', 'rb') as f:
    attachment = MIMEApplication(f.read(), _subtype='xlsx')
    attachment.add_header('content-disposition', 'attachment', filename='report.xlsx')
    msg.attach(attachment)

s = smtplib.SMTP('smtp.example.com', 587)
s.starttls()
s.login('sender@example.com', 'password')
s.sendmail('sender@example.com', 'receiver@example.com', msg.as_string())
s.quit()

代碼功能解釋：

• 使用 pandas 庫創建一個 DataFrame 對象，用於存儲表格數據。
• 使用 DataFrame.to_excel() 方法，將數據表格生成為一個 Excel 文件。
• 使用 MIME 多媒體郵件模塊，創建一個郵件實例，並設置郵件主題、發件人、收件人等信息。
• 將生成的 Excel 文件作為附件，添加到郵件實例中。
• 通過 SMTP 協議，連接郵件伺服器，登錄賬號，發送郵件。

三、自動化測試腳本開發

在進行軟體開發的過程中，我們需要對軟體進行測試，以確保軟體的穩定性和可靠性。這時，我們可以使用 Python 編寫自動化測試腳本，幫助我們快速檢測程序的功能模塊是否正常。

from selenium import webdriver
import time

# 初始化瀏覽器
driver = webdriver.Chrome()

# 打開網頁
driver.get('http://www.example.com')

# 登錄
driver.find_element_by_name('username').send_keys('admin')
driver.find_element_by_name('password').send_keys('password')
driver.find_element_by_tag_name('button').click()

# 進行操作
driver.find_element_by_class_name('title').click()
time.sleep(1)
driver.find_element_by_id('submit-button').click()
time.sleep(1)

# 斷言
result = driver.find_element_by_id('result').text
assert result == 'success'

# 結束測試
driver.quit()

代碼功能解釋：

• 使用 Python 的 selenium 庫，初始化一個瀏覽器實例。
• 使用 webdriver.get() 方法，打開指定網頁。
• 使用 find_element_by_xxx() 方法，找到登錄表單的各個元素，並進行操作。
• 使用 assert 語句，判斷操作結果是否與預期相同。
• 使用 webdriver.quit() 方法，關閉瀏覽器實例。