rbx.cp 0.9.1__tar.gz → 0.9.2__tar.gz

{rbx_cp-0.9.1 → rbx_cp-0.9.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: rbx.cp
-Version: 0.9.1
+Version: 0.9.2
 Summary: 
 Author: Roberto Sales
 Requires-Python: >=3.9.1,<4.0.0

{rbx_cp-0.9.1 → rbx_cp-0.9.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "rbx.cp"
-version = "0.9.1"
+version = "0.9.2"
 description = ""
 packages = [
   {include = "rbx"}

{rbx_cp-0.9.1 → rbx_cp-0.9.2}/rbx/grading/grading_context.py

@@ -1,7 +1,8 @@
 import contextvars
-from enum import Enum
 from typing import Callable, Optional, Union
 
+from rbx.autoenum import AutoEnum, alias
+
 Condition = Union[bool, Callable[[], bool]]
 
 
@@ -15,11 +16,11 @@ class ConditionedContext:
         return self.when()
 
 
-class CacheLevel(Enum):
-    NO_CACHE = 0
-    CACHE_TRANSIENTLY = 1
-    CACHE_COMPILATION = 2
-    CACHE_ALL = 3
+class CacheLevel(AutoEnum):
+    NO_CACHE = alias('none')
+    CACHE_TRANSIENTLY = alias('transient')
+    CACHE_COMPILATION = alias('compilation')
+    CACHE_ALL = alias('all')
 
 
 cache_level_var = contextvars.ContextVar('cache_level', default=CacheLevel.CACHE_ALL)
@@ -30,11 +31,11 @@ def is_compilation_only() -> bool:
 
 
 def is_transient() -> bool:
-    return cache_level_var.get().value <= CacheLevel.CACHE_TRANSIENTLY.value
+    return cache_level_var.get() == CacheLevel.CACHE_TRANSIENTLY or is_no_cache()
 
 
 def is_no_cache() -> bool:
-    return cache_level_var.get().value <= CacheLevel.NO_CACHE.value
+    return cache_level_var.get() == CacheLevel.NO_CACHE
 
 
 class cache_level(ConditionedContext):