Shorten urls using bitly and python

gistfile1.sh

function task() {
    shorten-urls task "$@"
}

shorten

#!/usr/bin/env python

import requests
import click

ENDPOINT = 'https://api-ssl.bitly.com/v3/shorten'

@click.command()
@click.option('--api-key', envvar='BITLY_API_KEY')
@click.argument('uri')
def main(api_key, uri):
    payload = {
        'access_token': api_key,
        'uri': uri,
        'format': 'json',
    }
    response = requests.get(ENDPOINT, params=payload)
    result = response.json()
    print(result['data']['url'])


if __name__ == '__main__':
    main()

shorten-urls

#!/usr/bin/env python2

import re
import sys
from subprocess import check_call, check_output


def shorten(url):
    if not re.match(r'(https?://\S+)', url):
        return url
    print("Shortening url")
    return check_output(['shorten', url])


def main():
    argv = list(map(shorten, sys.argv[1:]))
    exit(check_call(argv, shell=False))


if __name__ == '__main__':
    main()