def upload_file(image, existing_file=False):
config = lookitconfig.LookitConfig()
proto = config.get("Upload", "type")
if proto == "SSH":
liblookit.show_notification("Lookit", "Uploading image to {0}...".format(config.get("Upload", "hostname")))
success, data = upload_file_sftp(
image,
config.get("Upload", "hostname"),
int(config.get("Upload", "port")),
config.get("Upload", "username"),
config.get("Upload", "password"),
config.get("Upload", "directory"),
config.get("Upload", "url"),
)
elif proto == "FTP":
liblookit.show_notification("Lookit", "Uploading image to {0}...".format(config.get("Upload", "hostname")))
success, data = upload_file_ftp(
image,
config.get("Upload", "hostname"),
int(config.get("Upload", "port")),
config.get("Upload", "username"),
config.get("Upload", "password"),
config.get("Upload", "directory"),
config.get("Upload", "url"),
)
elif proto == "Omploader":
liblookit.show_notification("Lookit", "Uploading image to Omploader...")
success, data = upload_file_omploader(image)
try:
f = open(liblookit.LOG_FILE, "ab")
f.write(time.ctime() + " Uploaded screenshot to Omploader: " + data["original_image"] + "\n")
except IOError, e:
pass
finally:
f = open(liblookit.LOG_FILE, 'ab')
f.write(time.ctime() + ' Uploaded screenshot to Imgur: ' + data['original_image'] + '\n')
except IOError, e:
pass
finally:
f.close()
elif proto == 'CloudApp':
success, data = upload_file_cloud(image,
config.get('Upload', 'username'),
config.get('Upload', 'password'))
else:
success = False
data = "Error: no such protocol: {0}".format(proto)
if not success:
liblookit.show_notification('Lookit', 'Error: ' + data)
return
if data:
url = data['original_image']
else:
url = urlparse.urljoin(config.get('Upload', 'url'),
os.path.basename(image))
if config.getboolean('General', 'shortenurl') and proto != None:
url = urllib.urlopen('http://is.gd/api.php?longurl={0}'
.format(url)).readline()
if not existing_file:
if config.getboolean('General', 'trash'):
os.remove(os.path.abspath(image))
else:
f.write(time.ctime() + ' Uploaded screenshot to Imgur: ' +
data['original_image'] + '\n')
except IOError, e:
pass
finally:
f.close()
elif proto == 'CloudApp':
success, data = upload_file_cloud(image,
config.get('Upload', 'username'),
config.get('Upload', 'password'))
else:
success = False
data = "Error: no such protocol: {0}".format(proto)
if not success:
liblookit.show_notification('Lookit', 'Error: ' + data)
return
if data:
url = data['original_image']
else:
url = urlparse.urljoin(config.get('Upload', 'url'),
os.path.basename(image))
if config.getboolean('General', 'shortenurl') and proto != None:
url = urllib.urlopen(
'http://is.gd/api.php?longurl={0}'.format(url)).readline()
if not existing_file:
if config.getboolean('General', 'trash'):
os.remove(os.path.abspath(image))
else:
config.get("Upload", "password"),
config.get("Upload", "directory"),
config.get("Upload", "url"),
)
elif proto == "Omploader":
liblookit.show_notification("Lookit", "Uploading image to Omploader...")
success, data = upload_file_omploader(image)
try:
f = open(liblookit.LOG_FILE, "ab")
f.write(time.ctime() + " Uploaded screenshot to Omploader: " + data["original_image"] + "\n")
except IOError, e:
pass
finally:
f.close()
elif proto == "Imgur":
liblookit.show_notification("Lookit", "Uploading image to Imgur...")
success, data = upload_file_imgur(image)
try:
f = open(liblookit.LOG_FILE, "ab")
f.write(time.ctime() + " Uploaded screenshot to Imgur: " + data["original_image"] + "\n")
f.write("Delete url: " + data["delete_page"] + "\n")
except IOError, e:
pass
finally:
f.close()
elif proto == "CloudApp":
liblookit.show_notification("Lookit", "Uploading image to CloudApp...")
success, data = upload_file_cloud(image, config.get("Upload", "username"), config.get("Upload", "password"))
elif proto == "None":
success = True
data = False
