如何不从图片库iOS中获取截图

import UIKit

func captureScreenshot() -> UIImage? {
    guard let screen = UIScreen.main else {
        return nil
    }
    UIGraphicsBeginImageContextWithOptions(screen.bounds.size, false, 0)
    guard let context = UIGraphicsGetCurrentContext() else {
        return nil
    }
    screen.drawHierarchy(in: screen.bounds, afterScreenUpdates: false)
    let screenshot = UIGraphicsGetImageFromCurrentImageContext()
    UIGraphicsEndImageContext()
    return screenshot
}

// 调用方法
if let screenshot = captureScreenshot() {
    // 处理截图
}

import UIKit

func captureScreenshot() -> UIImage? {
    guard let window = UIApplication.shared.keyWindow else {
        return nil
    }
    UIGraphicsBeginImageContextWithOptions(window.bounds.size, false, 0)
    guard let context = UIGraphicsGetCurrentContext() else {
        return nil
    }
    window.drawHierarchy(in: window.bounds, afterScreenUpdates: false)
    let screenshot = UIGraphicsGetImageFromCurrentImageContext()
    UIGraphicsEndImageContext()
    return screenshot
}

// 调用方法
if let screenshot = captureScreenshot() {
    // 处理截图
}