untitled

import fileinput
import pyperclip
import sys, getopt, os
from pastebin_connection import paste_create, login_user


def add_paste(content, title, syntax, expiry, succ, error):
    if content:
        result = paste_create(content, title, syntax, expiry)
        if result['success']:
            succ.append(result['data'])
        else:
            error.append(result['data'])
        paste_content = ""

def handle_input(args, title, syntax, expiry, clip):
    paste_content = ""
    succ = []
    error = []

    for line in fileinput.input(args):
        if fileinput.isfirstline():
            add_paste(paste_content, title, syntax, expiry, succ, error)

        paste_content += line

    add_paste(paste_content, title, syntax, expiry, succ, error)
    if succ:
        print("Succesful pastes")
        for url in succ:
            print(url)

        print()

        if clip and len(succ) == 1:
            pyperclip.copy(succ[0])

    if error:
        print("Error")
        for url in error:
            print(url)

def print_help():
    print()
    print("Use like cat program to upload to pastebin.")
    print("[-u|--username] [-p|--password] - for username and assword")
    print()
    pass

def main(argv):
    username = ''
    password = ''
    syntax = ''
    expiry = ''
    clip = False
    silent = False
    try:
        opts, args = getopt.gnu_getopt(argv, 'hu:p:t:s:e:cs', ["username=", "password=", "title=", "syntax=", "expiry="])
    except getopt.GetoptError:
        print_help()
        sys.exit(2)

    for opt, arg in opts:
        if opt == '-h':
            print_help()
            sys.exit()
        elif opt in ('-u', '--username'):
            username = arg
        elif opt in ('-p', '--password'):
            password = arg
        elif opt in ('-t', '--title'):
            title = arg
        elif opt in ('-s', '--syntax'):
            syntax = arg
        elif opt in ('-e', '--expiry'):
            expiry = arg
        elif opt == '-c':
            clip = True
        elif opt == '-s':
            silent = True

    if silent:
        f = open(os.devnull, 'w')
        sys.stdout = f

    if username and password:
        res = login_user(username, password)
        if res['success']:
            print('Successful login')
        else:
            print(res['data'])

    handle_input(args, title, syntax, expiry, clip)

if __name__ == "__main__":
    main(sys.argv[1:])