class FileManager:
    def __init__(self, filename, mode):
        self.filename = filename
        self.mode = mode
        self.file = None

    def __enter__(self):
        print(f"? 正在打開文件: {self.filename}")
        self.file = open(self.filename, self.mode)
        return self.file

    def __exit__(self, exc_type, exc_val, exc_tb):
        if self.file:
            self.file.close()
            print(f"? 已關(guān)閉文件: {self.filename}")
        # 可選：異常處理邏輯
        if exc_type:
            print(f"? 發(fā)生異常: {exc_val}")
        return False  # 不抑制異常

# 使用示例
with FileManager("example.txt", "w") as f:
    f.write("Hello, World!\n")
    f.write("This is a test.\n")
    # 模擬異常
    # raise ValueError("Something went wrong!")

# ? 壞習(xí)慣：手動(dòng)管理
f = open("data.txt", "r")
content = f.read()
f.close()

# ? 好做法：使用 with
with open("data.txt", "r") as f:
    content = f.read()
    print(content)
# file 自動(dòng)關(guān)閉！

with open("log.txt", "a") as log_file:
    log_file.write(f"[{datetime.now()}] User logged in\n")
    log_file.write(f"[{datetime.now()}] Data processed successfully\n")

import os

def process_large_file(filename):
    total_lines = 0
    total_size = 0

    with open(filename, "r", encoding="utf-8") as file:
        for line in file:
            total_lines += 1
            total_size += len(line)
            if total_lines % 1000 == 0:
                print(f"?? 已處理 {total_lines} 行，當(dāng)前大小: {total_size} bytes")

    print(f"?? 總計(jì): {total_lines} 行，總大小: {total_size} bytes")

# 調(diào)用函數(shù)
process_large_file("huge_log.txt")

import random

with open("test.txt", "w") as f:
    f.write("First line\n")
    # 隨機(jī)拋出異常
    if random.choice([True, False]):
        raise RuntimeError("Simulated error during file write!")
    f.write("Second line\n")

from contextlib import contextmanager
import json
import pickle
from datetime import datetime

@contextmanager
def safe_file_handler(filename, mode="r", encoding="utf-8"):
    """
    安全的文件上下文管理器，支持自動(dòng)重試和日志記錄
    """
    file_obj = None
    retry_count = 0
    max_retries = 3

    while retry_count < max_retries:
        try:
            print(f"?? 嘗試打開文件: {filename} (第 {retry_count + 1} 次)")
            file_obj = open(filename, mode, encoding=encoding)
            break  # 成功打開，跳出循環(huán)
        except (IOError, OSError) as e:
            retry_count += 1
            print(f"?? 打開失敗: {e}, {max_retries - retry_count} 次重試機(jī)會(huì)")
            if retry_count >= max_retries:
                raise RuntimeError(f"? 無法打開文件 {filename}，已重試 {max_retries} 次") from e

    try:
        yield file_obj
    except Exception as e:
        print(f"?? 處理過程中發(fā)生異常: {e}")
        raise
    finally:
        if file_obj and not file_obj.closed:
            file_obj.close()
            print(f"? 已關(guān)閉文件: {filename}")

# ? 使用示例
if __name__ == "__main__":
    # 寫入數(shù)據(jù)
    with safe_file_handler("config.json", "w") as f:
        data = {"app": "MyApp", "version": "1.0", "created_at": str(datetime.now())}
        json.dump(data, f, indent=2)

    # 讀取數(shù)據(jù)
    with safe_file_handler("config.json", "r") as f:
        config = json.load(f)
        print("?? 配置加載成功:", config)

# 錯(cuò)誤認(rèn)知：only for files
with open("file.txt", "r") as f:
    pass

# 數(shù)據(jù)庫連接
import sqlite3

with sqlite3.connect("example.db") as conn:
    cursor = conn.cursor()
    cursor.execute("CREATE TABLE IF NOT EXISTS users (id INTEGER, name TEXT)")
    cursor.execute("INSERT INTO users VALUES (?, ?)", (1, "Alice"))
    conn.commit()

# 網(wǎng)絡(luò)請(qǐng)求（模擬）
import requests

with requests.Session() as session:
    resp = session.get("https://httpbin.org/get")
    print(resp.json())

# ? 錯(cuò)誤寫法（非必須）
with open("in.txt", "r") as fin:
    with open("out.txt", "w") as fout:
        for line in fin:
            fout.write(line.upper())

# ? 推薦寫法：鏈?zhǔn)角短?
with open("in.txt", "r") as fin, open("out.txt", "w") as fout:
    for line in fin:
        fout.write(line.upper())

from contextlib import contextmanager
import time

@contextmanager
def timer():
    start = time.time()
    print("?? 開始計(jì)時(shí)...")
    try:
        yield
    finally:
        end = time.time()
        print(f"? 任務(wù)耗時(shí): {end - start:.4f} 秒")

# 使用示例
with timer():
    time.sleep(1)
    print("? 任務(wù)完成")

import time
import os

def test_manual_close():
    files = []
    start = time.time()
    for i in range(1000):
        f = open(f"data_{i}.txt", "w")
        f.write(f"Line {i}\n")
        files.append(f)
    for f in files:
        f.close()
    print(f"?? 手動(dòng)關(guān)閉耗時(shí): {time.time() - start:.4f} 秒")

def test_with_statement():
    start = time.time()
    for i in range(1000):
        with open(f"data_{i}.txt", "w") as f:
            f.write(f"Line {i}\n")
    print(f"? with 語句耗時(shí): {time.time() - start:.4f} 秒")

# 執(zhí)行測(cè)試（注意：首次運(yùn)行會(huì)創(chuàng)建文件）
test_manual_close()
test_with_statement()