mirror of https://github.com/corca-ai/EVAL
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
19 lines
513 B
Python
19 lines
513 B
Python
1 year ago
|
from pathlib import Path
|
||
|
|
||
|
from env import settings
|
||
|
|
||
|
|
||
|
def verify(func):
|
||
|
def wrapper(*args, **kwargs):
|
||
|
try:
|
||
|
filepath = args[0].filepath
|
||
|
except:
|
||
|
raise Exception("This tool doesn't have filepath. Please check your code.")
|
||
|
if not str(Path(filepath).resolve()).startswith(
|
||
|
str(Path(settings["PLAYGROUND_DIR"]).resolve())
|
||
|
):
|
||
|
return "You can't access file outside of playground."
|
||
|
return func(*args, **kwargs)
|
||
|
|
||
|
return wrapper
|