mirror of
https://github.com/qbittorrent/qBittorrent.git
synced 2024-12-15 04:31:39 +03:00
4a1c8a7279
- Updated "buy it" icon
157 lines
5.2 KiB
Python
Executable file
157 lines
5.2 KiB
Python
Executable file
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# Redistribution and use in source and binary forms, with or without
|
|
# modification, are permitted provided that the following conditions are met:
|
|
#
|
|
# * Redistributions of source code must retain the above copyright notice,
|
|
# this list of conditions and the following disclaimer.
|
|
# * Redistributions in binary form must reproduce the above copyright
|
|
# notice, this list of conditions and the following disclaimer in the
|
|
# documentation and/or other materials provided with the distribution.
|
|
# * Neither the name of the author nor the names of its contributors may be
|
|
# used to endorse or promote products derived from this software without
|
|
# specific prior written permission.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
|
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
# POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
|
|
#VERSION: 1.23
|
|
|
|
# Author:
|
|
# Fabien Devaux <fab AT gnux DOT info>
|
|
# Contributors:
|
|
# Christophe Dumez <chris@qbittorrent.org> (qbittorrent integration)
|
|
# Thanks to gab #gcu @ irc.freenode.net (multipage support on PirateBay)
|
|
# Thanks to Elias <gekko04@users.sourceforge.net> (torrentreactor and isohunt search engines)
|
|
#
|
|
# Licence: BSD
|
|
|
|
import sys
|
|
import threading
|
|
import os
|
|
import glob
|
|
|
|
THREADED = True
|
|
CATEGORIES = ('all', 'movies', 'tv', 'music', 'games', 'anime', 'software', 'pictures', 'books')
|
|
|
|
################################################################################
|
|
# Every engine should have a "search" method taking
|
|
# a space-free string as parameter (ex. "family+guy")
|
|
# it should call prettyPrinter() with a dict as parameter.
|
|
# The keys in the dict must be: link,name,size,seeds,leech,engine_url
|
|
# As a convention, try to list results by decrasing number of seeds or similar
|
|
################################################################################
|
|
|
|
supported_engines = []
|
|
|
|
engines = glob.glob(os.path.join(os.path.dirname(__file__), 'engines','*.py'))
|
|
for engine in engines:
|
|
e = engine.split(os.sep)[-1][:-3]
|
|
if len(e.strip()) == 0: continue
|
|
if e.startswith('_'): continue
|
|
try:
|
|
exec "from engines.%s import %s"%(e,e)
|
|
supported_engines.append(e)
|
|
except:
|
|
pass
|
|
|
|
def engineToXml(short_name):
|
|
xml = "<%s>\n"%short_name
|
|
exec "engine = %s()"%short_name
|
|
xml += "<name>%s</name>\n"%engine.name
|
|
xml += "<url>%s</url>\n"%engine.url
|
|
xml += "<categories>"
|
|
if hasattr(engine, 'supported_categories'):
|
|
supported_categories = engine.supported_categories.keys()
|
|
supported_categories.remove('all')
|
|
xml += " ".join(supported_categories)
|
|
xml += "</categories>\n"
|
|
xml += "</%s>\n"%short_name
|
|
return xml
|
|
|
|
def displayCapabilities():
|
|
"""
|
|
Display capabilities in XML format
|
|
<capabilities>
|
|
<engine_short_name>
|
|
<name>long name</name>
|
|
<url>http://example.com</url>
|
|
<categories>movies music games</categories>
|
|
</engine_short_name>
|
|
</capabilities>
|
|
"""
|
|
xml = "<capabilities>"
|
|
for short_name in supported_engines:
|
|
xml += engineToXml(short_name)
|
|
xml += "</capabilities>"
|
|
print xml
|
|
|
|
class EngineLauncher(threading.Thread):
|
|
def __init__(self, engine, what, cat='all'):
|
|
threading.Thread.__init__(self)
|
|
self.engine = engine
|
|
self.what = what
|
|
self.cat = cat
|
|
def run(self):
|
|
if hasattr(self.engine, 'supported_categories'):
|
|
if self.cat == 'all' or self.cat in self.engine.supported_categories.keys():
|
|
self.engine.search(self.what, self.cat)
|
|
elif self.cat == 'all':
|
|
self.engine.search(self.what)
|
|
|
|
if __name__ == '__main__':
|
|
if len(sys.argv) < 2:
|
|
raise SystemExit('./nova2.py [all|engine1[,engine2]*] <category> <keywords>\navailable engines: %s'%
|
|
(','.join(supported_engines)))
|
|
|
|
if len(sys.argv) == 2:
|
|
if sys.argv[1] == "--capabilities":
|
|
displayCapabilities()
|
|
sys.exit(0)
|
|
else:
|
|
raise SystemExit('./nova.py [all|engine1[,engine2]*] <category> <keywords>\navailable engines: %s'%
|
|
(','.join(supported_engines)))
|
|
|
|
engines_list = [e.lower() for e in sys.argv[1].strip().split(',')]
|
|
|
|
if 'all' in engines_list:
|
|
engines_list = supported_engines
|
|
|
|
cat = sys.argv[2].lower()
|
|
|
|
if cat not in CATEGORIES:
|
|
raise SystemExit('Invalid category!')
|
|
|
|
what = '+'.join(sys.argv[3:])
|
|
|
|
threads = []
|
|
for engine in engines_list:
|
|
try:
|
|
if THREADED:
|
|
exec "l = EngineLauncher(%s(), what, cat)"%engine
|
|
threads.append(l)
|
|
l.start()
|
|
else:
|
|
exec "e = %s()"%engine
|
|
if hasattr(engine, 'supported_categories'):
|
|
if cat == 'all' or cat in e.supported_categories.keys():
|
|
e.search(what, cat)
|
|
elif self.cat == 'all':
|
|
e.search(what)
|
|
engine().search(what, cat)
|
|
except:
|
|
pass
|
|
if THREADED:
|
|
for t in threads:
|
|
t.join()
|