forked from 0x2620/pandora
73 lines
1.9 KiB
Python
73 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
# vi:si:et:sw=4:sts=4:ts=4
|
|
from __future__ import with_statement
|
|
import os
|
|
import time
|
|
import base64
|
|
from subprocess import Popen
|
|
|
|
from django.conf import settings
|
|
import ox.torrent
|
|
import transmissionrpc
|
|
|
|
def connect():
|
|
return transmissionrpc.Client(settings.TRANSMISSON_HOST,
|
|
port=settings.TRANSMISSON_PORT,
|
|
user=settings.TRANSMISSON_USER,
|
|
password=settings.TRANSMISSON_PASSWORD)
|
|
|
|
def remove(info_hash):
|
|
if settings.DEBUG:
|
|
print 'remove', info_hash
|
|
if info_hash:
|
|
try:
|
|
tc = connect()
|
|
tc.remove(info_hash.lower())
|
|
except:
|
|
if DEBUG:
|
|
import traceback
|
|
traceback.print_exc()
|
|
|
|
def add(torrent_file):
|
|
download_dir = os.path.dirname(torrent_file)
|
|
with open(torrent_file) as f:
|
|
torrent_data = base64.b64encode(f.read())
|
|
info_hash = ox.torrent.getInfoHash(torrent_file)
|
|
try:
|
|
tc = connect()
|
|
if not is_seeding(info_hash):
|
|
tc.add(torrent_data, download_dir=download_dir)
|
|
except:
|
|
if settings.DEBUG:
|
|
import traceback
|
|
traceback.print_exc()
|
|
|
|
def is_seeding(info_hash):
|
|
info_hash = info_hash.lower()
|
|
try:
|
|
tc = connect()
|
|
torrents = tc.info(info_hash)
|
|
except:
|
|
torrents = False
|
|
if settings.DEBUG:
|
|
import traceback
|
|
traceback.print_exc()
|
|
if torrents:
|
|
return True
|
|
return False
|
|
|
|
def start_daemon():
|
|
try:
|
|
tc = connect()
|
|
except:
|
|
Popen(['transmission-daemon',
|
|
'-a', '127.0.0.1',
|
|
'-r', '127.0.0.1',
|
|
'-p', str(settings.TRANSMISSON_PORT),
|
|
'--auth',
|
|
'-u', settings.TRANSMISSON_USER,
|
|
'-v', settings.TRANSMISSON_PASSWORD,
|
|
'-w', settings.MEDIA_ROOT,
|
|
])
|
|
time.sleep(1)
|
|
|