mirror/mylar
1
0
Fork 0
mirror of https://github.com/evilhero/mylar synced 2024-12-26 01:26:50 +00:00
Code Issues Releases Wiki Activity

FIX: Paramiko module error on startup if Paramiko not installed.

Browse source
This commit is contained in:
evilhero 2015-03-27 14:42:48 -04:00
parent cdc3e8a7a0
commit bacb9111ef
1 changed files with 12 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
mylar/ftpsshup.py
View file

@ -1,20 +1,11 @@
#!/usr/local/bin/python #!/usr/local/bin/python
import paramiko
import os import os
import time import time
import mylar import mylar
from mylar import logger from mylar import logger
class FastTransport(paramiko.Transport):
def __init__(self, sock):
super(FastTransport, self).__init__(sock)
self.window_size = 2147483647
self.packetizer.REKEY_BYTES = pow(2, 40)
self.packetizer.REKEY_PACKETS = pow(2, 40)
def putfile(localpath,file): #localpath=full path to .torrent (including filename), file=filename of torrent def putfile(localpath,file): #localpath=full path to .torrent (including filename), file=filename of torrent
try: try:
@ -77,6 +68,15 @@ def putfile(localpath,file): #localpath=full path to .torrent (including file
return "pass" return "pass"
def sendfiles(filelist): def sendfiles(filelist):
try:
import paramiko
except ImportError:
logger.fdebug('paramiko not found on system. Please install manually in order to use seedbox option')
logger.fdebug('get it at https://github.com/paramiko/paramiko')
logger.fdebug('to install: python setup.py install')
logger.fdebug('aborting send.')
return
fhost = mylar.TAB_HOST.find(':') fhost = mylar.TAB_HOST.find(':')
host = mylar.TAB_HOST[:fhost] host = mylar.TAB_HOST[:fhost]
port = int(mylar.TAB_HOST[fhost+1:]) port = int(mylar.TAB_HOST[fhost+1:])
@ -84,7 +84,7 @@ def sendfiles(filelist):
logger.fdebug('Destination: ' + host) logger.fdebug('Destination: ' + host)
logger.fdebug('Using SSH port : ' + str(port)) logger.fdebug('Using SSH port : ' + str(port))
transport = FastTransport((host, port)) transport = paramiko.Transport((host, port))
password = mylar.TAB_PASS password = mylar.TAB_PASS
username = mylar.TAB_USER username = mylar.TAB_USER
@ -92,7 +92,6 @@ def sendfiles(filelist):
sftp = paramiko.SFTPClient.from_transport(transport) sftp = paramiko.SFTPClient.from_transport(transport)
import sys
remotepath = mylar.TAB_DIRECTORY remotepath = mylar.TAB_DIRECTORY
logger.fdebug('remote path set to ' + remotepath) logger.fdebug('remote path set to ' + remotepath)
@ -151,7 +150,7 @@ def sendtohome(sftp, remotepath, filelist, transport):
sftp.close() sftp.close()
transport.close() transport.close()
#reload the transport here cause it locked up previously. #reload the transport here cause it locked up previously.
transport = FastTransport((host, port)) transport = paramiko.Transport((host, port))
transport.connect(username=mylar.TAB_USER, password=mylar.TAB_PASS) transport.connect(username=mylar.TAB_USER, password=mylar.TAB_PASS)
sftp = paramiko.SFTPClient.from_transport(transport) sftp = paramiko.SFTPClient.from_transport(transport)
count+=1 count+=1
@ -180,7 +179,7 @@ def sendtohome(sftp, remotepath, filelist, transport):
sftp.close() sftp.close()
transport.close() transport.close()
#reload the transport here cause it locked up previously. #reload the transport here cause it locked up previously.
transport = FastTransport((host, port)) transport = paramiko.Transport((host, port))
transport.connect(username=mylar.TAB_USER, password=mylar.TAB_PASS) transport.connect(username=mylar.TAB_USER, password=mylar.TAB_PASS)
sftp = paramiko.SFTPClient.from_transport(transport) sftp = paramiko.SFTPClient.from_transport(transport)
count+=1 count+=1