Python tikinter實現打開指定ip的電腦攝像頭

以下是一個使用Python的tkinter和OpenCV庫實現打開指定IP攝像頭的應用程序。這個程序允許用戶輸入IP攝像頭的URL，并實時顯示攝像頭畫面，同時支持截圖和錄制功能。

import tkinter as tk
from tkinter import ttk, messagebox, filedialog
import cv2
from PIL import Image, ImageTk
import threading
import time
import os
from datetime import datetimeclass IPCameraViewer:def __init__(self, root):self.root = rootself.root.title("IP攝像頭查看器")self.root.geometry("900x600")self.root.resizable(True, True)# 設置字體確保中文顯示正常self.font = ('SimHei', 10)self.title_font = ('SimHei', 14, 'bold')# 攝像頭相關變量self.cap = Noneself.ip_url = tk.StringVar(value="rtsp://admin:password@192.168.1.100:554/stream1")self.is_playing = Falseself.is_recording = Falseself.frame = Noneself.recorder = None# 創建UIself.create_widgets()def create_widgets(self):"""創建用戶界面"""# 標題區域title_frame = ttk.Frame(self.root, padding="10")title_frame.pack(fill=tk.X)title_label = ttk.Label(title_frame, text="IP攝像頭查看器", font=self.title_font)title_label.pack(pady=10)# 連接設置區域connect_frame = ttk.LabelFrame(self.root, text="攝像頭設置", padding="10")connect_frame.pack(fill=tk.X, padx=10, pady=5)ttk.Label(connect_frame, text="攝像頭URL:", font=self.font).pack(side=tk.LEFT, padx=5)url_entry = ttk.Entry(connect_frame,textvariable=self.ip_url,font=self.font,width=50)url_entry.pack(side=tk.LEFT, padx=5)self.connect_button = ttk.Button(connect_frame,text="連接",command=self.toggle_connection,style="Accent.TButton")self.connect_button.pack(side=tk.LEFT, padx=5)# 狀態區域status_frame = ttk.Frame(self.root, padding="5")status_frame.pack(fill=tk.X)self.status_var = tk.StringVar()self.status_var.set("未連接")status_label = ttk.Label(status_frame,textvariable=self.status_var,font=self.font,foreground="red")status_label.pack(side=tk.LEFT, padx=5)# 視頻顯示區域self.display_frame = ttk.LabelFrame(self.root, text="視頻預覽", padding="10")self.display_frame.pack(fill=tk.BOTH, expand=True, padx=10, pady=5)# 創建畫布用于顯示視頻self.canvas_width = 800self.canvas_height = 450self.canvas = tk.Canvas(self.display_frame,width=self.canvas_width,height=self.canvas_height,bg="black")self.canvas.pack(fill=tk.BOTH, expand=True)# 控制按鈕區域control_frame = ttk.Frame(self.root, padding="10")control_frame.pack(fill=tk.X)self.screenshot_button = ttk.Button(control_frame,text="截圖",command=self.take_screenshot,state=tk.DISABLED)self.screenshot_button.pack(side=tk.LEFT, padx=5)self.record_button = ttk.Button(control_frame,text="開始錄制",command=self.toggle_recording,state=tk.DISABLED)self.record_button.pack(side=tk.LEFT, padx=5)# 自定義按鈕樣式style = ttk.Style()style.configure('Accent.TButton', font=self.font)def toggle_connection(self):"""切換攝像頭連接狀態"""if self.is_playing:self.stop_camera()else:self.start_camera()def start_camera(self):"""啟動攝像頭"""try:self.status_var.set("正在連接...")self.connect_button.config(text="斷開連接")self.screenshot_button.config(state=tk.NORMAL)self.record_button.config(state=tk.NORMAL)# 打開攝像頭self.cap = cv2.VideoCapture(self.ip_url.get())if not self.cap.isOpened():raise Exception("無法連接到攝像頭，請檢查URL和網絡連接")# 獲取視頻幀尺寸self.frame_width = int(self.cap.get(3))self.frame_height = int(self.cap.get(4))# 啟動視頻播放線程self.is_playing = Trueself.status_var.set("已連接")threading.Thread(target=self.update_frame, daemon=True).start()except Exception as e:messagebox.showerror("錯誤", f"連接攝像頭時出錯: {str(e)}")self.status_var.set("連接失敗")self.connect_button.config(text="連接")self.screenshot_button.config(state=tk.DISABLED)self.record_button.config(state=tk.DISABLED)def stop_camera(self):"""停止攝像頭"""self.is_playing = Falseself.is_recording = Falseif self.cap:self.cap.release()self.cap = Noneif self.recorder:self.recorder.release()self.recorder = Noneself.connect_button.config(text="連接")self.status_var.set("未連接")self.record_button.config(text="開始錄制")self.screenshot_button.config(state=tk.DISABLED)self.record_button.config(state=tk.DISABLED)# 清空畫布self.canvas.delete("all")self.canvas.create_text(self.canvas_width/2, self.canvas_height/2, text="攝像頭已斷開", fill="white", font=self.font)def update_frame(self):"""更新視頻幀"""while self.is_playing:ret, frame = self.cap.read()if not ret:self.status_var.set("無法獲取視頻幀")break# 保存當前幀用于截圖self.frame = frame# 調整幀大小以適應畫布frame = cv2.resize(frame, (self.canvas_width, self.canvas_height))# 轉換為RGB并顯示rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)pil_image = Image.fromarray(rgb_frame)photo = ImageTk.PhotoImage(image=pil_image)# 在畫布上顯示圖像self.canvas.create_image(0, 0, image=photo, anchor=tk.NW)self.canvas.photo = photo# 錄制視頻if self.is_recording:self.recorder.write(frame)# 控制幀率time.sleep(0.03)  # 約30FPSdef toggle_recording(self):"""切換錄制狀態"""if self.is_recording:self.is_recording = Falseself.record_button.config(text="開始錄制")self.status_var.set("錄制已停止")if self.recorder:self.recorder.release()self.recorder = Noneelse:if not self.frame is None:# 創建保存目錄if not os.path.exists("recordings"):os.makedirs("recordings")# 設置輸出文件名和編碼器timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")output_file = f"recordings/camera_recording_{timestamp}.mp4"# 定義編碼器并創建VideoWriter對象fourcc = cv2.VideoWriter_fourcc(*'mp4v')self.recorder = cv2.VideoWriter(output_file, fourcc, 20.0,  # FPS(self.frame_width, self.frame_height))self.is_recording = Trueself.record_button.config(text="停止錄制")self.status_var.set("正在錄制...")else:messagebox.showinfo("提示", "沒有可用的視頻幀")def take_screenshot(self):"""拍攝截圖"""if not self.frame is None:# 創建保存目錄if not os.path.exists("screenshots"):os.makedirs("screenshots")# 設置輸出文件名timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")output_file = f"screenshots/camera_screenshot_{timestamp}.png"# 保存截圖cv2.imwrite(output_file, self.frame)self.status_var.set(f"截圖已保存至: {output_file}")# 顯示保存成功提示messagebox.showinfo("成功", f"截圖已保存至:\n{output_file}")else:messagebox.showinfo("提示", "沒有可用的視頻幀")if __name__ == "__main__":root = tk.Tk()app = IPCameraViewer(root)root.mainloop()