如何在swift中以编程方式多次约束不同情况下的视图？

import UIKit

class ViewController: UIViewController {
    var myView: UIView!
    var widthConstraint: NSLayoutConstraint!
    var heightConstraint: NSLayoutConstraint!

    override func viewDidLoad() {
        super.viewDidLoad()

        // 创建一个视图
        myView = UIView()
        myView.backgroundColor = UIColor.red
        myView.translatesAutoresizingMaskIntoConstraints = false
        view.addSubview(myView)

        // 设置视图的初始约束
        widthConstraint = myView.widthAnchor.constraint(equalToConstant: 100)
        heightConstraint = myView.heightAnchor.constraint(equalToConstant: 100)
        NSLayoutConstraint.activate([widthConstraint, heightConstraint])

        // 根据不同情况修改约束
        let condition = true // 根据实际情况设置条件
        if condition {
            widthConstraint.constant = 200
            heightConstraint.constant = 200
        } else {
            widthConstraint.constant = 50
            heightConstraint.constant = 50
        }

        // 更新视图布局
        view.layoutIfNeeded()
    }
}