fix(py): add pysince for timeit.default_timer

nimpylib · Oct 15, 2024 · 44316ec · 44316ec
1 parent 3ac6ae0
commit 44316ec
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/src/pylib/Lib/timeit.nim b/src/pylib/Lib/timeit.nim
@@ -6,7 +6,17 @@ import ../version
 import ./sys
 export list, noneType
 
-export n_timeit except newTimer, repeat, print_exc, default_repeat
+pysince(3,3):
+  const TimeItUseTime*{.booldefine: "timeit.usetime".} = true  ## \
+    ## disable this if don't wanna depending on `Lib/time`
+  when TimeItUseTime:
+    import ./time
+
+export n_timeit except newTimer, repeat, print_exc, default_repeat, default_timer
+
+when TimeItUseTime:
+  var default_timer* = pysince(3.3, time.perf_counter, n_timeit.default_timer)
+else: export n_timeit.default_timer
 
 template repeatImpl(xs: varargs[untyped]): untyped = n_timeit.repeat(xs)  ##\
 ## to avoid `repeat`'s repeat param being replaced

diff --git a/src/pylib/version.nim b/src/pylib/version.nim
@@ -71,7 +71,7 @@ func toVer(s: static float): MajorMinorVersion{.compileTime.} =
       " debug: delta=" & $(minorF.int.float - minorF)
   result.minor =  int minorF
 
-func pysince*[R](ver: static[float|MajorMinorVersion]; defExpr, elseExpr: R): R{.compileTime.} =
+func pysince*[R](ver: static[float|MajorMinorVersion]; defExpr, elseExpr: R): R{.inline.} =
   bind PyMajor, PyMinor, toVer
   when (PyMajor, PyMinor) >= toVer(ver): defExpr
   else: elseExpr