コード例 #1
0
def automatic_downloader(mode):
	dp = xbmcgui.DialogProgress()
	dp.create('TVLogo Downloader')
	dp.update(0,'Getting user channels')
	failed_log = []
	#grab all user channel groups
	#TV
	groupids = []
	ch_ids = []
	ch_names = []
	
	
	json_response = xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "PVR.GetChannelGroups", "params": {"channeltype" : "tv"}, "id": 1 }')
	decoded_data = json.loads(json_response)
	try: 
		groups = decoded_data['result']['channelgroups']
	except:
		mensagemok('TVLogo Downloader','Live TV is not enabled in kodi or no channels are available.')
		sys.exit(0)
	for x in range(0, len(decoded_data['result']['channelgroups'])):
		has_channels = False
		if groups[x]["channelgroupid"] > -1:
			json_response = xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "PVR.GetChannels", "params": {"channelgroupid" : '+str(groups[x]["channelgroupid"])+',"properties":["channel","channeltype","thumbnail","broadcastnow","broadcastnext"]}, "id": 1 }')
			channel_number = bool("channels" in json.loads(json_response)["result"])
			if channel_number: has_channels = True
		if has_channels:
			groupids.append(groups[x]["channelgroupid"])
	
	#Radio
	json_response = xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "PVR.GetChannelGroups", "params": {"channeltype" : "radio"}, "id": 1 }')
	decoded_data = json.loads(json_response)
	groups = decoded_data['result']['channelgroups']
	for x in range(0, len(decoded_data['result']['channelgroups'])):
	#check if group has channels
		has_channels = False
		if groups[x]["channelgroupid"] > -1:
			json_response = xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "PVR.GetChannels", "params": {"channelgroupid" : '+str(groups[x]["channelgroupid"])+',"properties":["channel","channeltype","thumbnail","broadcastnow","broadcastnext"]}, "id": 1 }')
			channel_number = bool("channels" in json.loads(json_response)["result"])
			if channel_number: has_channels = True
		if has_channels:
			groupids.append(groups[x]["channelgroupid"])
			
	#grab every channel name and label
	if groupids:
		for groupid in groupids:
			json_response = xbmc.executeJSONRPC('{"jsonrpc": "2.0", "method": "PVR.GetChannels", "params": {"channelgroupid" : '+str(groupid)+',"properties":["channel","channeltype","thumbnail"]}, "id": 1 }')
			decoded_data = json.loads(json_response)
			if "channels" in decoded_data["result"].keys():
				for channel in decoded_data["result"]["channels"]:
					if mode == 'all':
						if channel["label"].encode('utf-8') not in ch_names:
							ch_ids.append(channel["channelid"])
							ch_names.append(channel["label"])
					else:
						json_response = xbmc.executeJSONRPC('{"jsonrpc":"2.0","method":"Settings.GetSettingValue","params":{"setting":"pvrmenu.iconpath"},"id":9}')
						decoded_data = json.loads(json_response)
						logo_folder = decoded_data['result']['value'].encode('utf-8')
						label=channel["label"].encode('utf-8')
						logofile = os.path.join(logo_folder,label.replace('/','_')+'.png')
						if label not in ch_names and channel['thumbnail'].replace('image://','') != (logo_folder + label.replace(' ','%20')+'.png').replace('/','%2f') + '/' and not os.path.exists(logofile):
							ch_ids.append(channel["channelid"])
							ch_names.append(label)
	if ch_names:
		download_list = []
		initconst = const.Constr().restart_process()
		totalchannels = len(ch_names)
		
		i=1
		iscanceled = False
		for channel in ch_names:
			dp.update(i*100/totalchannels,'Processing channel '+channel)
			if not dp.iscanceled():
                                try:
                                        channel=channel.encode('utf-8')
                                except:pass
				match = thelogodb.Channels().by_keyword(urllib.quote_plus(get_replaced_names(channel)))
				if match:
					match = return_only_valid(match)
					if not match:
						if ' hq' in channel.lower():
							match = tvlogodownloader.get_nonhd_match(get_replaced_names(channel.lower().replace('hq',' hd')))
							match = return_only_valid(match)
						if not match:
							match = tvlogodownloader.get_nonhd_match(get_replaced_names(channel))
							match = return_only_valid(match)
							if not match:
								#check if nonascii version exists
								match = thelogodb.Channels().by_keyword(urllib.quote_plus(removeNonAscii(get_replaced_names(channel))))
								match = return_only_valid(match)
								#if no match check if channel is HD and grab logos for nonhd
								if not match:
									match = tvlogodownloader.get_nonhd_match(removeNonAscii(get_replaced_names(channel)))
									match = return_only_valid(match)
				else:
					if ' hq' in channel.lower():
						match = tvlogodownloader.get_nonhd_match(get_replaced_names(channel.lower().replace('hq',' hd')))
						match = return_only_valid(match)
					if not match:
						match = tvlogodownloader.get_nonhd_match(get_replaced_names(channel))
						match = return_only_valid(match)
						if not match:
							#check if nonascii version exists
							match = thelogodb.Channels().by_keyword(urllib.quote_plus(removeNonAscii(get_replaced_names(channel))))
							match = return_only_valid(match)
							#if no match check if channel is HD and grab logos for nonhd
							if not match:
								match = tvlogodownloader.get_nonhd_match(removeNonAscii(get_replaced_names(channel)))
								match = return_only_valid(match)
				
				if not match:
					if channel not in failed_log:
						failed_log.append(channel)
				if match:
					if len(match) == 1:
						obj = {'channel_name': match[0]["strChannel"],'channel_logo': match[0]["strLogoWide"],'selected_channel':channel}
						const.Constr().add_to_array(obj)
					else:
						#manipulations
						seqmatch = {}
						for canal in match:
							#try:print canal["strChannel"].lower(),channel.lower()
							#except: pass
							ratio = int(difflib.SequenceMatcher(None, canal["strChannel"].lower(),channel.lower()).ratio()*100)
							seqmatch[ratio] = canal
						
						already = False
						
						if settings.getSetting('auto_if_100') == 'true':
							if max(seqmatch.keys()) == 100: 
								already = True
								obj = {'channel_name': seqmatch[max(seqmatch.keys())]["strChannel"],'channel_logo': seqmatch[max(seqmatch.keys())]["strLogoWide"],'selected_channel':channel}
								const.Constr().add_to_array(obj)
						if settings.getSetting('auto_if_multiple') == 'true' and not already:
							if max(seqmatch.keys()) >= int(settings.getSetting("minimum_ratio")):
								already = True
								obj = {'channel_name': seqmatch[max(seqmatch.keys())]["strChannel"],'channel_logo': seqmatch[max(seqmatch.keys())]["strLogoWide"],'selected_channel':channel}
								const.Constr().add_to_array(obj)
						
						if not already:
							xbmc.sleep(3000)
							before = const.Constr().return_array()
							logowindow.start(match,"False","False",selected_channel=channel)
							now = const.Constr().return_array()
							if before == now:
								if channel not in failed_log:
									failed_log.append(channel)
							del before
							del now
				else:
					if channel not in failed_log:
						failed_log.append(channel)
				i +=1
			else:
				iscanceled = True
				dp.close()
		if not iscanceled:
			dp.update(100)
			dp.close()
			logos_to_download = const.Constr().return_array()
			downloader.Downloader(logos_to_download,True,failed_log)
			postprocessing.run()
			tvlogodownloader.main_menu(select=False,choose='')
コード例 #2
0
def specific_channels():
    options = ["By Country", "By Country Package", "By Package", "Search"]
    optionsvar = ["country", "country_package", "package", "search"]
    choose = xbmcgui.Dialog().select('TVLogo Downloader', options)
    if choose > -1:
        if optionsvar[choose] == 'country':
            countries = thelogodb.Channels().get_countries()
            if countries:
                country_list = []
                for country in countries:
                    if country["strCountry"] and country != 'None':
                        country_list.append(country["strCountry"])
                choose = xbmcgui.Dialog().select('TVLogo Downloader',
                                                 country_list)
                if choose > -1:
                    channels = thelogodb.Channels().by_country(
                        urllib.quote(country_list[choose]))
                    if channels:
                        logowindow.start(channels, "True", "True")
                        specific_channels()
                    else:
                        mensagemok('TVLogo Downloader',
                                   'No channels with logos in thelogodb!')
                        specific_channels()
                else:
                    specific_channels()
            else:
                specific_channels()

        elif optionsvar[choose] == 'country_package':
            country_list = []
            packages = thelogodb.Packages().get_all()
            if packages:
                for package in packages:
                    country = package["strCountry"]
                    if country and country != 'None' and country not in country_list:
                        country_list.append(country)
                if country_list:
                    country_list = sorted(country_list)
                    choose = xbmcgui.Dialog().select('TVLogo Downloader',
                                                     country_list)
                    if choose > -1:
                        country = country_list[choose]
                        packages = thelogodb.Packages().get_all()
                        package_list = []
                        package_id_list = []
                        if packages:
                            for package in packages:
                                if package["strCountry"] == country:
                                    package_label = '[' + str(
                                        package['strCountry']
                                    ) + '] ' + package[
                                        'strPackage'] + ' (' + str(
                                            package['strType']) + ')'
                                    package_id = package['idPackage']
                                    package_list.append(package_label)
                                    package_id_list.append(package_id)
                            if package_list:
                                choose = xbmcgui.Dialog().select(
                                    'TVLogo Downloader', package_list)
                                if choose > -1:
                                    channels = thelogodb.Channels().by_package(
                                        package_id_list[choose])
                                    channels_have_logos = False
                                    for channel in channels:
                                        if channel["strLogoWide"]:
                                            channels_have_logos = True
                                    if channels and channels_have_logos:
                                        logowindow.start(
                                            channels, "True", "True")
                                        specific_channels()
                                    else:
                                        mensagemok(
                                            'TVLogo Downloader',
                                            'No logos available for this package!'
                                        )
                                        specific_channels()
                                else:
                                    entire_packages()
                    else:
                        mensagemok('TVLogo Downloader',
                                   'No packages available!')
                        entire_packages()
                else:
                    mensagemok('TVLogo Downloader', 'Error getting packages!')
                    entire_packages()
            else:
                mensagemok('TVLogo Downloader', 'No packages available!')
                entire_packages()
        elif optionsvar[choose] == 'package':
            packages = thelogodb.Packages().get_all()
            package_list = []
            package_id_list = []
            if packages:
                for package in packages:
                    package_label = '[' + str(
                        package['strCountry']
                    ) + '] ' + package['strPackage'] + ' (' + str(
                        package['strType']) + ')'
                    package_id = package['idPackage']
                    package_list.append(package_label)
                    package_id_list.append(package_id)
                if package_list:
                    choose = xbmcgui.Dialog().select('TVLogo Downloader',
                                                     package_list)
                    if choose > -1:
                        channels = thelogodb.Channels().by_package(
                            package_id_list[choose])
                        if channels:
                            logowindow.start(channels, "True", "True")
                            specific_channels()
                        else:
                            mensagemok('TVLogo Downloader',
                                       'No logos available for this package!')
                            specific_channels()
                    else:
                        entire_packages()
            else:
                mensagemok('TVLogo Downloader', 'Error getting packages!')
                entire_packages()

        elif optionsvar[choose] == 'search':
            keyb = xbmc.Keyboard('', 'Enter channel')
            keyb.doModal()
            if (keyb.isConfirmed()):
                search_parameter = urllib.quote_plus(keyb.getText())
                if search_parameter:
                    channels = thelogodb.Channels().by_keyword(
                        search_parameter)
                    if channels:
                        logowindow.start(channels, "True", "True")
                    else:
                        mensagemok('TVLogo Downloader',
                                   'No packages available!')
                        specific_channels()
                else:
                    specific_channels()
    else:
        main_menu()
コード例 #3
0
def automatic_downloader(mode):
    dp = xbmcgui.DialogProgress()
    dp.create('TVLogo Downloader')
    dp.update(0, 'Getting user channels')
    failed_log = []
    #grab all user channel groups
    #TV
    groupids = []
    ch_ids = []
    ch_names = []

    json_response = xbmc.executeJSONRPC(
        '{"jsonrpc": "2.0", "method": "PVR.GetChannelGroups", "params": {"channeltype" : "tv"}, "id": 1 }'
    )
    decoded_data = json.loads(json_response)
    try:
        groups = decoded_data['result']['channelgroups']
    except:
        mensagemok(
            'TVLogo Downloader',
            'Live TV is not enabled in kodi or no channels are available.')
        sys.exit(0)
    for x in range(0, len(decoded_data['result']['channelgroups'])):
        has_channels = False
        if groups[x]["channelgroupid"] > -1:
            json_response = xbmc.executeJSONRPC(
                '{"jsonrpc": "2.0", "method": "PVR.GetChannels", "params": {"channelgroupid" : '
                + str(groups[x]["channelgroupid"]) +
                ',"properties":["channel","channeltype","thumbnail","broadcastnow","broadcastnext"]}, "id": 1 }'
            )
            channel_number = bool(
                "channels" in json.loads(json_response)["result"])
            if channel_number: has_channels = True
        if has_channels:
            groupids.append(groups[x]["channelgroupid"])

    #Radio
    json_response = xbmc.executeJSONRPC(
        '{"jsonrpc": "2.0", "method": "PVR.GetChannelGroups", "params": {"channeltype" : "radio"}, "id": 1 }'
    )
    decoded_data = json.loads(json_response)
    groups = decoded_data['result']['channelgroups']
    for x in range(0, len(decoded_data['result']['channelgroups'])):
        #check if group has channels
        has_channels = False
        if groups[x]["channelgroupid"] > -1:
            json_response = xbmc.executeJSONRPC(
                '{"jsonrpc": "2.0", "method": "PVR.GetChannels", "params": {"channelgroupid" : '
                + str(groups[x]["channelgroupid"]) +
                ',"properties":["channel","channeltype","thumbnail","broadcastnow","broadcastnext"]}, "id": 1 }'
            )
            channel_number = bool(
                "channels" in json.loads(json_response)["result"])
            if channel_number: has_channels = True
        if has_channels:
            groupids.append(groups[x]["channelgroupid"])

    #grab every channel name and label
    if groupids:
        for groupid in groupids:
            json_response = xbmc.executeJSONRPC(
                '{"jsonrpc": "2.0", "method": "PVR.GetChannels", "params": {"channelgroupid" : '
                + str(groupid) +
                ',"properties":["channel","channeltype","thumbnail"]}, "id": 1 }'
            )
            decoded_data = json.loads(json_response)
            if "channels" in decoded_data["result"].keys():
                for channel in decoded_data["result"]["channels"]:
                    if mode == 'all':
                        if channel["label"] not in ch_names:
                            ch_ids.append(channel["channelid"])
                            ch_names.append(channel["label"])
                    else:
                        json_response = xbmc.executeJSONRPC(
                            '{"jsonrpc":"2.0","method":"Settings.GetSettingValue","params":{"setting":"pvrmenu.iconpath"},"id":9}'
                        )
                        decoded_data = json.loads(json_response)
                        logo_folder = decoded_data['result']['value']
                        logofile = os.path.join(
                            logo_folder,
                            channel["label"].replace('/', '_') + '.png')
                        if channel["label"] not in ch_names and channel[
                                'thumbnail'].replace(
                                    'image://', ''
                                ) != (logo_folder + channel["label"].replace(
                                    ' ', '%20') + '.png').replace(
                                        '/', '%2f'
                                    ) + '/' and not os.path.exists(logofile):
                            ch_ids.append(channel["channelid"])
                            ch_names.append(channel["label"])
    if ch_names:
        download_list = []
        initconst = const.Constr().restart_process()
        totalchannels = len(ch_names)

        i = 1
        iscanceled = False
        for channel in ch_names:
            dp.update(int(i / float(totalchannels)),
                      'Processing channel ' + channel)
            if not dp.iscanceled():
                match = thelogodb.Channels().by_keyword(
                    urllib.quote_plus(
                        get_replaced_names(channel.encode('utf-8'))))
                if match:
                    match = return_only_valid(match)
                    if not match:
                        if ' hq' in channel.encode('utf-8').lower():
                            match = tvlogodownloader.get_nonhd_match(
                                get_replaced_names(
                                    channel.encode('utf-8').lower().replace(
                                        'hq', ' hd')))
                            match = return_only_valid(match)
                        if not match:
                            match = tvlogodownloader.get_nonhd_match(
                                get_replaced_names(channel.encode('utf-8')))
                            match = return_only_valid(match)
                            if not match:
                                #check if nonascii version exists
                                match = thelogodb.Channels().by_keyword(
                                    urllib.quote_plus(
                                        removeNonAscii(
                                            get_replaced_names(channel))))
                                match = return_only_valid(match)
                                #if no match check if channel is HD and grab logos for nonhd
                                if not match:
                                    match = tvlogodownloader.get_nonhd_match(
                                        removeNonAscii(
                                            get_replaced_names(channel)))
                                    match = return_only_valid(match)
                else:
                    if ' hq' in channel.encode('utf-8').lower():
                        match = tvlogodownloader.get_nonhd_match(
                            get_replaced_names(
                                channel.encode('utf-8').lower().replace(
                                    'hq', ' hd')))
                        match = return_only_valid(match)
                    if not match:
                        match = tvlogodownloader.get_nonhd_match(
                            get_replaced_names(channel.encode('utf-8')))
                        match = return_only_valid(match)
                        if not match:
                            #check if nonascii version exists
                            match = thelogodb.Channels().by_keyword(
                                urllib.quote_plus(
                                    removeNonAscii(
                                        get_replaced_names(channel))))
                            match = return_only_valid(match)
                            #if no match check if channel is HD and grab logos for nonhd
                            if not match:
                                match = tvlogodownloader.get_nonhd_match(
                                    removeNonAscii(
                                        get_replaced_names(channel)))
                                match = return_only_valid(match)

                if not match:
                    if channel not in failed_log:
                        failed_log.append(channel)
                if match:
                    if len(match) == 1:
                        obj = {
                            'channel_name': match[0]["strChannel"],
                            'channel_logo': match[0]["strLogoWide"],
                            'selected_channel': channel
                        }
                        const.Constr().add_to_array(obj)
                    else:
                        #manipulations
                        seqmatch = {}
                        for canal in match:
                            try:
                                print canal["strChannel"].lower(
                                ), channel.lower()
                            except:
                                pass
                            ratio = int(
                                difflib.SequenceMatcher(
                                    None, canal["strChannel"].lower(),
                                    channel.lower()).ratio() * 100)
                            seqmatch[ratio] = canal

                        already = False

                        if settings.getSetting('auto_if_100') == 'true':
                            if max(seqmatch.keys()) == 100:
                                already = True
                                obj = {
                                    'channel_name':
                                    seqmatch[max(
                                        seqmatch.keys())]["strChannel"],
                                    'channel_logo':
                                    seqmatch[max(
                                        seqmatch.keys())]["strLogoWide"],
                                    'selected_channel':
                                    channel
                                }
                                const.Constr().add_to_array(obj)
                        if settings.getSetting(
                                'auto_if_multiple') == 'true' and not already:
                            if max(seqmatch.keys()) >= int(
                                    settings.getSetting("minimum_ratio")):
                                already = True
                                obj = {
                                    'channel_name':
                                    seqmatch[max(
                                        seqmatch.keys())]["strChannel"],
                                    'channel_logo':
                                    seqmatch[max(
                                        seqmatch.keys())]["strLogoWide"],
                                    'selected_channel':
                                    channel
                                }
                                const.Constr().add_to_array(obj)

                        if not already:
                            xbmc.sleep(3000)
                            before = const.Constr().return_array()
                            logowindow.start(match,
                                             "False",
                                             "False",
                                             selected_channel=channel)
                            now = const.Constr().return_array()
                            if before == now:
                                if channel not in failed_log:
                                    failed_log.append(channel)
                            del before
                            del now
                else:
                    if channel not in failed_log:
                        failed_log.append(channel)
                i += 1
            else:
                iscanceled = True
                dp.close()
        if not iscanceled:
            dp.update(100)
            dp.close()
            logos_to_download = const.Constr().return_array()
            downloader.Downloader(logos_to_download, True, failed_log)
            postprocessing.run()
            tvlogodownloader.main_menu(select=False, choose='')
コード例 #4
0
def run(channel_name):
	channel_name = urllib.unquote(channel_name)
	menu_labels = ['Exact channel name','Custom channel name']
	menu_functions = ['auto','custom']
	initconst = const.Constr().restart_process()
	choose = xbmcgui.Dialog().select('TVLogo Downloader',menu_labels)
	if choose > -1:
		if menu_functions[choose] == 'auto':
			channel = urllib.quote_plus(get_replaced_names(channel_name))
		elif menu_functions[choose] == 'custom':
			keyb = xbmc.Keyboard('', 'Enter channel')
			keyb.doModal()
			if (keyb.isConfirmed()):
				search_parameter = urllib.quote_plus(get_replaced_names(keyb.getText()))
				if search_parameter:
					channel = search_parameter
					del search_parameter
		
		match = thelogodb.Channels().by_keyword(channel)
		if match:
			match = return_only_valid(match)
			if not match:
				#check if HQ exists and replace it by HD
				if '+hq' in channel.lower():
					match = thelogodb.Channels().by_keyword(channel.lower().replace('+hq','+hd'))
					match = return_only_valid(match)
				if not match:
					match = tvlogodownloader.get_nonhd_match(urllib.unquote_plus(channel))
					match = return_only_valid(match)
					if not match:
						#check if nonascii version exists
						match = thelogodb.Channels().by_keyword(urllib.quote_plus(removeNonAscii(urllib.unquote_plus(channel))))
						match = return_only_valid(match)
						#if no match check if channel is HD and grab logos for nonhd
						if not match:
							match = tvlogodownloader.get_nonhd_match(removeNonAscii(urllib.unquote_plus(channel)))
							match = return_only_valid(match)
		else:
			#check if HQ exists and replace it by HD
			if '+hq' in channel.lower():
				match = thelogodb.Channels().by_keyword(channel.lower().replace('+hq','+hd'))
				match = return_only_valid(match)
			if not match:
				match = tvlogodownloader.get_nonhd_match(urllib.unquote_plus(channel))
				match = return_only_valid(match)
				if not match:
					#check if nonascii version exists
					match = thelogodb.Channels().by_keyword(urllib.quote_plus(removeNonAscii(urllib.unquote_plus(channel))))
					match = return_only_valid(match)
					#if no match check if channel is HD and grab logos for nonhd
					if not match:
						match = tvlogodownloader.get_nonhd_match(removeNonAscii(urllib.unquote_plus(channel)))
						match = return_only_valid(match)

		if match:
			if len(match) == 1:
				obj = {'channel_name': match[0]["strChannel"],'channel_logo': match[0]["strLogoWide"],'selected_channel':channel_name}
				const.Constr().add_to_array(obj)
			else:
				#manipulations
				seqmatch = {}
				for canal in match:
					ratio = int(difflib.SequenceMatcher(None, canal["strChannel"].lower(),urllib.unquote_plus(channel).lower()).ratio()*100)
					seqmatch[ratio] = canal
						
				already = False
									
				if settings.getSetting('auto_if_100') == 'true':
					if max(seqmatch.keys()) == 100: 
						already = True
						obj = {'channel_name': seqmatch[max(seqmatch.keys())]["strChannel"],'channel_logo': seqmatch[max(seqmatch.keys())]["strLogoWide"],'selected_channel':channel_name}
						const.Constr().add_to_array(obj)
				if settings.getSetting('auto_if_multiple') == 'true' and not already:
					if max(seqmatch.keys()) >= int(settings.getSetting("minimum_ratio")):
						already = True
						obj = {'channel_name': seqmatch[max(seqmatch.keys())]["strChannel"],'channel_logo': seqmatch[max(seqmatch.keys())]["strLogoWide"],'selected_channel':channel_name}
						const.Constr().add_to_array(obj)
						
				if not already:
					logowindow.start(match,"False","False",selected_channel=channel_name)
		else: mensagemok('TVLogo Downloader','No channels match on the db. Try to use a custom search.')
		#download
		logos_to_download = const.Constr().return_array()
		if logos_to_download:
			downloader.Downloader(logos_to_download,True,'')
			postprocessing.run()
	return