def mainloop(self):
""" The main loop.
"""
if len(self.args) == 1 and "=urn:btih:" in self.args[0]:
# Handle magnet link
self.make_magnet_meta(self.args[0])
return
if not self.args:
self.parser.print_help()
self.parser.exit()
elif len(self.args) < 2:
self.parser.error("Expected a path and at least one announce URL, got: %s" % (' '.join(self.args),))
# Create and configure metafile factory
datapath = self.args[0].rstrip(os.sep)
metapath = datapath
if self.options.output_filename:
metapath = self.options.output_filename
if os.path.isdir(metapath):
metapath = os.path.join(metapath, os.path.basename(datapath))
if not metapath.endswith(".torrent"):
metapath += ".torrent"
torrent = metafile.Metafile(metapath)
torrent.ignore.extend(self.options.exclude)
def callback(meta):
"Callback to set label and resume data."
if self.options.cross_seed:
meta["info"]["x_cross_seed_label"] = self.options.cross_seed
if self.options.no_cross_seed:
del meta["info"]["x_cross_seed"]
# Set specific keys?
metafile.assign_fields(meta, self.options.set)
# Create and write the metafile(s)
# TODO: make it work better with multiple trackers (hash only once), also create fast-resume file for each tracker
meta = torrent.create(datapath, self.args[1:],
progress=None if self.options.quiet else metafile.console_progress(),
root_name=self.options.root_name, private=self.options.private, no_date=self.options.no_date,
comment=self.options.comment, created_by="PyroScope %s" % self.version, callback=callback
)
# Create second metafile with fast-resume?
if self.options.hashed:
try:
metafile.add_fast_resume(meta, datapath)
except EnvironmentError, exc:
self.fatal("Error making fast-resume data (%s)" % (exc,))
raise
hashed_path = re.sub(r"\.torrent$", "", metapath) + "-resume.torrent"
self.LOG.info("Writing fast-resume metafile %r..." % (hashed_path,))
try:
bencode.bwrite(hashed_path, meta)
except EnvironmentError, exc:
self.fatal("Error writing fast-resume metafile %r (%s)" % (hashed_path, exc,))
raise
def check_torrent(torrent_file, seed_dir):
metainfo = bencode.bread(torrent_file)
name = metainfo['info']['name']
print name
torrent = metafile.Metafile(torrent_file)
path = os.path.join(seed_dir, name)
try:
success = torrent.check(metainfo, path, metafile.console_progress())
except OSError:
success = False
dest_dir = os.path.join(os.path.dirname(torrent_file),
'success' if success else 'failure')
if not os.path.isdir(dest_dir):
os.mkdir(dest_dir)
shutil.move(torrent_file, dest_dir)
def mainloop(self):
""" The main loop.
"""
if not self.args:
self.parser.print_help()
self.parser.exit()
elif len(self.args) < 1:
self.parser.error("Expecting at least a metafile name")
# Read metafile
metapath = self.args[0]
try:
metainfo = bencode.bread(metapath)
except (KeyError, bencode.BencodeError) as exc:
self.fatal("Bad metafile %r (%s)" % (metapath, type(exc).__name__), exc)
raise
else:
# Check metafile integrity
try:
metafile.check_meta(metainfo)
except ValueError as exc:
self.fatal("Metafile %r failed integrity check" % (metapath,), exc)
raise
else:
if len(self.args) > 1:
datapath = self.args[1].rstrip(os.sep)
else:
datapath = metainfo["info"]["name"]
# Check the hashes
torrent = metafile.Metafile(metapath)
try:
ok = torrent.check(metainfo, datapath,
progress=None if self.options.quiet else metafile.console_progress())
if not ok:
self.fatal("Metafile %r has checksum errors" % (metapath,))
sys.exit(1)
except OSError as exc:
self.fatal("Torrent data file missing", exc)
raise
def mainloop(self):
""" The main loop.
"""
if len(self.args) == 1 and "=urn:btih:" in self.args[0]:
# Handle magnet link
self.make_magnet_meta(self.args[0])
return
if not self.args:
self.parser.print_help()
self.parser.exit()
elif len(self.args) < 2:
self.parser.error(
"Expected a path and at least one announce URL, got: %s" %
(' '.join(self.args), ))
# Create and configure metafile factory
datapath = self.args[0].rstrip(os.sep)
metapath = datapath
if self.options.output_filename:
metapath = self.options.output_filename
if os.path.isdir(metapath):
metapath = os.path.join(metapath, os.path.basename(datapath))
if not metapath.endswith(".torrent"):
metapath += ".torrent"
torrent = metafile.Metafile(metapath)
torrent.ignore.extend(self.options.exclude)
def callback(meta):
"Callback to set label and resume data."
if self.options.cross_seed:
if self.options.cross_seed == "@entropy":
meta["info"]["entropy"] = format(
random.getrandbits(self.ENTROPY_BITS),
'x').zfill(self.ENTROPY_BITS // 4)
else:
meta["info"][
"x_cross_seed_label"] = self.options.cross_seed
if self.options.no_cross_seed:
del meta["info"]["x_cross_seed"]
# Set specific keys?
metafile.assign_fields(meta, self.options.set, self.options.debug)
# Create and write the metafile(s)
# TODO: make it work better with multiple trackers (hash only once), also create fast-resume file for each tracker
meta = torrent.create(datapath,
self.args[1:],
progress=None if self.options.quiet else
metafile.console_progress(),
root_name=self.options.root_name,
private=self.options.private,
no_date=self.options.no_date,
comment=self.options.comment,
created_by="PyroScope %s" % self.version,
callback=callback)
tied_file = metapath
# Create second metafile with fast-resume?
if self.options.hashed:
try:
metafile.add_fast_resume(meta, datapath)
except EnvironmentError as exc:
self.fatal("Error making fast-resume data (%s)" % (exc, ))
raise
hashed_path = re.sub(r"\.torrent$", "",
metapath) + "-resume.torrent"
self.LOG.info("Writing fast-resume metafile %r..." %
(hashed_path, ))
try:
bencode.bwrite(hashed_path, meta)
tied_file = hashed_path
except EnvironmentError as exc:
self.fatal("Error writing fast-resume metafile %r (%s)" % (
hashed_path,
exc,
))
raise
# Load into client on demand
if self.options.load or self.options.start:
proxy = config.engine.open()
info_hash = metafile.info_hash(meta)
try:
item_name = proxy.d.name(info_hash, fail_silently=True)
except xmlrpc.HashNotFound:
load_item = proxy.load.start_verbose if self.options.start else proxy.load.verbose
load_item(xmlrpc.NOHASH, os.path.abspath(tied_file))
time.sleep(.05) # let things settle
try:
item_name = proxy.d.name(info_hash, fail_silently=True)
self.LOG.info(
"OK: Item #%s %s client.", info_hash,
'started in' if self.options.start else 'loaded into')
except xmlrpc.HashNotFound as exc:
self.fatal("Error while loading item #%s into client: %s" %
(
info_hash,
exc,
))
else:
self.LOG.warning(
"Item #%s already exists in client, --load/--start is ignored!",
info_hash)
def mainloop(self):
""" The main loop.
"""
if len(self.args) == 1 and "=urn:btih:" in self.args[0]:
# Handle magnet link
self.make_magnet_meta(self.args[0])
return
if not self.args:
self.parser.print_help()
self.parser.exit()
elif len(self.args) < 2:
self.parser.error("Expected a path and at least one announce URL, got: %s" % (' '.join(self.args),))
# Create and configure metafile factory
datapath = self.args[0].rstrip(os.sep)
metapath = datapath
if self.options.output_filename:
metapath = self.options.output_filename
if os.path.isdir(metapath):
metapath = os.path.join(metapath, os.path.basename(datapath))
if not metapath.endswith(".torrent"):
metapath += ".torrent"
torrent = metafile.Metafile(metapath)
torrent.ignore.extend(self.options.exclude)
def callback(meta):
"Callback to set label and resume data."
if self.options.cross_seed:
if self.options.cross_seed == "@entropy":
meta["info"]["entropy"] = format(random.getrandbits(self.ENTROPY_BITS),
'x').zfill(self.ENTROPY_BITS//4)
else:
meta["info"]["x_cross_seed_label"] = self.options.cross_seed
if self.options.no_cross_seed:
del meta["info"]["x_cross_seed"]
# Set specific keys?
metafile.assign_fields(meta, self.options.set)
# Create and write the metafile(s)
# TODO: make it work better with multiple trackers (hash only once), also create fast-resume file for each tracker
meta = torrent.create(datapath, self.args[1:],
progress=None if self.options.quiet else metafile.console_progress(),
root_name=self.options.root_name, private=self.options.private, no_date=self.options.no_date,
comment=self.options.comment, created_by="PyroScope %s" % self.version, callback=callback
)
# Create second metafile with fast-resume?
if self.options.hashed:
try:
metafile.add_fast_resume(meta, datapath)
except EnvironmentError as exc:
self.fatal("Error making fast-resume data (%s)" % (exc,))
raise
hashed_path = re.sub(r"\.torrent$", "", metapath) + "-resume.torrent"
self.LOG.info("Writing fast-resume metafile %r..." % (hashed_path,))
try:
bencode.bwrite(hashed_path, meta)
except EnvironmentError as exc:
self.fatal("Error writing fast-resume metafile %r (%s)" % (hashed_path, exc,))
raise
metainfo = bencode.bread(metapath)
except (KeyError, bencode.BencodeError), exc:
self.LOG.error("Bad metafile %r (%s: %s)" % (metapath, type(exc).__name__, exc))
else:
# Check metafile integrity
try:
metafile.check_meta(metainfo)
except ValueError, exc:
self.LOG.error("Metafile %r failed integrity check: %s" % (metapath, exc,))
else:
if len(self.args) > 1:
datapath = self.args[1].rstrip(os.sep)
else:
datapath = metainfo["info"]["name"]
# Check the hashes
torrent = metafile.Metafile(metapath)
torrent.check(metainfo, datapath, progress=None if self.options.quiet else metafile.console_progress())
def run(): #pragma: no cover
""" The entry point.
"""
ScriptBase.setup()
MetafileChecker().run()
if __name__ == "__main__":
run()
