Example extension UIView for creating IOS screenshot

ExtensionView.swift

//
//  ExtensionView.swift
//
//  Created by Vasiliy Savchuk on 2/6/17.
//  Copyright © 2017 All rights reserved.
//
// Usage this snippet:
// yourImageView.image = view.screenshot

import UIKit

extension UIView {
    var screenshot: UIImage {
        UIGraphicsBeginImageContext(self.bounds.size)
        let context = UIGraphicsGetCurrentContext()
        self.layer.render(in: context!)
        let screenShot = UIGraphicsGetImageFromCurrentImageContext()
        UIGraphicsEndImageContext()
        return screenShot!
    }
}