CIRCL
/
lookyloo
mirror of https://github.com/CIRCL/lookyloo
2
0
Fork 0
Code Issues Releases Wiki Activity

chg: Remove unused parameter.

pull/78/head
Raphaël Vinot 2020-07-06 18:15:03 +02:00
parent 0c5501016c
commit db0d5b3f2d
1 changed files with 7 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
lookyloo/lookyloo.py
View File

@ -210,24 +210,14 @@ class Lookyloo():
with self_generated_ua_file.open('w') as f: with self_generated_ua_file.open('w') as f:
json.dump(to_store, f, indent=2) json.dump(to_store, f, indent=2)
def cache_tree(self, capture_uuid: Optional[str]=None, capture_dir: Optional[Union[str, Path]]=None) -> None: def cache_tree(self, capture_uuid: str) -> None:
c_dir = None capture_dir = self.lookup_capture_dir(capture_uuid)
if capture_uuid: if not capture_dir:
c_dir = self.lookup_capture_dir(capture_uuid)
if not c_dir:
raise MissingUUID(f'Unable to find UUID {capture_uuid} in the cache') raise MissingUUID(f'Unable to find UUID {capture_uuid} in the cache')
elif capture_dir:
if isinstance(capture_dir, str):
c_dir = Path(capture_dir)
else:
c_dir = capture_dir
if not c_dir: with open((capture_dir / 'uuid'), 'r') as f:
raise Exception('Need either capture_uuid or capture_dir')
with open((c_dir / 'uuid'), 'r') as f:
uuid = f.read() uuid = f.read()
har_files = sorted(c_dir.glob('*.har')) har_files = sorted(capture_dir.glob('*.har'))
try: try:
ct = CrawledTree(har_files, uuid) ct = CrawledTree(har_files, uuid)
self.indexing.index_cookies_capture(ct) self.indexing.index_cookies_capture(ct)
@ -235,7 +225,7 @@ class Lookyloo():
except Har2TreeError as e: except Har2TreeError as e:
raise NoValidHarFile(e.message) raise NoValidHarFile(e.message)
with (c_dir / 'tree.pickle').open('wb') as _p: with (capture_dir / 'tree.pickle').open('wb') as _p:
pickle.dump(ct, _p) pickle.dump(ct, _p)
def get_crawled_tree(self, capture_uuid: str) -> CrawledTree: def get_crawled_tree(self, capture_uuid: str) -> CrawledTree: