JOJ/JOJ3-config-generator
27
1
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

feat: only support str limit in toml
Some checks failed
build / build (push) Failing after 2m35s
Details
build / build (pull_request) Failing after 2m33s
Details

Browse Source
This commit is contained in:
张泊明518370910136 2025-03-04 22:44:37 -05:00
parent 719f19232b
commit d6a33d2f20
Signed by: 张泊明518370910136
GPG Key ID: D47306D7062CDA9D
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
joj3_config_generator/models/task.py
View File

@ -60,20 +60,24 @@ class Files(BaseModel):
class Limit(BaseModel): class Limit(BaseModel):
mem: int = DEFAULT_MEMORY_LIMIT mem: int = DEFAULT_MEMORY_LIMIT
cpu: int = DEFAULT_CPU_LIMIT cpu: int = DEFAULT_CPU_LIMIT
stderr: int = DEFAULT_FILE_LIMIT
stdout: int = DEFAULT_FILE_LIMIT stdout: int = DEFAULT_FILE_LIMIT
stderr: int = DEFAULT_FILE_LIMIT
model_config = ConfigDict(validate_assignment=True) model_config = ConfigDict(validate_assignment=True)
@field_validator("cpu", mode="before") @field_validator("cpu", mode="before")
@classmethod @classmethod
def ensure_time(cls, v: Any) -> Time: def ensure_time(cls, v: Any) -> Time:
return Time(v) if isinstance(v, str):
return Time(v)
raise ValueError("Must be a string")
@field_validator("mem", "stdout", "stderr", mode="before") @field_validator("mem", "stdout", "stderr", mode="before")
@classmethod @classmethod
def ensure_mem(cls, v: Any) -> Memory: def ensure_mem(cls, v: Any) -> Memory:
return Memory(v) if isinstance(v, str):
return Memory(v)
raise ValueError("Must be a string")
class Stage(BaseModel): class Stage(BaseModel):