Skip to content
This repository was archived by the owner on Apr 30, 2021. It is now read-only.

Simplify a bit the code for the Corpus' constructor #5

Merged
yevgenypats merged 1 commit into from
Dec 18, 2019
Merged

Simplify a bit the code for the Corpus' constructor #5

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions pythonfuzz/corpus.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ class Corpus(object):
def __init__(self, dirs=None, max_input_size=4096):
self._inputs = []
self._max_input_size = max_input_size
self._dirs = [] if dirs is None else dirs
self._dirs = dirs if dirs else []
for i, path in enumerate(dirs):
if i == 0 and not os.path.exists(path):
os.mkdir(path)
Expand All @@ -27,9 +27,9 @@ def __init__(self, dirs=None, max_input_size=4096):
fname = os.path.join(path, i)
if os.path.isfile(fname):
self._add_file(fname)
self._seed_run_finished = True if len(self._inputs) == 0 else False
self._seed_run_finished = not self._inputs
self._seed_idx = 0
self._save_corpus = True if len(dirs) > 0 and os.path.isdir(dirs[0]) else False
self._save_corpus = dirs and os.path.isdir(dirs[0])

def _add_file(self, path):
with open(path, 'rb') as f:
Expand Down