def subtitles(request,path):
try:
proc = Process.objects.filter(name=path.split('/')[0])[0]
except:
return '/archive/notFound.vtt'
if proc.subp ==1:
return proc.mlink
try:
opsub = OpenSubtitles()
opsub.login('hgeg','sokoban')
except:
return '/archive/error.vtt'
subsFile = "/archive/internal.vtt"
try:
data = guessFileData(path)
if(data['type']=='tvshow'):
sub = opsub.search_subtitles([{'sublanguageid':'eng','query':data['name'],'season':data['season'],'episode':data['episode']}])[0]
else:
sub = opsub.search_subtitles([{'sublanguageid':'eng','query':data['name']}])[0]
subsFile = createFile(path,sub['ZipDownloadLink'])
proc.subp = 1
proc.mlink = subsFile
proc.save()
return proc.mlink
except: pass
if request != None:
return HttpResponse(str(subsFile))
else: return subsFile
class OpenSubtitlesModel(object):
def __init__(self):
self.ost = OpenSubtitles('en')
def _refresh_auth(self):
return self.ost.login('doctest', 'doctest')
def search_by_name_get_first_n(self, name, n):
self._refresh_auth()
search_result = self.ost.search_subtitles([{
'sublanguageid': 'eng',
'query': name
}])
if type(search_result) == list and len(search_result) > 0:
parsed_result = search_result[0:n]
return parsed_result
else:
return
def search_by_name_get_most_fit(self, name):
return self.search_by_name_get_first_n(name, 1)[0]
def download_by_file_id(self, id_subtitle_file):
self._refresh_auth()
path_to_files_by_id_dict = self.ost.download_subtitles(
[id_subtitle_file]) # Files are placed at project root
path_to_subtitle_file = list(path_to_files_by_id_dict.values())[
0] # Get value from single entry dict
return path_to_subtitle_file
class OST:
def __init__(self):
self.api = OpenSubtitles()
def find_max_dl_count(self, **kwargs):
target = 0
for i, e in enumerate(kwargs['data']):
if int(e['IDMovieImdb']) == int(kwargs['imdb_id'].strip('t')) and \
e['SubFormat'] == 'srt':
if int(kwargs['data'][target]['SubDownloadsCnt']) < \
int(e['SubDownloadsCnt']):
target = i
return target
def link(self, **kwargs):
token = self.api.login("doctest", 'doctest')
data = self.api.search_subtitles([{
'query': kwargs['query'],
'sublanguageid': kwargs['sublanid']
}])
target = self.find_max_dl_count(data, kwargs['imdb_id'])
if len(data) == 0:
return None
return data[target]['ZipDownloadLink']
def main():
parser = argparse.ArgumentParser(description="List available subtitles for a show")
parser.add_argument("show", help="Show you want to list the subtitles for")
parser.add_argument("-s", "--season", help="Season you want to list", type=int, default=1)
parser.add_argument("-l", "--language", help="Language you want to list subtitles in", default='en')
args = parser.parse_args()
os = OpenSubtitles()
os.login(username=None, password=None)
show_name = args.show
sub_language = args.language
season = str(args.season)
results = os.search_subtitles([{'sublanguageid': sub_language, 'query': show_name, 'season': season}])
sorted_results = sorted(results, key=lambda x: int(x['SeriesEpisode']))
table = []
for result in sorted_results:
name = result['MovieName'].strip()
match_exact_show_name_regex = r'\"{}\"'.format(show_name)
if re.search(match_exact_show_name_regex, name):
username_raw = result['UserNickName']
username = username_raw if username_raw else 'nobody'
size = result['SubSize']
date_raw = result['SubAddDate']
date = datetime.strptime(date_raw, '%Y-%m-%d %H:%M:%S')
filename = result['SubFileName']
table.append(['-r--r--r--', '1', username, username, size,
date.strftime('%b %d %H:%M'), filename])
print "total {}".format(len(table))
print tabulate(table, tablefmt="plain")
def main():
parser = argparse.ArgumentParser()
parser.add_argument(
"file",
help="Path to the file you want to download subtitles for",
type=str)
args = parser.parse_args()
if os.path.exists(args.file):
f = File(os.path.abspath(args.file))
path, file = os.path.split(args.file)
path_full_nofile = os.path.abspath(path)
path_file = file
path_file_noext = os.path.splitext(args.file)[0]
else:
sys.exit("File does not exist")
ost = OpenSubtitles()
username = input("Please input your OpenSubtitles.org username: "******"Please input your OpenSubtitles.org password: "******"Logging in, please wait...")
ost.login(username, password)
print("Successfully logged in!")
hash = f.get_hash()
size = f.size
print("Scouring the web for subtitles, please wait...")
data = ost.search_subtitles([{
'sublanguageid': 'eng',
'moviehash': hash,
'moviebytesize': size
}])
bestdic = None
highestdownloadcount = 0
for dict in data:
downloadcount = int(dict.get('SubDownloadsCnt'))
if downloadcount > highestdownloadcount:
downloadcount = highestdownloadcount
bestdic = dict
print("Subtitle found in", bestdic.get('SubLanguageID'), "with",
bestdic.get('SubDownloadsCnt'), "downloads.")
print("Downloading subtitles, please wait...")
urllib.request.urlretrieve(
bestdic.get('SubDownloadLink'),
path_full_nofile + "/" + path_file_noext + ".srt.gz")
print("Subtitle downloaded!")
print("Unzipping subtitle, please wait...")
inF = gzip.open(path_full_nofile + "/" + path_file_noext + ".srt.gz", "rb")
outF = open(path_full_nofile + "/" + path_file_noext + ".srt", "wb")
outF.write(inF.read())
inF.close()
outF.close()
os.remove(path_full_nofile + "/" + path_file_noext + ".srt.gz")
print("Done!")
def __init__(self, username, password, language=None, **kwargs):
"""
:param username: Your OpenSubtitles username
:type username: str
:param password: Your OpenSubtitles password
:type password: str
:param language: Preferred language name, ISO639 code or OpenSubtitles
language ID to be used for the subtitles. Also supports an (ordered)
list of preferred languages
:type language: str or list[str]
"""
from pythonopensubtitles.opensubtitles import OpenSubtitles
super().__init__(**kwargs)
self._ost = OpenSubtitles()
self._token = self._ost.login(username, password)
self.languages = []
self._file_lock = threading.RLock()
if language:
if isinstance(language, str):
self.languages.append(language.lower())
elif isinstance(language, list):
self.languages.extend([l.lower() for l in language])
else:
raise AttributeError(
'{} is neither a string nor a list'.format(language))
def download_subs_and_cache_results():
import argparse
parser = argparse.ArgumentParser(
description='TV Series Tools: Download subtitles for movies specified'
' by passed IMDB IDs'
)
parser.add_argument('--input', '-i', dest='inputfile', required=True,
help='movies CSV file: 1st column = IMDB ID, 2nd column = title')
parser.add_argument('--output', '-o', dest='outputdir', required=True,
help='path to a directory in which the subtitle SRT'
' files will be downloaded')
parser.add_argument('--cache', '-c', dest='cachedir', required=True,
help='cache location')
parser.add_argument('--shuffle', '-s', dest='shuffle', action='store_true',
help='shuffle the list of IMDB IDs')
parser.add_argument('--timeout', '-t', dest='timeout', type=int,
default=DEFAULT_TIMEOUT,
help='time in seconds between OpenSubtitles\'s API'
' calls, if not set {} is used'
.format(DEFAULT_TIMEOUT))
args = parser.parse_args()
opensub = OpenSubtitles()
opensub.login(os.environ['OPENSUB_USER'], os.environ['OPENSUB_PASSWD'])
movies = listio.read_map(args.inputfile)
if args.shuffle:
movies = list(movies)
random.shuffle(movies)
excl_ids = []
for f in (FILE_NAME_DOWNLOADED, FILE_NAME_FAILED):
try:
lines = listio.read_map(os.path.join(args.cachedir, f))
for line in lines:
excl_ids.append(line[0])
except FileNotFoundError:
pass
if not os.path.isdir(args.cachedir):
os.makedirs(args.cachedir)
if not os.path.isdir(args.outputdir):
os.makedirs(args.outputdir)
write_result(
os.path.join(args.cachedir, FILE_NAME_DOWNLOADED),
os.path.join(args.cachedir, FILE_NAME_FAILED),
download_subs(
movies,
excl_ids,
args.outputdir,
opensub,
timeout=args.timeout
)
)
sys.exit()
def main():
pathList = []
downloadList = []
#if len(sys.argv) < 1 :
# print "Missing Arguments"
# quit()
folder = "/home/Downloads/torrents"
#print sys.argv[1]
for root, dirs, files in os.walk(folder):
path = root.split('/')
#print root
for file in files:
if(file.endswith(".mp4") or file.endswith(".avi") or file.endswith(".mkv") ):
print os.path.join(root,file)
pathList.append(os.path.join(root,file))
print "Found %d files" % len(pathList)
if len(pathList) >= 1:
ops = OpenSubtitles()
token = ops.login("","")
print token
for subToFind in pathList:
f = File(subToFind)
dirname = os.path.normpath(subToFind)
subData = ops.search_subtitles([{'sublanguageid': 'pob','moviehash': f.get_hash() , 'moviebytesize': f.size }])
if not subData:
print "Sub not found for %s " % f.path
else:
filename = ''
if f.path.endswith(".mp4"):
filename = dirname.replace('.mp4', '.srt')
elif f.path.endswith('.mkv'):
filename = dirname.replace('.mkv', '.srt')
elif f.path.endswith('.avi'):
filename = dirname.replace('.avi', '.srt')
response = urllib2.urlopen(subData[0]['SubDownloadLink'])
compressedFile = StringIO.StringIO()
compressedFile.write(response.read())
response.close()
compressedFile.seek(0)
decompressedFile = gzip.GzipFile(fileobj=compressedFile,mode='rb')
with open(filename ,'w') as outfile:
outfile.write(decompressedFile.read())
time.sleep(1)
ops.logout()
def descargar_subs_de_lista(listfile, outputpath, min_line=0, max_line=10):
ost = OpenSubtitles()
token = ost.login('doctest', 'doctest')
with codecs.open(listfile, "r") as lf:
for i, line in enumerate(lf):
if i < min_line: continue
if i >= max_line: break
movieid = line.split(" ")[0]
print("Descargando #{0} - {1}".format(i, movieid))
descargar_sub(movieid, outputpath, ost=ost)
def printmovielist(query, outfile, num=300, check_if_subtitles=True):
results = 0
start_index = 1
imdb_api = imdb.IMDb()
if check_if_subtitles:
ost = OpenSubtitles()
token = ost.login('doctest', 'doctest')
print(token)
with codecs.open(outfile, "w", encoding="utf-8") as out:
while results < num:
page = requests.get(query + "&start={0}".format(start_index))
soup = BeautifulSoup(page.content, "html.parser")
titles = soup.find_all('span', class_="lister-item-header")
idregex = re.compile("tt([0-9]*)")
print("Revisando resultados del {0} al {1}:", start_index,
start_index + 49)
for title in titles:
movielink = title.find('a')
movieid = idregex.search(movielink['href']).group(1)
movietitle = title.text.replace("\n", " ").strip()
# Se ignoran los resultados que no sean peliculas:
movie = imdb_api.get_movie(movieid)
if movie is None or movie['kind'] != 'movie':
continue
usable = True
# Se ignoran los resultados para los que no haya subtitulos en los lenguajes de nuestro interes:
if check_if_subtitles:
languages = ['eng', 'spa', 'fre']
for lang in languages:
found = ost.search_subtitles([{
'sublanguageid': lang,
'imdbid': movieid
}])
if not found or len(found) < 1:
usable = False
if usable:
print(movieid, movietitle)
out.write("{0} {1}\n".format(movieid, movietitle))
results += 1
if results == num:
return
start_index += 50
def __init__(self, skip_subsync, ost_username, ost_password, tmdb_key):
self.skip_subsync = skip_subsync
self.ost = OpenSubtitles()
try:
self.ost.login(ost_username, ost_password)
except Exception as e:
logging.error("Failed to log into opensubtitles.org.")
raise e
self.ost_language = 'eng'
if tmdb_key:
self.media_searcher = MediaSearcher(tmdb_key)
if not self.skip_subsync:
self.subsync_parser = ffsubsync.make_parser()
def crawler(title):
ost = OpenSubtitles()
_ = ost.login("doctest", 'doctest')
data = ost.search_subtitles([{'query': title, 'sublanguageid': 'zht'}])
# pp.pprint(data)
highest_download_count = -1
link = ''
for d in data:
if int(d['SubDownloadsCnt']) > highest_download_count:
highest_download_count = int(d['SubDownloadsCnt'])
link = d['ZipDownloadLink']
webbrowser.open(link, new=2, autoraise=True)
print(title, link, sep='\t', flush=True)
def download_subtitle(self):
"""
Downloads subtitles from opensubtitles.org, in the defined language
and stores them in a tempfile.
Search is trying to match movie by hash and if it is unsuccessful,
it searches by movie name.
Only first match is considered.
"""
ost = OpenSubtitles()
logged_in = ost.login(self.os_username, self.os_password)
if not logged_in:
raise Error("Wrong opensubtitles credentials")
# TODO refactor to consume path on event, potentially from queue
mkv_files = [mkv for mkv in self.watch_path.glob("*.mkv")]
subs = list()
for movie in mkv_files:
movie_file = File(movie.absolute())
# search by hash, if not, by name
ost_data = ost.search_subtitles([
{
"sublanguageid":
self.os_language
if len(self.os_language) == 3 else self.os_language,
"moviehash":
movie_file.get_hash(),
},
{
"sublanguageid":
self.os_language
if len(self.os_language) == 3 else self.os_language,
"query":
movie.name,
},
])
if ost_data:
# # downloading first subtitle
plain_link = ost_data[0]["SubDownloadLink"]
sub_link_parts = plain_link.split("/download/")
# rebuilding link to get utf-8 subtitle
sub_link = (sub_link_parts[0] + "/download/subencoding-utf8/" +
sub_link_parts[1])
response = requests.get(sub_link)
tmp, tmp_name = tempfile.mkstemp()
with open(tmp, "wb") as srt_out:
srt_out.write(gzip.decompress(response.content))
subs.append({"file_path": movie, "sub": tmp_name})
else:
subs.append({"file_path": movie, "sub": None})
return subs
def __init__(self, username, password):
"""Instantiate the shit out of this
Arguments:
- username (str): Your OpenSubtitles username
- password (str): Your OpenSubtitles password
Returns:
- Nothing: Sets all parameters of the class.
"""
self.username = username
self.password = password
self.opensubs = OpenSubtitles()
self.token = self.opensubs.login(self.username, self.password)
assert self.token != None, "Incorrect username/password (or something else went wrong when loging in...)"
class OpenSubtitleSession:
def __init__(self, username, password):
self.username = username
self.password = password
self._ost = OpenSubtitles()
def __enter__(self):
token = self._ost.login(self.username, self.password)
if not token:
raise ValueError('Failed to login')
return self._ost
def __exit__(self, exc_type, exc_val, exc_tb):
self._ost.logout()
def __init__(self, search_term = None, data_path = ".", verbose = 2):
"""
Initialize the SubDownloader object
"""
self.ost = OpenSubtitles()
self.ia = IMDb()
self.password_array = []
self.used_accounts = []
self.data_path = data_path
self.verbose = verbose
self.search_term = search_term
self.current_account = None
def _download_subtitle_chunk(retries=5):
nonlocal ost
try:
ost.download_subtitles(
[_id for _id in sub_ids.keys()],
override_filenames=sub_ids,
output_directory=dirname,
extension="srt",
)
except ProtocolError as e:
if retries == 0:
raise e
time.sleep(10)
ost = OpenSubtitles()
ost.login(None, None)
_download_subtitle_chunk(retries=retries - 1)
def get_subtitle(filename):
os_sub = OpenSubtitles()
try:
token = os_sub.login('thegyro','idontlikepasswords')
if os.path.exists(os.path.join(source_dir,filename)):
video_file = File(os.path.join(source_dir,filename))
video_hash = video_file.get_hash()
video_size = video_file.size
sub_param = {'sublanguageid':'English','moviehash':video_hash,'moviebytesize': video_size}
subtitles = os_sub.search_subtitles([sub_param])
print subtitles
else:
raise Exception("File doesn't exist")
except Exception as ex:
print str(ex)
def opensub_initiate():
"""OpenSubtitles initiation for use
Login to OpenSubtitles using API to retrieve a token
for identifying movies.
Failure results in a null token.
Network errors are caused to service unavailability, or DNS problems
Args:
None
Returns:
token(str): Authentication Token for OpenSubtitles
None: if token cannot be validated
Raises:
None
"""
try:
# login to opensub using API
sub = OpenSubtitles()
token = sub.login(OpenSubKey.get_solo().uid, OpenSubKey.get_solo().key)
if not token:
# return sub
print 'null token'
log.error('open sub null token')
return
return sub
except ProtocolError as e:
# most likely network error or API server error
print "E: " + str(e)
except AssertionError:
print 'Failed to authenticate opensubtitles login.'
except ProtocolError as e:
"""
TODO: gaierror:
[Errno 8] nodename nor servname provided, or not known
"""
if e.errcode == 503:
# ProtocolError - 503 Service Unavailable
print 'Check network connection and try again.'
log.error('open sub error occured')
log.error(traceback.format_exc())
except Exception:
log.error('open sub error occured')
log.error(traceback.format_exc())
def downloadSubs(Parameters):
#Assign/allocate object and get token after logging in with credentials from the Parameters object
opensubs = OpenSubtitles()
token = opensubs.login(Parameters.username, Parameters.password)
if token is None:
print '\n*** Login failed! ***\n'
sys.exit()
#Get hash and size of file from Parameters object
f = File(os.path.join(Parameters.path, Parameters.video))
print '\tPath: %s' % Parameters.path
print '\tFile: %s' % Parameters.video
hash = f.get_hash()
size = f.get_size()
#Search subtitles DB using file hash and size. Looks like the first result is the best matching result
data = opensubs.search_subtitles([{'sublanguageid': 'eng', 'moviehash': hash, 'moviebytesize': size}])
if data:
#Download first result, decode it from BASE64, add gz extension, save file
download = opensubs.download_subtitles([data[0]['IDSubtitleFile']])
data_decoded = base64.b64decode(unicode(download[0]['data']))
gz_file = os.path.join(Parameters.path, Parameters.subtitle) + '.gz'
print '\nCreating gz file: %s' % gz_file
download_file = open(gz_file,'w')
download_file.write(data_decoded)
download_file.close()
print 'Created gz file: %s' % gz_file
#Extract SRT file from gz file and place it in the same folder
print 'Opening gz file: %s' % gz_file
srt_file_buffer = gzip.open(gz_file, 'r')
srt_file_name = os.path.join(Parameters.path, Parameters.subtitle)
print 'Creating SRT file: %s' % srt_file_name
srt_file = open(srt_file_name,'w')
srt_file.write(srt_file_buffer.read())
srt_file.close()
print 'Created SRT file: %s' % srt_file_name
#Delete .gz file
print 'Deleting %s' % gz_file
os.remove(gz_file)
else:
print '*** No match found for file! ***'
class subtitleFinder:
def __init__(self, username="", password="", token=""):
self.subs = OpenSubtitles()
if token:
self.subs.token = token # Set token to that provided -- untested
else:
self.subs.login(username, password)
# result = [{'MatchedBy': 'imdbid', 'IDSubMovieFile': '0', 'MovieHash': '0', 'MovieByteSize': '0', 'MovieTimeMS': '0', 'IDSubtitleFile': '1952931532', 'SubFileName': 'Person.of.Interest.S01E01.HDTV.en.srt', 'SubActualCD': '1', 'SubSize': '51140', 'SubHash': '1e591e0dcfa8197e70d05ab3a68c958d', 'SubLastTS': '00:43:38', 'SubTSGroup': '2', 'IDSubtitle': '4242217', 'UserID': '419677', 'SubLanguageID': 'eng', 'SubFormat': 'srt', 'SubSumCD': '1', 'SubAuthorComment': '', 'SubAddDate': '2011-09-23 15:40:55', 'SubBad': '0', 'SubRating': '0.0', 'SubDownloadsCnt': '166380', 'MovieReleaseName': 'S01E01.HDTV.XviD-ASAP / 720p-IMMERSE', 'MovieFPS': '23.976', 'IDMovie': '83999', 'IDMovieImdb': '1941917', 'MovieName': '"Person of Interest" Pilot', 'MovieNameEng': '', 'MovieYear': '2011', 'MovieImdbRating': '8.5', 'SubFeatured': '0', 'UserNickName': 'gutoresquin', 'SubTranslator': '', 'ISO639': 'en', 'LanguageName': 'English', 'SubComments': '0', 'SubHearingImpaired': '0', 'UserRank': 'administrator', 'SeriesSeason': '1', 'SeriesEpisode': '1', 'MovieKind': 'episode', 'SubHD': '1', 'SeriesIMDBParent': '1839578', 'SubEncoding': 'UTF-8', 'SubAutoTranslation': '0', 'SubForeignPartsOnly': '0', 'SubFromTrusted': '1', 'QueryCached': 1, 'SubTSGroupHash': '17b1cfb76c2c9188e3772b16eb3b18be', 'SubDownloadLink': 'http://dl.opensubtitles.org/en/download/src-api/vrf-19b40c51/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/filead/1952931532.gz', 'ZipDownloadLink': 'http://dl.opensubtitles.org/en/download/src-api/vrf-f51c0baf/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/subad/4242217', 'SubtitlesLink': 'http://www.opensubtitles.org/en/subtitles/4242217/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/person-of-interest-pilot-en', 'QueryNumber': '0', 'QueryParameters': {'imdbid': '1941917', 'sublanguageid': 'eng'}, 'Score': 34.6638}, {'MatchedBy': 'imdbid', 'IDSubMovieFile': '0', 'MovieHash': '0', 'MovieByteSize': '0', 'MovieTimeMS': '0', 'IDSubtitleFile': '1953205010', 'SubFileName': 'person.of.interest.s01e01.dvdrip.xvid-reward.srt', 'SubActualCD': '1', 'SubSize': '43805', 'SubHash': '12ba13bf716e293c52b3fa174b3c8f1b', 'SubLastTS': '00:42:27', 'SubTSGroup': '8', 'IDSubtitle': '4657669', 'UserID': '0', 'SubLanguageID': 'eng', 'SubFormat': 'srt', 'SubSumCD': '1', 'SubAuthorComment': '', 'SubAddDate': '2012-09-09 11:22:04', 'SubBad': '0', 'SubRating': '0.0', 'SubDownloadsCnt': '10630', 'MovieReleaseName': ' person.of.interest.s01e01.dvdrip.xvid-reward', 'MovieFPS': '23.976', 'IDMovie': '83999', 'IDMovieImdb': '1941917', 'MovieName': '"Person of Interest" Pilot', 'MovieNameEng': '', 'MovieYear': '2011', 'MovieImdbRating': '8.5', 'SubFeatured': '0', 'UserNickName': '', 'SubTranslator': '', 'ISO639': 'en', 'LanguageName': 'English', 'SubComments': '0', 'SubHearingImpaired': '0', 'UserRank': '', 'SeriesSeason': '1', 'SeriesEpisode': '1', 'MovieKind': 'episode', 'SubHD': '0', 'SeriesIMDBParent': '1839578', 'SubEncoding': 'ASCII', 'SubAutoTranslation': '0', 'SubForeignPartsOnly': '0', 'SubFromTrusted': '0', 'QueryCached': 1, 'SubTSGroupHash': '80ff21163490f8d5e7d2817a6db92ceb', 'SubDownloadLink': 'http://dl.opensubtitles.org/en/download/src-api/vrf-197d0c43/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/filead/1953205010.gz', 'ZipDownloadLink': 'http://dl.opensubtitles.org/en/download/src-api/vrf-f5650bc4/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/subad/4657669', 'SubtitlesLink': 'http://www.opensubtitles.org/en/subtitles/4657669/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/person-of-interest-pilot-en', 'QueryNumber': '0', 'QueryParameters': {'imdbid': '1941917', 'sublanguageid': 'eng'}, 'Score': 10.1063}, {'MatchedBy': 'imdbid', 'IDSubMovieFile': '0', 'MovieHash': '0', 'MovieByteSize': '0', 'MovieTimeMS': '0', 'IDSubtitleFile': '1953205011', 'SubFileName': 'person.of.interest.s01e01.extended.pilot.dvdrip.xvid-reward.srt', 'SubActualCD': '1', 'SubSize': '56390', 'SubHash': '478a88c9b3e48f1b1683ec70a86d060f', 'SubLastTS': '00:54:05', 'SubTSGroup': '1', 'IDSubtitle': '4657670', 'UserID': '0', 'SubLanguageID': 'eng', 'SubFormat': 'srt', 'SubSumCD': '1', 'SubAuthorComment': '', 'SubAddDate': '2012-09-09 11:22:04', 'SubBad': '0', 'SubRating': '0.0', 'SubDownloadsCnt': '23690', 'MovieReleaseName': ' person.of.interest.s01e01.extended.pilot.dvdrip.xvid-reward', 'MovieFPS': '23.976', 'IDMovie': '83999', 'IDMovieImdb': '1941917', 'MovieName': '"Person of Interest" Pilot', 'MovieNameEng': '', 'MovieYear': '2011', 'MovieImdbRating': '8.5', 'SubFeatured': '0', 'UserNickName': '', 'SubTranslator': '', 'ISO639': 'en', 'LanguageName': 'English', 'SubComments': '0', 'SubHearingImpaired': '0', 'UserRank': '', 'SeriesSeason': '1', 'SeriesEpisode': '1', 'MovieKind': 'episode', 'SubHD': '0', 'SeriesIMDBParent': '1839578', 'SubEncoding': 'ASCII', 'SubAutoTranslation': '0', 'SubForeignPartsOnly': '0', 'SubFromTrusted': '0', 'QueryCached': 1, 'SubTSGroupHash': 'd322313b0ee4ef22a36e7020b5e7ec95', 'SubDownloadLink': 'http://dl.opensubtitles.org/en/download/src-api/vrf-197e0c44/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/filead/1953205011.gz', 'ZipDownloadLink': 'http://dl.opensubtitles.org/en/download/src-api/vrf-f55e0bbc/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/subad/4657670', 'SubtitlesLink': 'http://www.opensubtitles.org/en/subtitles/4657670/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/person-of-interest-pilot-en', 'QueryNumber': '0', 'QueryParameters': {'imdbid': '1941917', 'sublanguageid': 'eng'}, 'Score': 10.2369}, {'MatchedBy': 'imdbid', 'IDSubMovieFile': '0', 'MovieHash': '0', 'MovieByteSize': '0', 'MovieTimeMS': '0', 'IDSubtitleFile': '1953652407', 'SubFileName': 'Person of Interest Ep 1 Pilot.srt', 'SubActualCD': '1', 'SubSize': '54011', 'SubHash': '34831fe210538fd18e8d0b1dd7b81ade', 'SubLastTS': '00:43:38', 'SubTSGroup': '2', 'IDSubtitle': '5110515', 'UserID': '0', 'SubLanguageID': 'eng', 'SubFormat': 'srt', 'SubSumCD': '1', 'SubAuthorComment': '', 'SubAddDate': '2013-07-29 13:23:14', 'SubBad': '0', 'SubRating': '0.0', 'SubDownloadsCnt': '9827', 'MovieReleaseName': ' Person of Interest Ep 1 Pilot', 'MovieFPS': '0.000', 'IDMovie': '83999', 'IDMovieImdb': '1941917', 'MovieName': '"Person of Interest" Pilot', 'MovieNameEng': '', 'MovieYear': '2011', 'MovieImdbRating': '8.5', 'SubFeatured': '0', 'UserNickName': '', 'SubTranslator': '', 'ISO639': 'en', 'LanguageName': 'English', 'SubComments': '0', 'SubHearingImpaired': '0', 'UserRank': '', 'SeriesSeason': '1', 'SeriesEpisode': '1', 'MovieKind': 'episode', 'SubHD': '0', 'SeriesIMDBParent': '1839578', 'SubEncoding': 'UTF-8', 'SubAutoTranslation': '0', 'SubForeignPartsOnly': '0', 'SubFromTrusted': '0', 'QueryCached': 1, 'SubTSGroupHash': '17b1cfb76c2c9188e3772b16eb3b18be', 'SubDownloadLink': 'http://dl.opensubtitles.org/en/download/src-api/vrf-19b30c53/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/filead/1953652407.gz', 'ZipDownloadLink': 'http://dl.opensubtitles.org/en/download/src-api/vrf-f50d0bab/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/subad/5110515', 'SubtitlesLink': 'http://www.opensubtitles.org/en/subtitles/5110515/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/person-of-interest-pilot-en', 'QueryNumber': '0', 'QueryParameters': {'imdbid': '1941917', 'sublanguageid': 'eng'}, 'Score': 10.09827}, {'MatchedBy': 'imdbid', 'IDSubMovieFile': '0', 'MovieHash': '0', 'MovieByteSize': '0', 'MovieTimeMS': '0', 'IDSubtitleFile': '1953655535', 'SubFileName': 'Person of Interest 1x01 - Pilot.srt', 'SubActualCD': '1', 'SubSize': '59830', 'SubHash': '29281fa9d767fe24abd56bf5289995a9', 'SubLastTS': '00:55:44', 'SubTSGroup': '1', 'IDSubtitle': '5113606', 'UserID': '1683613', 'SubLanguageID': 'eng', 'SubFormat': 'srt', 'SubSumCD': '1', 'SubAuthorComment': '', 'SubAddDate': '2013-07-31 13:05:24', 'SubBad': '0', 'SubRating': '0.0', 'SubDownloadsCnt': '21434', 'MovieReleaseName': 'Person of Interest 1x01 - Pilot', 'MovieFPS': '25.000', 'IDMovie': '83999', 'IDMovieImdb': '1941917', 'MovieName': '"Person of Interest" Pilot', 'MovieNameEng': '', 'MovieYear': '2011', 'MovieImdbRating': '8.5', 'SubFeatured': '0', 'UserNickName': 'james66', 'SubTranslator': '', 'ISO639': 'en', 'LanguageName': 'English', 'SubComments': '0', 'SubHearingImpaired': '1', 'UserRank': 'sub leecher', 'SeriesSeason': '1', 'SeriesEpisode': '1', 'MovieKind': 'episode', 'SubHD': '0', 'SeriesIMDBParent': '1839578', 'SubEncoding': 'ASCII', 'SubAutoTranslation': '0', 'SubForeignPartsOnly': '0', 'SubFromTrusted': '0', 'QueryCached': 1, 'SubTSGroupHash': 'd322313b0ee4ef22a36e7020b5e7ec95', 'SubDownloadLink': 'http://dl.opensubtitles.org/en/download/src-api/vrf-19c60c58/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/filead/1953655535.gz', 'ZipDownloadLink': 'http://dl.opensubtitles.org/en/download/src-api/vrf-f51b0baf/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/subad/5113606', 'SubtitlesLink': 'http://www.opensubtitles.org/en/subtitles/5113606/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/person-of-interest-pilot-en', 'QueryNumber': '0', 'QueryParameters': {'imdbid': '1941917', 'sublanguageid': 'eng'}, 'Score': 11.21434}, {'MatchedBy': 'imdbid', 'IDSubMovieFile': '0', 'MovieHash': '0', 'MovieByteSize': '0', 'MovieTimeMS': '0', 'IDSubtitleFile': '1954390918', 'SubFileName': 'Person of Interest - 01x01 - Pilot.Web-DL.English.C.updated.Addic7ed.com.srt', 'SubActualCD': '1', 'SubSize': '47522', 'SubHash': '8ad5288e042ffdbdfdf7b7daf514b7b7', 'SubLastTS': '00:43:00', 'SubTSGroup': '8', 'IDSubtitle': '5812638', 'UserID': '0', 'SubLanguageID': 'eng', 'SubFormat': 'srt', 'SubSumCD': '1', 'SubAuthorComment': '', 'SubAddDate': '2014-09-05 00:10:54', 'SubBad': '0', 'SubRating': '0.0', 'SubDownloadsCnt': '5324', 'MovieReleaseName': ' Person of Interest - 01x01 - Pilot.Web-DL.English.C.updated.Addic7ed.com', 'MovieFPS': '0.000', 'IDMovie': '83999', 'IDMovieImdb': '1941917', 'MovieName': '"Person of Interest" Pilot', 'MovieNameEng': '', 'MovieYear': '2011', 'MovieImdbRating': '8.5', 'SubFeatured': '0', 'UserNickName': '', 'SubTranslator': '', 'ISO639': 'en', 'LanguageName': 'English', 'SubComments': '0', 'SubHearingImpaired': '0', 'UserRank': '', 'SeriesSeason': '1', 'SeriesEpisode': '1', 'MovieKind': 'episode', 'SubHD': '0', 'SeriesIMDBParent': '1839578', 'SubEncoding': 'ASCII', 'SubAutoTranslation': '0', 'SubForeignPartsOnly': '0', 'SubFromTrusted': '0', 'QueryCached': 1, 'SubTSGroupHash': '80ff21163490f8d5e7d2817a6db92ceb', 'SubDownloadLink': 'http://dl.opensubtitles.org/en/download/src-api/vrf-19c60c5a/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/filead/1954390918.gz', 'ZipDownloadLink': 'http://dl.opensubtitles.org/en/download/src-api/vrf-f5490bba/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/subad/5812638', 'SubtitlesLink': 'http://www.opensubtitles.org/en/subtitles/5812638/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/person-of-interest-pilot-en', 'QueryNumber': '0', 'QueryParameters': {'imdbid': '1941917', 'sublanguageid': 'eng'}, 'Score': 10.05324}, {'MatchedBy': 'imdbid', 'IDSubMovieFile': '0', 'MovieHash': '0', 'MovieByteSize': '0', 'MovieTimeMS': '0', 'IDSubtitleFile': '1955349491', 'SubFileName': 'Person.of.Interest.S01E01.720p.HDTV.ReEnc-Max.srt', 'SubActualCD': '1', 'SubSize': '46822', 'SubHash': '72ac04035e444e22927b2020f930b9fc', 'SubLastTS': '00:42:32', 'SubTSGroup': '2', 'IDSubtitle': '6769377', 'UserID': '4332442', 'SubLanguageID': 'eng', 'SubFormat': 'srt', 'SubSumCD': '1', 'SubAuthorComment': '', 'SubAddDate': '2016-10-20 15:17:38', 'SubBad': '0', 'SubRating': '0.0', 'SubDownloadsCnt': '1471', 'MovieReleaseName': ' Person.of.Interest.S01E01.720p.HDTV.ReEnc-Max', 'MovieFPS': '23.976', 'IDMovie': '83999', 'IDMovieImdb': '1941917', 'MovieName': '"Person of Interest" Pilot', 'MovieNameEng': '', 'MovieYear': '2011', 'MovieImdbRating': '8.5', 'SubFeatured': '0', 'UserNickName': 'red1jhon', 'SubTranslator': '', 'ISO639': 'en', 'LanguageName': 'English', 'SubComments': '0', 'SubHearingImpaired': '0', 'UserRank': 'bronze member', 'SeriesSeason': '1', 'SeriesEpisode': '1', 'MovieKind': 'episode', 'SubHD': '1', 'SeriesIMDBParent': '1839578', 'SubEncoding': 'ASCII', 'SubAutoTranslation': '0', 'SubForeignPartsOnly': '0', 'SubFromTrusted': '0', 'QueryCached': 1, 'SubTSGroupHash': '17b1cfb76c2c9188e3772b16eb3b18be', 'SubDownloadLink': 'http://dl.opensubtitles.org/en/download/src-api/vrf-19d20c5b/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/filead/1955349491.gz', 'ZipDownloadLink': 'http://dl.opensubtitles.org/en/download/src-api/vrf-f57d0bc6/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/subad/6769377', 'SubtitlesLink': 'http://www.opensubtitles.org/en/subtitles/6769377/sid-wOog0I4sWDjuitBOr-UG5SbNRZ8/person-of-interest-pilot-en', 'QueryNumber': '0', 'QueryParameters': {'imdbid': '1941917', 'sublanguageid': 'eng'}, 'Score': 16.01471}]
def _search(self, imdb_id, language="eng"):
return self.subs.search_subtitles([{
"sublanguageid": language,
"imdbid": imdb_id
}])
def _search_multiple(self, imdb_ids, language="eng"):
request = []
for imdb_id in imdb_ids:
request.append({"sublanguageid": language, "imdbid": imdb_id})
return self.subs.search_subtitles(request)
def get_sub_link(self, imdb_id, language="eng"):
result = self._search(imdb_id, language)
sorted1 = sortby(result, "SubDownloadsCnt")
return sorted1[0]["SubDownloadLink"]
def get_sub_links(self, imdb_ids, language="eng"):
results = self._search_multiple(imdb_ids, language)
separated = self._separate_episodes(imdb_ids, results)
for key in separated:
sorted1 = sortby(separated[key], "SubDownloadsCnt")
#[print(x["SubFileName"],x["SubDownloadsCnt"]) for x in sorted1]
#print(sorted1[0]["SubFileName"],sorted1[0]["SubDownloadLink"])
separated[key] = sorted1[0]["SubDownloadLink"]
return separated
def _separate_episodes(self, imdb_ids, array):
separated = {}
for sub in array:
try:
separated[sub["IDMovieImdb"]].append(sub)
except KeyError:
separated[sub["IDMovieImdb"]] = [sub]
return separated
def __init__(self,
opensubs_email,
opensubs_password,
path,
language="eng"):
"""
init opensubtitles object, and token
:param path: path to media file
:param language: language of subtitles needed
:param opensubs_email: Email of OpenSubtitles account
:param opensubs_password: Password of OpenSubtitles account
"""
self.path = path
self.language = language
self.opensubtitles = OpenSubtitles()
self.token = self.opensubtitles.login(opensubs_email,
opensubs_password)
self.is_auth = False
if self.token:
self.is_auth = True
def main():
ost = OpenSubtitles()
token = ost.login(LOGIN, PASSWORD)
subs = get_list()
id_list = []
filenames = dict()
for sub in subs:
sub_name = get_filename(sub)
#subtitle already in directory
if sub_name in os.listdir():
continue
data = ost.search_subtitles([sub])
#no sub found
if len(data) == 0:
continue
sub_id = data[0]["IDSubtitleFile"]
id_list.append(int(sub_id))
filenames[sub_id] = sub_name
for file in id_list:
x = ost.download_subtitles([file], override_filenames = filenames)
def descargar_sub(movieid, path, ost=None):
if ost is None:
ost = OpenSubtitles()
token = ost.login('doctest', 'doctest')
langs = ['eng', 'spa', 'fre']
directory = os.path.join(path, movieid)
if not os.path.isdir(directory):
os.makedirs(directory)
for lang in langs:
found = ost.search_subtitles([{
'sublanguageid': lang,
'imdbid': movieid
}])
if found and len(found) > 0:
subid = found[0].get('IDSubtitleFile')
sub = ost.download_subtitles([subid],
output_directory=directory,
extension=lang + '.srt')
def download_all_subtitles(filepath):
dirname = os.path.dirname(filepath)
basename = os.path.basename(filepath)
basename_without_ext = os.path.splitext(basename)[0]
ost = OpenSubtitles()
ost.login(None, None)
f = File(filepath)
h = f.get_hash()
results_from_hash = (ost.search_subtitles([{
"sublanguageid": "all",
"moviehash": h
}]) or [])
languages_in_results_from_hash = [
lang_id
for lang_id in [r.get("SubLanguageID") for r in results_from_hash]
]
results_from_filename = [
r for r in ost.search_subtitles([{
"sublanguageid": "all",
"query": basename_without_ext
}])
]
results_from_filename_but_not_from_hash = [
r for r in results_from_filename if r.get("SubLanguageID")
and r.get("SubLanguageID") not in languages_in_results_from_hash
]
results = results_from_hash + results_from_filename_but_not_from_hash
wait_before_next_chunk = False
for chunk in _chunks(results, 10):
sub_ids = {
r["IDSubtitleFile"]: f'{basename_without_ext}.{r["ISO639"]}.srt'
for r in chunk
}
def _download_subtitle_chunk(retries=5):
nonlocal ost
try:
ost.download_subtitles(
[_id for _id in sub_ids.keys()],
override_filenames=sub_ids,
output_directory=dirname,
extension="srt",
)
except ProtocolError as e:
if retries == 0:
raise e
time.sleep(10)
ost = OpenSubtitles()
ost.login(None, None)
_download_subtitle_chunk(retries=retries - 1)
if wait_before_next_chunk:
time.sleep(10)
_download_subtitle_chunk()
wait_before_next_chunk = True
def main(video):
log("info", "Loaded module subtitles")
# Logging in to OST
log("info", "Logging in to OpenSubtitles")
ost = OpenSubtitles()
token = ost.login(opensubtitles_username, opensubtitles_password)
if isinstance(token, str):
log("success", "Logged in to OpenSubtitles")
else:
log("critical", "Invalid username / password entered")
return
# Opening File
file_path = video_path + video
if not path.exists(file_path):
log("critical", "Could not find specified video file '" + file_path + "'")
return
# Hashing file
log("info", "Generating video hash...")
f = File(file_path)
hash = f.get_hash()
log("success", "File hash generated: " + hash)
# Searching OST
log("info", "Querying OpenSubtitles for subtitles...")
data = ost.search_subtitles([{'sublanguageid': 'all', 'moviehash': hash}])
if len(data) > 0:
log("success", f"Found {len(data)} results.")
else:
log("warning", "No results found.")
# TODO: Implement series and episode-based downloading
return
subtitle_id = data[0]["IDSubtitleFile"]
log("info", "Attempting download of subtitles with ID " + str(subtitle_id))
try:
if isinstance(ost.download_subtitles([subtitle_id], override_filenames={subtitle_id: video + '.srt'}, output_directory=video_path, extension='srt'), dict):
log("success", "Subtitles successfully downloaded. Enjoy your video!")
else:
log("critical", "Subtitle download failed.")
return
except:
#TODO: Make neater
log("error", "Something went wrong. Trying second option in list(?)")
subtitle_id = data[1]["IDSubtitleFile"]
log("info", "Attempting download of subtitles with ID " + str(subtitle_id))
try:
if isinstance(ost.download_subtitles([subtitle_id], override_filenames={subtitle_id: video + '.srt'}, output_directory=video_path, extension='srt'), dict):
log("success", "Subtitles successfully downloaded. Enjoy your video!")
else:
log("critical", "Subtitle download failed.")
return
except:
log("critical", "giving up after too many tries")
return
def main():
# Run the program in current directory
# Get login details from user
( username, password ) = get_login_credentials()
# The OpenSubtitles object
ost = OpenSubtitles()
# Login to opensubtitles.org through the API
print("Logging in...")
while ost.login(username, password) is None:
print("Login failed, try again.")
( username, password ) = get_login_credentials()
# Creates the Subs directory if it doesn't exist
create_subs_dir()
# Gets a list of all the video file names in the directory (.mp4 or .mkv)
filename_list = get_video_filenames()
video_list = to_video_object_list(filename_list)
print("Downloading all subs...")
# Search and download subtitles for all videos
for video in video_list:
print("Downloading " + video.search_name + " S" + video.season + " E" + video.episode + "Subs")
# Creates a list of all the searched subtitles
search_data = ost.search_subtitles([{
'sublanguageid': 'eng',
'query': video.search_name.lower(),
'season': video.season,
'episode': video.episode
}])
# Finds subtitle file ID of most downloaded subtitles file
subtitle_file_id = find_most_downloaded(search_data)
ost.download_subtitles([subtitle_file_id], output_directory='.\\Subs\\',
override_filenames={str(subtitle_file_id): video.file_name + '-eng.srt'},
extension='srt')
print("Done!")
print("Logging out...")
ost.logout()
def download_all_subtitles(filepath):
dirname = os.path.dirname(filepath)
basename = os.path.basename(filepath)
basename_without_ext = os.path.splitext(basename)[0]
ost = OpenSubtitles()
ost.login(None, None)
f = File(filepath)
h = f.get_hash()
results = ost.search_subtitles([{"sublanguageid": "all", "moviehash": h}])
for chunk in _chunks(results, 20):
sub_ids = {
r["IDSubtitleFile"]:
f'{basename_without_ext}.{r["SubLanguageID"]}.srt'
for r in chunk
}
ost.download_subtitles(
[_id for _id in sub_ids.keys()],
override_filenames=sub_ids,
output_directory=dirname,
extension="srt",
)
class CommonMetadataOpenSubtitles(object):
"""
Class for interfacing with Opensubtitles
"""
def __init__(self, user_name, user_password):
self.opensubtitles_inst = OpenSubtitles()
self.token = self.opensubtitles_inst.login(user_name, user_password)
def com_meta_opensub_search(self, file_name):
f = File(file_name)
return self.opensubtitles_inst.search_subtitles([{'sublanguageid': 'all',
'moviehash': f.get_hash(),
'moviebytesize': f.size}])
def com_meta_opensub_ping(self):
self.opensubtitles_inst.no_operation()
def com_meta_opensub_logoff(self):
self.opensubtitles_inst.logout()
def __init__(self, user_name, user_password):
self.opensubtitles_inst = OpenSubtitles()
self.token = self.opensubtitles_inst.login(user_name, user_password)
class OpenSubs:
def __init__(self, username, password):
"""Instantiate the shit out of this
Arguments:
- username (str): Your OpenSubtitles username
- password (str): Your OpenSubtitles password
Returns:
- Nothing: Sets all parameters of the class.
"""
self.username = username
self.password = password
self.opensubs = OpenSubtitles()
self.token = self.opensubs.login(self.username, self.password)
assert self.token != None, "Incorrect username/password (or something else went wrong when loging in...)"
def subtitlesInfo(self, imdbID, language="eng"):
"""Get info from OpenSubtitles for a movie'
Arguments:
- imdbID (str): IMDB ID of the movie (with or without the "tt" prefix)
- language (str): ISO 639-2 Code for the language you want the subtitles in (English by default)
Check http://www.loc.gov/standards/iso639-2/php/code_list.php for other language codes
Returns:
If subtitles found:
- subtitlesInfo (dict): Dictionary containing information about the movie's subtitles, including download links.
Otherwise:
- None
"""
# Process imdbID to remove TT
if (imdbID.startswith("tt")):
imdbID = imdbID[2:]
# Create search parameters dict
searchParams = {}
searchParams["sublanguageid"] = language
searchParams["imdbid"] = str(imdbID)
# Make sure it downloaded something
returnedDict = self.opensubs.search_subtitles([searchParams])
assert returnedDict != None, "OpenSubtitles returned nothing, check that you input the correct IMDBid and you input the correct username and password when instantiating this object."
# Check not null
if not returnedDict:
return
# Index 0 for top result
return returnedDict[0]
def downloadSubtitles(self, imdbID, language="eng", outputFolder=None):
"""Download whatever "subtitlesInfo(self,imdbID, language)" outputs into outputFolder
Arguments:
- imdbID (str): IMDB ID of the movie (with or without the "tt" prefix)
- language (str): ISO 639-2 Code for the language you want the subtitles in (English by default)
Check http://www.loc.gov/standards/iso639-2/php/code_list.php for other language codes
- outputFolder (str): FULL path to folder where you want to download subtitles.
* If (outputPath == None) then outputPath = currentWorkingDirectory/subtitles
* If you specify outputFolder please make sure you're inputting the FULL (and not relative) path,
as well as making sure the folder already exists
Returns:
If subtitles found:
- Nothing: Downloads file that subtitlesInfo outputs into outputFolder/imdbID.sub (or whatever other format OpenSubtitles returns)
Otherwise:
If subtitles not available in OpenSubtitles.org return (int) -999
If HTTP error (most likely you've reached your daily download limit return (int) HTTP error code)
"""
# Output path shenanigans
if outputFolder == None:
outputFolder = os.getcwd() + "/subtitles"
# Create subtitles folder if it doesn't exist
try:
os.makedirs(outputFolder)
except OSError as e:
if e.errno != errno.EEXIST:
raise
# Process imdbID to remove TT
if (imdbID.startswith("tt")):
imdbID = imdbID[2:]
# Get subtitles info
subtitlesInfo = self.subtitlesInfo(imdbID, language)
# If no subtitles found print warning and return -999
if not subtitlesInfo:
print("No subtitles found for", imdbID)
return -999
# Generate filename and download
isGZ = False
isGZ_string = subtitlesInfo['SubDownloadLink'].split(".")[-1]
if isGZ_string == "gz":
isGZ = True
filename = outputFolder + "/tt" + str(
imdbID) + "." + subtitlesInfo['SubFormat']
if isGZ:
filename += ".gz"
try:
# Download
urllib.request.urlretrieve(subtitlesInfo['SubDownloadLink'],
filename)
except Exception as e:
# If HTTP error print error and return HTTP code (e.g. 404)
print(e)
return e.code
def download_subs_and_cache_results():
import argparse
parser = argparse.ArgumentParser(
description='TV Series Tools: Download subtitles for movies specified'
' by passed IMDB IDs')
parser.add_argument(
'--input',
'-i',
dest='inputfile',
required=True,
help='movies CSV file: 1st column = IMDB ID, 2nd column = title')
parser.add_argument('--output',
'-o',
dest='outputdir',
required=True,
help='path to a directory in which the subtitle SRT'
' files will be downloaded')
parser.add_argument('--cache',
'-c',
dest='cachedir',
required=True,
help='cache location')
parser.add_argument('--shuffle',
'-s',
dest='shuffle',
action='store_true',
help='shuffle the list of IMDB IDs')
parser.add_argument(
'--timeout',
'-t',
dest='timeout',
type=int,
default=DEFAULT_TIMEOUT,
help='time in seconds between OpenSubtitles\'s API'
' calls, if not set {} is used'.format(DEFAULT_TIMEOUT))
args = parser.parse_args()
opensub = OpenSubtitles()
opensub.login(os.environ['OPENSUB_USER'], os.environ['OPENSUB_PASSWD'])
movies = listio.read_map(args.inputfile)
if args.shuffle:
movies = list(movies)
random.shuffle(movies)
excl_ids = []
for f in (FILE_NAME_DOWNLOADED, FILE_NAME_FAILED):
try:
lines = listio.read_map(os.path.join(args.cachedir, f))
for line in lines:
excl_ids.append(line[0])
except FileNotFoundError:
pass
if not os.path.isdir(args.cachedir):
os.makedirs(args.cachedir)
if not os.path.isdir(args.outputdir):
os.makedirs(args.outputdir)
write_result(
os.path.join(args.cachedir, FILE_NAME_DOWNLOADED),
os.path.join(args.cachedir, FILE_NAME_FAILED),
download_subs(movies,
excl_ids,
args.outputdir,
opensub,
timeout=args.timeout))
sys.exit()
# Simple Python script to download subtitles of movies # https://github.com/prashantsengar/SubtitlesDL # [email protected] import requests from pythonopensubtitles.utils import File from pythonopensubtitles.opensubtitles import OpenSubtitles ost = OpenSubtitles() from zipfile import ZipFile import os import time EMAIL = "YOUR OpenSubtitles email" PASSW = "YOUR OpenSubtitles password" token = ost.login(EMAIL, PASSW) filename=input("Enter path to file: ") sub_path = os.path.dirname(filename) name = '.'.join(filename.split('.')[0:-1]) sname = name+'.srt' zname = name+'.zip' f = File(filename) hashh = f.get_hash() size = f.size print(hashh) print(size) data = ost.search_subtitles([{'sublanguageid': 'all', 'moviehash': hashh, 'moviebytesize': size}]) assert type(data)!=None
#!/usr/bin/env python
from pythonopensubtitles.opensubtitles import OpenSubtitles
from pythonopensubtitles.utils import File
import ConfigParser
import urllib
from tempfile import mktemp
import gzip
from numbers import Number
import os
config = ConfigParser.RawConfigParser()
config.read(os.path.realpath(__file__)[:-2] + 'cfg')
osmgr = OpenSubtitles()
osmgr.login(config.get('Login', 'user'), config.get('Login', 'pass'))
def find_subtitles(path, langid='all'):
''' Get a list of subtitles for a given file '''
try:
f = File(path)
hash = f.get_hash()
assert type(hash) == str
size = f.size
assert isinstance(size, Number)
data = osmgr.search_subtitles([{
'sublanguageid': langid,
'moviehash': hash,
'moviebytesize': size
}])
assert type(data) == list
return data
f.write(content.decode(encoding))
break
if args.move:
destination = os.path.join(folder, title + ext)
if os.path.abspath(video_path) != os.path.abspath(destination):
print('Moving video: ' + title + ext)
shutil.move(video_path, destination)
else:
print('Not moving: same location')
return True
Settings.USER_AGENT = 'OSTestUserAgentTemp'
opensubtitles = OpenSubtitles()
token = opensubtitles.login('marxin', 'spartapraha')
if not type(token) == str:
print("Can't login to opensubtitles.org")
exit(1)
file_list = []
for root, dirs, files in os.walk(args.folder):
for file in files:
if is_video_ext(os.path.splitext(file)[1]):
file_list.append(os.path.join(root, file))
unresolved = 0
#!/usr/bin/env python
from pythonopensubtitles.opensubtitles import OpenSubtitles
from pythonopensubtitles.utils import File
import ConfigParser
import urllib
from tempfile import mktemp
import gzip
from numbers import Number
import os
config = ConfigParser.RawConfigParser()
config.read(os.path.realpath(__file__)[:-2] + 'cfg')
osmgr = OpenSubtitles()
osmgr.login(config.get('Login', 'user'), config.get('Login', 'pass'))
def find_subtitles(path, langid='all'):
''' Get a list of subtitles for a given file '''
try:
f = File(path)
hash = f.get_hash()
assert type(hash) == str
size = f.size
assert isinstance(size, Number)
data = osmgr.search_subtitles([{'sublanguageid': langid,
'moviehash': hash,
'moviebytesize': size}])
assert type(data) == list
return data
except:
import sys
import os
from pythonopensubtitles.opensubtitles import OpenSubtitles
fd = OpenSubtitles()
class Data(object):
username = '******'
password = '******'
name = 'Trance'
path = ('/home/seberm/Downloads/'
'Trance.2013 WEBRip XViD juggs')
video = 'Trance.2013 WEBRip XViD juggs.avi'
#title = 'Dark.City.1998.Directors.Cut.BRRip.H264.AAC.5.1ch.Gopo.srt'
#subtitle = 'Dark.City.1998.Directors.Cut.BRRip.H264.AAC.5.1ch.Gopo.srt'
token = fd.login(Data.username, Data.password)
if not token:
print("Chyba prihlaseni")
sys.exit(1)
print(token)
from pythonopensubtitles.utils import File
f = File(os.path.join(Data.path, Data.video))
h = f.get_hash()
print("Hash: %s" % h)
print("Size: %f" % f.size)
from pathlib import Path
from pythonopensubtitles.opensubtitles import OpenSubtitles
from pythonopensubtitles.utils import File
import json
import time
import pysrt
import pandas as pd
from sources.common_functions import *
data_path = Path('./data')
with open('../.config', 'r') as r:
creds = json.load(r)['opensubtitles']
# -
ost = OpenSubtitles()
ost.login(creds['user'], creds['key'])
# # Download Subtitles
# +
movies_raw_path = data_path / 'movies_raw'
movies_prep_path = data_path / 'movies_prep'
movie_file_dict = get_movie_file_dict(movies_raw_path)
movies_features_to_extract = get_movies_status(movies_raw_path,
movies_prep_path)
movies_to_extract = [
k for k, v in movies_features_to_extract.items() if 'subtitle' in v
]
movies_to_extract
def __init__(self, username, password):
self.username = username
self.password = password
self._ost = OpenSubtitles()
class SubtitleFinder:
def __init__(self, skip_subsync, ost_username, ost_password, tmdb_key):
self.skip_subsync = skip_subsync
self.ost = OpenSubtitles()
try:
self.ost.login(ost_username, ost_password)
except Exception as e:
logging.error("Failed to log into opensubtitles.org.")
raise e
self.ost_language = 'eng'
if tmdb_key:
self.media_searcher = MediaSearcher(tmdb_key)
if not self.skip_subsync:
self.subsync_parser = ffsubsync.make_parser()
def find_and_download(self, source, imdb_id):
# subtitles matching on hash are already synced
subs_data = self.find_subtitles_by_hash(source)
if subs_data:
return self.download_subtitles(subs_data)
parsed_media = filename_parser.parse(source)
if imdb_id is None and self.media_searcher is not None:
imdb_id = self.media_searcher.search(parsed_media)
if imdb_id is not None:
subs_data = self.find_subtitles_by_id(imdb_id)
elif isinstance(parsed_media, ParsedMovie):
subs_data = self.find_subtitles_for_movie(parsed_media.title)
elif isinstance(parsed_media, ParsedSeries):
subs_data = self.find_subtitles_for_episode(parsed_media.title, parsed_media.season, parsed_media.episode)
# sync subs unless explicitly asked not to
if subs_data is not None:
subs = self.download_subtitles(subs_data)
if self.skip_subsync:
return subs
else:
return self.sync_subtitles(source, subs)
def find_subtitles_by_hash(self, source):
f = File(source)
return self.find_subtitles(moviehash=f.get_hash(), moviebytesize=f.size)
def find_subtitles_by_id(self, imdb_id):
return self.find_subtitles(imdbid=imdb_id)
def find_subtitles_for_movie(self, title):
return self.find_subtitles(query=title)
def find_subtitles_for_episode(self, title, season, episode):
return self.find_subtitles(query=title, season=season, episode=episode)
def find_subtitles(self, **request):
request.update(sublanguageid=self.ost_language)
if 'imdbid' in request and request['imdbid'][:2] == 'tt':
request.update(imdbid=request['imdbid'][2:])
subs_data = self.ost.search_subtitles([request])
return subs_data
def download_subtitles(self, subs_data):
id_subtitle_file = subs_data[0].get('IDSubtitleFile')
subs_dict = self.ost.download_subtitles([id_subtitle_file], return_decoded_data=True)
raw_subs = subs_dict.get(id_subtitle_file)
return pysubs2.SSAFile.from_string(raw_subs)
def sync_subtitles(self, video_filename, subtitles):
with tempfile.NamedTemporaryFile(delete=False, suffix='.srt') as tmp_unsynced:
tmp_unsynced.write(subtitles.to_string('srt').encode())
tmp_unsynced.close()
with tempfile.NamedTemporaryFile(suffix='.srt') as tmp_synced:
tmp_synced.close()
self.run_subsync(video_filename, tmp_unsynced.name, tmp_synced.name)
return pysubs2.load(tmp_synced.name)
def run_subsync(self, reference, srtin, srtout):
ffsubsync.run(self.subsync_parser.parse_args([
reference,
'-i', srtin,
'-o', srtout
]))
