X-Git-Url: https://git.phdru.name/?p=extfs.d.git;a=blobdiff_plain;f=torrent;h=7841fa7e672c950180e94414f2426a5e440891ba;hp=cc9dc3dcc2d46691f63591888777f0858123cdc0;hb=HEAD;hpb=f8149878b4b42033610eef9b305e6b71c2e5f549 diff --git a/torrent b/torrent index cc9dc3d..a0f827e 100755 --- a/torrent +++ b/torrent @@ -1,4 +1,4 @@ -#! /usr/bin/env python +#! /usr/bin/env python3 """Torrent Virtual FileSystem for Midnight Commander The script requires Midnight Commander 3.1+ @@ -15,7 +15,7 @@ file the command is "%cd"): cd file/torrent://; In older versions it is cd file#torrent, where "file" is the name of your torrent metafile. See detailed installation instructions at -http://phdru.name/Software/mc/torrent_INSTALL.html. +https://phdru.name/Software/mc/torrent_INSTALL.html. The VFS lists all files and directories from the torrent metafile; all files appear empty, of course, but the sizes are shown. Filenames are reencoded from @@ -27,46 +27,53 @@ contents of these files are taken from the corresponding fields in the torrent metafile. The script doesn't check if the torrent consists of a .META file or directory (quite unlikely). -Date/time for all files is set to midnight of the 1st January of the current -year. The filesystem is, naturally, read-only. +Date/time for all directories/files is set to the value of 'creation date' +field, if it exists; if not date/time is set to the last modification time of +the torrent file itself. + +The filesystem is, naturally, read-only. """ -__version__ = "1.2.3" +__version__ = "1.3.1" __author__ = "Oleg Broytman " -__copyright__ = "Copyright (C) 2010-2015 PhiloSoft Design" +__copyright__ = "Copyright (C) 2010-2023 PhiloSoft Design" __license__ = "GPL" -from os.path import getmtime +from datetime import datetime +from os.path import dirname, getmtime import sys from time import localtime, asctime from eff_bdecode import decode try: - import locale - use_locale = True + import locale + use_locale = True except ImportError: - use_locale = False + use_locale = False if use_locale: - # Get the default charset. - try: - lcAll = locale.getdefaultlocale() - except locale.Error, err: - print >>sys.stderr, "WARNING:", err - lcAll = [] - - if len(lcAll) == 2: - default_encoding = lcAll[1] - else: - try: - default_encoding = locale.getpreferredencoding() - except locale.Error, err: - print >>sys.stderr, "WARNING:", err - default_encoding = sys.getdefaultencoding() + # Get the default charset. + try: + if sys.version_info[:2] < (3, 11): + lcAll = locale.getdefaultlocale() + else: + lcAll = [] + except locale.Error as err: + #print("WARNING:", err, file=sys.stderr) + lcAll = [] + + if len(lcAll) == 2: + default_encoding = lcAll[1] + else: + try: + default_encoding = locale.getpreferredencoding() + except locale.Error as err: + #print("WARNING:", err, file=sys.stderr) + default_encoding = sys.getdefaultencoding() else: - default_encoding = sys.getdefaultencoding() + default_encoding = sys.getdefaultencoding() import logging logger = logging.getLogger('torrent-mcextfs') @@ -82,19 +89,23 @@ Author: %s This is not a program. Put the script in $HOME/[.local/share/].mc/extfs.d or /usr/[local/][lib|share]/mc/extfs. For more information read the source!""", - __version__, __author__, __copyright__ -) + __version__, __author__, __copyright__) sys.exit(1) locale.setlocale(locale.LC_ALL, '') +PY3 = (sys.version_info[0] >= 3) +if PY3: + def output(s): + sys.stdout.buffer.write(s.encode(default_encoding, 'replace') + b'\n') +else: + def output(s): + sys.stdout.write(s + '\n') + def mctorrent_list(): """List the entire VFS""" - if 'info' not in torrent: - torrent_error('Info absent') - info = torrent['info'] if 'name' not in info and 'name.utf-8' not in info: torrent_error('Unknown name') @@ -119,27 +130,13 @@ def mctorrent_list(): if 'path.utf-8' in file: if name_utf8: path = '/'.join([name_utf8] + file['path.utf-8']) - if default_encoding != 'utf-8': - path = path.decode('utf-8', 'replace').encode(default_encoding, 'replace') else: - _name_utf8 = name - if encoding and (encoding != 'utf-8'): - _name_utf8 = _name_utf8.decode(encoding, 'replace').encode('utf-8', 'replace') - path = '/'.join([_name_utf8] + file['path.utf-8']) - if default_encoding != 'utf-8': - path = path.decode('utf-8', 'replace').encode(default_encoding, 'replace') + path = '/'.join([name] + file['path.utf-8']) else: if name_utf8: - path = file['path'] - if encoding and (encoding != 'utf-8'): - path = path.decode(encoding, 'replace').encode('utf-8', 'replace') path = '/'.join([name_utf8] + path) - if default_encoding != 'utf-8': - path = path.decode('utf-8', 'replace').encode(default_encoding, 'replace') else: path = '/'.join([name] + file['path']) - if encoding and (default_encoding != encoding): - path = path.decode(encoding, 'replace').encode(default_encoding, 'replace') length = file['length'] paths.append((path, length)) else: # One-file torrent @@ -147,10 +144,7 @@ def mctorrent_list(): torrent_error('Unknown length') length = info['length'] if name_utf8: - if default_encoding != 'utf-8': - name = name_utf8.decode('utf-8', 'replace').encode(default_encoding, 'replace') - elif encoding and (default_encoding != encoding): - name = name.decode(encoding, 'replace').encode(default_encoding, 'replace') + name = name_utf8 paths = [(name, length)] meta = [] @@ -158,7 +152,7 @@ def mctorrent_list(): 'created by', 'creation date', 'encoding', \ 'nodes', 'publisher', 'publisher-url': if name == 'comment' and 'comment.utf-8' in torrent: - data = torrent['comment.utf-8'].decode('utf-8').encode(default_encoding, 'replace') + data = torrent['comment.utf-8'] meta.append(('.META/' + name, len(data))) elif name in torrent: if name == 'announce-list': @@ -182,11 +176,20 @@ def mctorrent_list(): if 'piece length' in info: meta.append(('.META/piece length', len(str(info['piece length'])))) + paths += meta + dirs = set() + for name, size in paths: + if '/' in name: + dirs.add(dirname(name)) + if not dt: dt = decode_datetime(getmtime(sys.argv[2])) - for name, size in paths + meta: - print "-r--r--r-- 1 user group %d %s %s" % (size, dt, name) + for name in sorted(dirs): + output("dr-xr-xr-x 1 user group 0 %s %s" % (dt, name)) + + for name, size in sorted(paths): + output("-r--r--r-- 1 user group %d %s %s" % (size, dt, name)) def mctorrent_copyout(): @@ -200,7 +203,7 @@ def mctorrent_copyout(): 'created by', 'creation date', 'encoding', \ 'nodes', 'publisher', 'publisher-url': if name == 'comment' and 'comment.utf-8' in torrent: - data = torrent['comment.utf-8'].decode('utf-8').encode(default_encoding, 'replace') + data = torrent['comment.utf-8'] elif torrent_filename == '.META/' + name: if name in torrent: if name == 'announce-list': @@ -210,7 +213,8 @@ def mctorrent_copyout(): elif name == 'creation date': data = decode_datetime_asc(torrent[name]) elif name == 'nodes': - data = ['%s:%s' % (host, port) for host, port in torrent[name]] + data = ['%s:%s' % (host, port) + for host, port in torrent[name]] data = '\n'.join(data) else: data = str(torrent[name]) @@ -219,15 +223,13 @@ def mctorrent_copyout(): break if torrent_filename in ('.META/private', '.META/piece length'): - if 'info' not in torrent: - torrent_error('Info absent') info = torrent['info'] if torrent_filename == '.META/private': if 'private' not in info: - torrent_error('Info absent') + torrent_error('Private absent') if torrent_filename == '.META/piece length': if 'piece length' not in info: - torrent_error('Info absent') + torrent_error('Piece length absent') data = str(info[torrent_filename[len('.META/'):]]) if not torrent_filename.startswith('.META/'): @@ -236,56 +238,122 @@ def mctorrent_copyout(): if data is None: torrent_error('Unknown file name') else: - outfile = open(real_filename, 'w') + outfile = open(real_filename, 'wt') outfile.write(data) outfile.close() def mctorrent_copyin(): """Put a file to the VFS""" - sys.exit("Torrent VFS doesn't support adding files (read-only filesystem)") + sys.exit("Torrent VFS doesn't support adding/overwriting files " + "(read-only filesystem)") + def mctorrent_rm(): """Remove a file from the VFS""" - sys.exit("Torrent VFS doesn't support removing files/directories (read-only filesystem)") + sys.exit("Torrent VFS doesn't support removing files/directories " + "(read-only filesystem)") + mctorrent_rmdir = mctorrent_rm + def mctorrent_mkdir(): """Create a directory in the VFS""" - sys.exit("Torrent VFS doesn't support creating directories (read-only filesystem)") + sys.exit("Torrent VFS doesn't support creating directories " + "(read-only filesystem)") def torrent_error(error_str): logger.critical("Error parsing the torrent metafile: %s", error_str) sys.exit(1) + +def decode_dict(d, encoding): + new_d = {} + for k in d: + v = d[k] + k = k.decode(encoding) + if isinstance(v, dict): + v = decode_dict(v, encoding) + elif isinstance(v, list): + v = decode_list(v, encoding) + elif isinstance(v, bytes): + v = v.decode(encoding) + new_d[k] = v + return new_d + + +def decode_list(l, encoding): + new_l = [] + for v in l: + if isinstance(v, dict): + v = decode_dict(v, encoding) + elif isinstance(v, list): + v = decode_list(v, encoding) + elif isinstance(v, bytes): + v = v.decode(encoding) + new_l.append(v) + return new_l + + def decode_torrent(): try: - torrent_file = open(sys.argv[2], 'r') + torrent_file = open(sys.argv[2], 'rb') data = torrent_file.read() torrent_file.close() - return decode(data) - except IOError, error_str: + torrent = decode(data) + except IOError as error_str: torrent_error(error_str) + del torrent[b'info'][b'pieces'] + if b'info' not in torrent: + torrent_error('Info absent') + + if PY3: + codepage = torrent.get(b'codepage', None) + encoding = torrent.get(b'encoding', None) + if encoding: + encoding = encoding.decode('ascii') + elif codepage: + encoding = codepage.decode('ascii') + else: + for encoding in ('ascii', 'utf-8', default_encoding): + try: + return decode_dict(torrent, encoding) + except UnicodeDecodeError: + pass + torrent_error('UnicodeDecodeError') + return decode_dict(torrent, encoding) + + return torrent + def decode_datetime_asc(dt): - return asctime(localtime(float(dt))) + try: + return asctime(localtime(float(dt))) + except ValueError: + return datetime.max.ctime() + def decode_datetime(dt): - Y, m, d, H, M = localtime(float(dt))[0:5] - return "%02d-%02d-%d %02d:%02d" % (m, d, Y, H, M) + try: + Y, m, d, H, M = localtime(float(dt))[0:5] + except ValueError: + return datetime.max.ctime() + else: + return "%02d-%02d-%d %02d:%02d" % (m, d, Y, H, M) + def decode_announce_list(announce): - return '\n'.join(l[0] for l in announce) + return '\n'.join(a[0] for a in announce if a) command = sys.argv[1] procname = "mctorrent_" + command g = globals() -if not g.has_key(procname): +if procname not in g: logger.critical("Unknown command %s", command) sys.exit(1) @@ -295,5 +363,5 @@ try: g[procname]() except SystemExit: raise -except: +except Exception: logger.exception("Error during run")