pjht/buku
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chg: dev: use safe load only

Browse Source
This commit is contained in:
rachmadaniHaryono 2021-05-01 23:17:24 +08:00
parent 58adeb78bf
commit 03d808072f
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_bukuDb.py
View File

@ -1327,10 +1327,7 @@ def test_load_firefox_database(firefox_db, add_pt):
res_yaml_file = firefox_db[1] if add_pt else firefox_db[2] res_yaml_file = firefox_db[1] if add_pt else firefox_db[2]
if not dump_data: if not dump_data:
with open(res_yaml_file, "r") as f: with open(res_yaml_file, "r") as f:
try: res_yaml = yaml.load(f, Loader=PrettySafeLoader)
res_yaml = yaml.load(f)
except RuntimeError:
res_yaml = yaml.load(f, Loader=PrettySafeLoader)
# init # init
import buku import buku