浏览代码

support more padding char

BLUELOVETH 2 年之前
父节点
当前提交
d3d07d060c
共有 1 个文件被更改,包括 6 次插入3 次删除
  1. 6 3
      src/vm.cpp

+ 6 - 3
src/vm.cpp

@@ -397,9 +397,12 @@ PyObject* VM::format(Str spec, PyObject* obj){
     }
 
     char pad_c = ' ';
-    if(spec[0] == '0'){
-        pad_c = '0';
-        spec = spec.substr(1);
+    for(char c: std::string_view("0-=*#@!~")){
+        if(spec[0] == c){
+            pad_c = c;
+            spec = spec.substr(1);
+            break;
+        }
     }
     char align;
     if(spec[0] == '>'){