Script for uploading files to transfer.sh (https://transfer.sh/)

transfer_sh_uploader.py

#!/usr/bin/env python3

"""
Author: Arsen Losenko (github: arsenlosenko)
Short description: Script that uploads files in entered directory as an archive to transfer.sh, for easy downloading.
"""

import os
import sys
import zipfile
import requests
import datetime


def create_zip():
    path_to_folder = input("Please enter directory: \n")
    if os.path.isdir(path_to_folder):
        os.chdir(path_to_folder)
        zip_name = 'files_archive_{}.zip'.format(str(datetime.datetime.now())[5:16].replace(' ', "_"))
        files = os.listdir()
        print("Creating zipfile from files in...", path_to_folder)
        with zipfile.ZipFile(zip_name, 'w') as zip:
            for img in files:
                zip.write(img)
                print("Added file: ", img)

        return zip_name
    else:
        print("Please enter a valid absolute path to directory")
        sys.exit()


def send_zip():
    zip_file = create_zip()
    print("\nSending zipfile: ", zip_file)
    url = 'https://transfer.sh/'
    file = {'file': open(zip_file, 'rb')}
    response = requests.post(url, files=file)
    download_link = response.content.decode('utf-8')
    print("Link to download zipfile:\n", download_link)
    confirm_removal()


def confirm_removal():
    confirmation = input('\nDelete files in the directory?(y/n, Y/N): ')
    if (confirmation == 'y') or (confirmation == 'Y'):
        clean_up()
    elif (confirmation == 'n') or (confirmation == 'N'):
        print("OK, files will be there")
        sys.exit()
    else:
        print("Please enter a valid answer (y/n, Y/N)")
        confirm_removal()


def clean_up():
    print('\nCleaning up...')
    files = os.listdir()
    for f in files:
        os.remove(f)
        print("Removed file: ", f)


if __name__ == '__main__':
    send_zip()