feat: implement constant folding for str.format

Author: BobTheBuidler

mypy/constant_fold.py

@@ -77,22 +77,34 @@ def constant_fold_expr(expr: Expression, cur_mod_id: str) -> ConstantValue | Non
         value = constant_fold_expr(expr.expr, cur_mod_id)
         if value is not None:
             return constant_fold_unary_op(expr.op, value)
-    # --- partial str.join support in preparation for f-string constant folding ---
+    # --- f-string requires partial support for both str.join and str.format ---
     elif (
         isinstance(expr, CallExpr)
         and isinstance(callee := expr.callee, MemberExpr)
         and isinstance(folded_callee := constant_fold_expr(callee.expr, cur_mod_id), str)
-        and callee.name == "join"
-        and len(args := expr.args) == 1
-        and isinstance(arg := args[0], (ListExpr, TupleExpr))
     ):
-        folded_items = []
-        for item in arg.items:
-            val = constant_fold_expr(item, cur_mod_id)
-            if not isinstance(val, str):
-                return None
-            folded_items.append(val)
-        return folded_callee.join(folded_items)
+        # --- partial str.join constant folding ---
+        if (
+            callee.name == "join"
+            and len(args := expr.args) == 1
+            and isinstance(arg := args[0], (ListExpr, TupleExpr))
+        ):
+            folded_items: list[str] = []
+            for item in arg.items:
+                val = constant_fold_expr(item, cur_mod_id)
+                if not isinstance(val, str):
+                    return None
+                folded_items.append(val)
+            return folded_callee.join(folded_items)
+        # --- str.format constant folding
+        elif callee.name == "format":
+            folded_args: list[str] = []
+            for arg in expr.args:
+                arg_val = constant_fold_expr(arg, cur_mod_id)
+                if arg_val is None:
+                    return None
+                folded_args.append(arg_val)
+            return folded_callee.format(*folded_args)
     return None