83 lines
2.2 KiB
Python
83 lines
2.2 KiB
Python
# The contents of this file are subject to the BitTorrent Open Source License
|
|
# Version 1.1 (the License). You may not copy or use this file, in either
|
|
# source code or executable form, except in compliance with the License. You
|
|
# may obtain a copy of the License at http://www.bittorrent.com/license/.
|
|
#
|
|
# Software distributed under the License is distributed on an AS IS basis,
|
|
# WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
|
|
# for the specific language governing rights and limitations under the
|
|
# License.
|
|
|
|
# GetTorrent -- abstraction which can get a .torrent file from multiple
|
|
# sources: local file, url, etc.
|
|
|
|
# written by Matt Chisholm
|
|
|
|
import os
|
|
import re
|
|
import zurllib
|
|
from bencode import bdecode
|
|
from BitTorrent.platform import get_cache_dir
|
|
|
|
urlpat = re.compile('^\w+://')
|
|
|
|
def get_quietly(arg):
|
|
(data, errors) = get(arg)
|
|
# If there's an error opening a file from the IE cache,
|
|
# act like we simply didn't get a file (because we didn't)
|
|
if errors:
|
|
cache = get_cache_dir()
|
|
if (cache is not None) and (cache in arg):
|
|
errors = []
|
|
return data, errors
|
|
|
|
def get(arg):
|
|
data = None
|
|
errors = []
|
|
if os.access(arg, os.F_OK):
|
|
data, errors = get_file(arg)
|
|
elif urlpat.match(arg):
|
|
data, errors = get_url(arg)
|
|
else:
|
|
errors.append(_("Could not read %s") % arg)
|
|
return data, errors
|
|
|
|
|
|
def get_url(url):
|
|
data = None
|
|
errors = []
|
|
err_str = _("Could not download or open \n%s\n"
|
|
"Try using a web browser to download the torrent file.") % url
|
|
u = None
|
|
try:
|
|
u = zurllib.urlopen(url)
|
|
data = u.read()
|
|
u.close()
|
|
b = bdecode(data)
|
|
except Exception, e:
|
|
if u is not None:
|
|
u.close()
|
|
errors.append(err_str + "\n(%s)" % e)
|
|
data = None
|
|
else:
|
|
if u is not None:
|
|
u.close()
|
|
|
|
return data, errors
|
|
|
|
|
|
def get_file(filename):
|
|
data = None
|
|
errors = []
|
|
f = None
|
|
try:
|
|
f = file(filename, 'rb')
|
|
data = f.read()
|
|
f.close()
|
|
except Exception, e:
|
|
if f is not None:
|
|
f.close()
|
|
errors.append((_("Could not read %s") % filename) + (': %s' % str(e)))
|
|
|
|
return data, errors
|