You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
2.3 KiB
71 lines
2.3 KiB
from hashlib import md5
|
|
from CodernityDB.hash_index import HashIndex
|
|
from CodernityDB.tree_index import TreeBasedIndex
|
|
|
|
|
|
class ReleaseIndex(TreeBasedIndex):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
kwargs['key_format'] = '32s'
|
|
super(ReleaseIndex, self).__init__(*args, **kwargs)
|
|
|
|
def make_key(self, key):
|
|
return key
|
|
|
|
def make_key_value(self, data):
|
|
if data.get('_t') == 'release' and data.get('media_id'):
|
|
return data['media_id'], None
|
|
|
|
def run_for_media(self, db, media_id):
|
|
for release in db.get_many('release', media_id, with_doc = True):
|
|
yield release['doc']
|
|
|
|
def run_with_status(self, db, status = [], with_doc = True):
|
|
|
|
status = list(status if isinstance(status, (list, tuple)) else [status])
|
|
|
|
for s in status:
|
|
for ms in db.get_many('release_status', s, with_doc = with_doc):
|
|
yield ms['doc'] if with_doc else ms
|
|
|
|
|
|
class ReleaseStatusIndex(TreeBasedIndex):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
kwargs['key_format'] = '32s'
|
|
super(ReleaseStatusIndex, self).__init__(*args, **kwargs)
|
|
|
|
def make_key(self, key):
|
|
return md5(key).hexdigest()
|
|
|
|
def make_key_value(self, data):
|
|
if data.get('_t') == 'release' and data.get('status'):
|
|
return md5(data.get('status')).hexdigest(), {'media_id': data.get('media_id')}
|
|
|
|
|
|
class ReleaseIDIndex(HashIndex):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
kwargs['key_format'] = '32s'
|
|
super(ReleaseIDIndex, self).__init__(*args, **kwargs)
|
|
|
|
def make_key(self, key):
|
|
return md5(key).hexdigest()
|
|
|
|
def make_key_value(self, data):
|
|
if data.get('_t') == 'release' and data.get('identifier'):
|
|
return md5(data.get('identifier')).hexdigest(), {'media_id': data.get('media_id')}
|
|
|
|
|
|
class ReleaseDownloadIndex(HashIndex):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
kwargs['key_format'] = '32s'
|
|
super(ReleaseDownloadIndex, self).__init__(*args, **kwargs)
|
|
|
|
def make_key(self, key):
|
|
return md5(key).hexdigest()
|
|
|
|
def make_key_value(self, data):
|
|
if data.get('_t') == 'release' and data.get('download_info') and data['download_info']['id'] and data['download_info']['downloader']:
|
|
return md5('%s-%s' % (data['download_info']['downloader'], data['download_info']['id'])).hexdigest(), None
|
|
|