kageru/pyshare
1
0
Fork 0
Code Issues 2 Pull Requests Releases Wiki Activity

more refactoring for more modes

Browse Source
This commit is contained in:
kageru 2017-10-25 14:22:58 +02:00
parent edf2544c4a
commit 6ce497b653
1 changed files with 28 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
pyshare.py
View File

@ -10,11 +10,13 @@ from random import choices
import config import config
character_pool = ascii_letters + digits character_pool = ascii_letters + digits
tk = Tk()
def generate_filename(prefix, length, ext): def generate_filename(prefix, length, ext):
return prefix + ''.join(choices(character_pool, k=length)) + '.' + ext return prefix + ''.join(choices(character_pool, k=length)) + '.' + ext
def find_filename(prefix, length, ext, conn): def find_filename(prefix, length, ext, conn):
filename = generate_filename(prefix, length, ext) filename = generate_filename(prefix, length, ext)
i = 0 i = 0
@ -22,12 +24,12 @@ def find_filename(prefix, length, ext, conn):
filename = generate_filename(prefix, length, ext) filename = generate_filename(prefix, length, ext)
i += 1 i += 1
if i > 1000: if i > 1000:
# using recursion here feels... questionable at best, but it's faster than using % every loop # completely, definitely, totally justified recursion... yay?
find_filename(prefix, length + 1, ext, conn) find_filename(prefix, length + 1, ext, conn)
def upload_local_file(path: str, def upload_local_file(path: str, conn: Connection) -> Exception:
conn: Connection) -> Exception: # does this even return an exception? probably not. does it matter? definitely not # does this even return an exception? probably not. does it matter? definitely not
raise NotImplementedError('soon(tm)') raise NotImplementedError('soon(tm)')
@ -54,23 +56,27 @@ def curl_upload(filename):
if config.custom_curl_command is not None: if config.custom_curl_command is not None:
return call(config.custom_curl_command) return call(config.custom_curl_command)
else: else:
return call(f'curl -k -F"file=@{filename}" -F"name={config.username}" -F"passwd={config.password}" {config.curl_target}') return call(
f'curl -k -F"file=@{filename}" -F"name={config.username}" -F"passwd={config.password}" {config.curl_target}')
def set_clipboard(text):
tk.clipboard_clear()
tk.clipboard_append(text)
def get_clipboard():
result = tk.selection_get(selection="CLIPBOARD")
def notify_user(url): def notify_user(url):
print(url) print(url)
# copy link to clipboard
# https://stackoverflow.com/questions/579687/how-do-i-copy-a-string-to-the-clipboard-on-windows-using-python#4203897
r = Tk()
r.clipboard_clear()
r.clipboard_append(url)
r.after(2000, sys.exit)
# also show a little button that does nothing. # also show a little button that does nothing.
# Well, it informs you that your link is ready, so that's something, I guess # Well, it informs you that your link is ready, so that's something, I guess
rButton = Button(r, text=f"{url}", font=("Verdana", 12), bg="black", command=sys.exit) rButton = Button(tk, text=f"{url}", font=("Verdana", 12), bg="black", command=sys.exit)
rButton.pack() rButton.pack()
r.geometry('400x50+700+500') tk.geometry('400x50+700+500')
r.mainloop() tk.mainloop()
if __name__ == '__main__': if __name__ == '__main__':
@ -81,6 +87,14 @@ if __name__ == '__main__':
else: else:
mode = 'screenshot' mode = 'screenshot'
ext = 'png' ext = 'png'
if mode == 'screenshot':
pass
elif mode=='file':
pass
elif mode=='text':
pass
"""
if config.uploader in ['ftp', 'sftp']: if config.uploader in ['ftp', 'sftp']:
if mode != 'screenshot' and '.' in file: if mode != 'screenshot' and '.' in file:
ext = '.' + file.rsplit('.', 1)[1] ext = '.' + file.rsplit('.', 1)[1]
@ -99,3 +113,4 @@ if __name__ == '__main__':
sys.exit(-1) sys.exit(-1)
url = config.url_template.format(filename) url = config.url_template.format(filename)
notify_user(url) notify_user(url)
"""