47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
#!/usr/bin/env python
|
|
|
|
import sys
|
|
from subprocess import call
|
|
import os
|
|
from pysftp import Connection
|
|
from tkinter import Tk, Button
|
|
from string import ascii_letters, digits
|
|
from random import choices
|
|
import config
|
|
|
|
character_pool = ascii_letters + digits
|
|
|
|
def generate_filename(prefix, length):
|
|
return prefix + ''.join(choices(character_pool, k=length)) + '.png'
|
|
|
|
|
|
filename = generate_filename(config.prefix, config.length)
|
|
|
|
with Connection(config.sftp_address, username=config.username, password=config.password) as conn:
|
|
with conn.cd(config.remote_directory):
|
|
# there's potential for an endless loop here...
|
|
while conn.exists(filename):
|
|
filename = generate_filename(config.prefix, config.length)
|
|
|
|
filepath = os.path.join(config.local_directory, filename)
|
|
|
|
call(["escrotum", "{}".format(filepath), "-s"])
|
|
|
|
conn.put(filepath)
|
|
|
|
link = config.link_template.format(filename)
|
|
|
|
# copy link to clipboard
|
|
# https://stackoverflow.com/questions/579687/how-do-i-copy-a-string-to-the-clipboard-on-windows-using-python#4203897
|
|
print(link)
|
|
r = Tk()
|
|
r.clipboard_clear()
|
|
r.clipboard_append(link)
|
|
r.after(2000, sys.exit)
|
|
# also show a little button that does nothing.
|
|
# Well, it informs you that your link is ready, so that's something, I guess
|
|
rButton = Button(r, text=f"{link}", font=("Verdana", 12), bg="black", command=sys.exit)
|
|
rButton.pack()
|
|
r.geometry('400x50+700+500')
|
|
r.mainloop()
|