mirror of
https://github.com/rembo10/headphones.git
synced 2026-05-08 12:49:32 +01:00
Exception handling for torrent name regular expression
Fall back to the torrent file's name in case reular expression extraction method from the torrent file doesn't work.
This commit is contained in:
@@ -74,8 +74,15 @@ def add_torrent(link, data=None):
|
||||
with open(link, 'rb') as f:
|
||||
metainfo = str(base64.b64encode(f.read()))
|
||||
# Extract torrent name from .torrent
|
||||
name_length = int( re.findall( 'name([0-9]*)\:.*?\:', base64.b64encode(metainfo) )[0] )
|
||||
name = re.findall('name[0-9]*\:(.*?)\:', base64.b64encode(metainfo) )[0][:size]
|
||||
try:
|
||||
name_length = int( re.findall( 'name([0-9]*)\:.*?\:', base64.b64encode(metainfo) )[0] )
|
||||
name = re.findall('name[0-9]*\:(.*?)\:', base64.b64encode(metainfo) )[0][:size]
|
||||
except:
|
||||
# get last part of link/path (name only)
|
||||
name = link.split('\\')[-1].split('/')[-1]
|
||||
# remove '.torrent' suffix
|
||||
if name[-len('.torrent'):] == '.torrent':
|
||||
name = name[:-len('.torrent')]
|
||||
result = {
|
||||
'type' : 'torrent',
|
||||
'name' : name,
|
||||
|
||||
Reference in New Issue
Block a user