mirror of
https://github.com/mvt-project/mvt.git
synced 2024-06-28 07:09:00 +00:00
24 lines
539 B
Python
24 lines
539 B
Python
|
import os
|
||
|
|
||
|
def get_artifact(fname):
|
||
|
"""
|
||
|
Return the artifact path in the artifact folder
|
||
|
"""
|
||
|
DATA_FOLDER = os.path.join(os.path.dirname(__file__), "artifacts")
|
||
|
fpath = os.path.join(DATA_FOLDER, fname)
|
||
|
if os.path.isfile(fpath):
|
||
|
return fpath
|
||
|
return
|
||
|
|
||
|
def get_artifact_folder():
|
||
|
return os.path.join(os.path.dirname(__file__), "artifacts")
|
||
|
|
||
|
def init_setup():
|
||
|
"""
|
||
|
init data to have a clean state before testing
|
||
|
"""
|
||
|
try:
|
||
|
del os.environ['MVT_STIX2']
|
||
|
except KeyError:
|
||
|
pass
|