forked from premiere/premiere-libtorrent
54 lines
1.1 KiB
Python
54 lines
1.1 KiB
Python
#!/bin/python
|
|
|
|
import sys
|
|
import os
|
|
import libtorrent
|
|
|
|
if len(sys.argv) < 3:
|
|
print 'usage make_torrent.py file tracker-url'
|
|
sys.exit(1)
|
|
|
|
input = os.path.abspath(sys.argv[1])
|
|
|
|
fs = libtorrent.file_storage()
|
|
|
|
#def predicate(f):
|
|
# print f
|
|
# return True
|
|
#libtorrent.add_files(fs, input, predicate)
|
|
|
|
parent_input = os.path.split(input)[0]
|
|
|
|
for root, dirs, files in os.walk(input):
|
|
# skip directories starting with .
|
|
if os.path.split(root)[1][0] == '.': continue
|
|
|
|
for f in files:
|
|
# skip files starting with .
|
|
if f[0] == '.': continue
|
|
|
|
# skip thumbs.db on windows
|
|
if f == 'Thumbs.db': continue
|
|
|
|
fname = os.path.join(root[len(parent_input)+1:], f)
|
|
size = os.path.getsize(os.path.join(parent_input, fname))
|
|
print '%10d kiB %s' % (size / 1024, fname)
|
|
fs.add_file(fname, size);
|
|
|
|
if fs.num_files() == 0:
|
|
print 'no files added'
|
|
sys.exit(1)
|
|
|
|
t = libtorrent.create_torrent(fs, 0, 4 * 1024 * 1024)
|
|
|
|
t.add_tracker(sys.argv[2])
|
|
t.set_creator('libtorrent %s' % libtorrent.version)
|
|
|
|
libtorrent.set_piece_hashes(t, parent_input, lambda x: sys.stderr.write('.'))
|
|
sys.stderr.write('\n')
|
|
|
|
f = open('out.torrent', 'wb+')
|
|
print >>f, libtorrent.bencode(t.generate())
|
|
f.close()
|
|
|