# -*- coding: utf-8 -*- import time,xbmc,logging from resources.modules.client import get_html from resources.modules import log class AllDebrid: def __init__(self): from resources.modules import tools self.tools=tools self.agent_identifier = self.tools.addonName self.token = self.tools.getSetting('alldebrid.token') self.base_url = 'https://api.alldebrid.com/v4/' if self.token=='': self.auth() def get_url(self, url, token_req=False): if self.token == '': return url = '{}{}'.format(self.base_url, url) if not '?' in url: url += '?' url += 'agent={}'.format(self.agent_identifier) else: url += '&agent={}'.format(self.agent_identifier) if token_req: url += '&apikey={}'.format(self.token) return get_html(url).json() def post_url(self, url, post_data=None, token_req=False): if self.token == '': return url = '{}{}'.format(self.base_url, url) if not '?' in url: url += '?' url += 'agent={}'.format(self.agent_identifier) else: url += '&agent={}'.format(self.agent_identifier) if token_req: url += '&apikey={}'.format(self.token) log.warning('AD test:'+url) a=get_html(url, data=post_data).json() log.warning(a) return a def auth(self): pin_url = '{}pin/get?agent={}'.format(self.base_url, self.agent_identifier) resp = get_html(pin_url).json()['data'] expiry = pin_ttl = int(resp['expires_in']) auth_complete = False self.tools.copy2clip(resp['pin']) self.tools.progressDialog.create('{} - {}'.format(self.tools.addonName, 'AllDebrid Auth')) self.tools.progressDialog.update(100, 'Open this link in a browser: {}'.format(self.tools.colorString(resp['base_url']))+'\n'+ 'Enter the code: {}'.format(self.tools.colorString( resp['pin']))+'\n'+ 'This code has been copied to your clipboard') # Seems the All Debrid servers need some time do something with the pin before polling # Polling to early will cause an invalid pin error time.sleep(5) while not auth_complete and not expiry <= 0 and not self.tools.progressDialog.iscanceled(): auth_complete, expiry = self.poll_auth(resp['check_url']) progress_percent = 100 - int((float(pin_ttl - expiry) / pin_ttl) * 100) self.tools.progressDialog.update(progress_percent) time.sleep(1) try:self.tools.progressDialog.close() except:pass self.store_user_info() if auth_complete: xbmc.executebuiltin((u'Notification(%s,%s)' % (self.tools.addonName, ('Authentication is completed')))) else: return def poll_auth(self, poll_url): resp = get_html(poll_url).json()['data'] if resp['activated']: self.tools.setSetting('alldebrid.token', resp['apikey']) self.token = resp['apikey'] return True, 0 return False, int(resp['expires_in']) def store_user_info(self): user_information = self.get_url('user', True) self.tools.setSetting('alldebrid.username', user_information['data']['user']['username']) return def check_hash(self, hash_list): all_mag=[] for itt in hash_list: all_mag.append(itt) post_data = {'magnets[]': hash_list} return self.get_url('magnet/instant?magnets[]='+'&magnets[]='.join(all_mag), token_req=True) return self.post_url('magnet/instant', post_data, True) def upload_magnet(self, hash): return self.get_url('magnet/upload?magnet={}'.format(hash), token_req=True) def update_relevant_hosters(self): return self.get_url('hosts') def get_hosters(self, hosters): import database host_list = database.get(self.update_relevant_hosters, 1) if host_list is None: host_list = self.update_relevant_hosters() if host_list is not None: hosters['premium']['all_debrid'] = [(i['domain'], i['domain'].split('.')[0]) for i in host_list['hosts'] if i['status']] hosters['premium']['all_debrid'] += [(host, host.split('.')[0]) for i in host_list['hosts'] if 'altDomains' in i and i['status'] for host in i['altDomains']] else: import traceback traceback.print_exc() hosters['premium']['all_debrid'] = [] def resolve_hoster(self, url): url = self.tools.quote(url) resolve = self.get_url('link/unlock?link={}'.format(url), token_req=True) if resolve['status']=='success': return resolve['data']['link'] else: return None def magnet_status(self, magnet_id): return self.get_url('magnet/status?id={}'.format(magnet_id), token_req=True) def movie_magnet_to_stream(self, magnet): selectedFile = None magnet_id = self.upload_magnet(magnet) log.warning(magnet_id) if 'error' in magnet_id: xbmc.executebuiltin(u'Notification(%s,%s)' % (self.tools.addonName+' Error', magnet_id['error']['message'])) return 0 magnet_id =magnet_id ['data']['magnets'][0]['id'] all_lk=(self.magnet_status(magnet_id)) folder_details = self.magnet_status(magnet_id)['data']['magnets']['links'] for items in folder_details: if 'mkv' in items['filename'] or 'avi' in items['filename'] or 'mp4' in items['filename']: selectedFile = items['link'] self.delete_magnet(magnet_id) return self.resolve_hoster(selectedFile) def resolve_magnet(self, magnet, args, torrent, pack_select=False): import source_utils if 'showInfo' not in args: return self.movie_magnet_to_stream(magnet, args) magnet_id = self.upload_magnet(magnet) magnet_id = magnet_id['id'] episodeStrings, seasonStrings = source_utils.torrentCacheStrings(args) try: folder_details = self.magnet_status(magnet_id) if folder_details['status'] != 'Ready': return folder_details = [{'link': key, 'filename': value} for key, value in folder_details['links'].items()] if 'extra' not in args['info']['title'] and 'extra' not in args['showInfo']['info']['tvshowtitle'] \ and int(args['info']['season']) != 0: folder_details = [i for i in folder_details if 'extra' not in source_utils.cleanTitle(i['filename'].replace('&', ' ').lower())] if 'special' not in args['info']['title'] and 'special' not in args['showInfo']['info']['tvshowtitle'] \ and int(args['info']['season']) != 0: folder_details = [i for i in folder_details if 'special' not in source_utils.cleanTitle(i['filename'].replace('&', ' ').lower())] streamLink = self.check_episode_string(folder_details, episodeStrings) if streamLink is None: return self.delete_magnet(magnet_id) return self.resolve_hoster(streamLink) except: import traceback traceback.print_exc() pass def check_episode_string(self, folder_details, episodeStrings): import source_utils for i in folder_details: for epstring in episodeStrings: if epstring in source_utils.cleanTitle(i['filename'].replace('&', ' ').lower()): if any(i['filename'].endswith(ext) for ext in source_utils.COMMON_VIDEO_EXTENSIONS): return i['link'] return None def delete_magnet(self, magnet_id): return self.get_url('magnet/delete?id={}'.format(magnet_id), token_req=True)