mirror of https://gitlab.com/SIGBUS/nyaa.git
62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
import hashlib
|
|
import functools
|
|
from collections import OrderedDict
|
|
|
|
|
|
def sha1_hash(input_bytes):
|
|
""" Hash given bytes with hashlib.sha1 and return the digest (as bytes) """
|
|
return hashlib.sha1(input_bytes).digest()
|
|
|
|
|
|
def sorted_pathdict(input_dict):
|
|
""" Sorts a parsed torrent filelist dict by alphabat, directories first """
|
|
directories = OrderedDict()
|
|
files = OrderedDict()
|
|
|
|
for key, value in input_dict.items():
|
|
if isinstance(value, dict):
|
|
directories[key] = sorted_pathdict(value)
|
|
else:
|
|
files[key] = value
|
|
|
|
return OrderedDict(sorted(directories.items()) + sorted(files.items()))
|
|
|
|
|
|
def cached_function(f):
|
|
sentinel = object()
|
|
f._cached_value = sentinel
|
|
|
|
@functools.wraps(f)
|
|
def decorator(*args, **kwargs):
|
|
if f._cached_value is sentinel:
|
|
print('Evaluating', f, args, kwargs)
|
|
f._cached_value = f(*args, **kwargs)
|
|
return f._cached_value
|
|
return decorator
|
|
|
|
|
|
def flattenDict(d, result=None):
|
|
if result is None:
|
|
result = {}
|
|
for key in d:
|
|
value = d[key]
|
|
if isinstance(value, dict):
|
|
value1 = {}
|
|
for keyIn in value:
|
|
value1["/".join([key, keyIn])] = value[keyIn]
|
|
flattenDict(value1, result)
|
|
elif isinstance(value, (list, tuple)):
|
|
for indexB, element in enumerate(value):
|
|
if isinstance(element, dict):
|
|
value1 = {}
|
|
index = 0
|
|
for keyIn in element:
|
|
newkey = "/".join([key, keyIn])
|
|
value1["/".join([key, keyIn])] = value[indexB][keyIn]
|
|
index += 1
|
|
for keyA in value1:
|
|
flattenDict(value1, result)
|
|
else:
|
|
result[key] = value
|
|
return result
|