main.py

import requests
import os
import argparse
from dotenv import load_dotenv

ENTRY_URL = 'https://api-ssl.bitly.com/v4/bitlinks'
DESCRIPTION_TEXT = 'Сокращение ссылок с помощью сервиса bit.ly и вывод статистики по переходам'
HELP_TEXT = 'Введите ссылку для сокращения или bitly-ссылку для просмотра количества переходов.'


def make_headers(token):
    headers = {
        "Authorization": "Bearer {}".format(token)
    }
    return headers


def make_payload(url):
    payload = {
        "long_url": url
    }
    return payload


def shorten_link(token, url):
    response = requests.post(ENTRY_URL, json=make_payload(url), headers=make_headers(token))
    response.raise_for_status()
    short_link_data = response.json()
    return short_link_data['link']


def count_clicks(token, short_link):
    response = requests.get(ENTRY_URL + '/{}/clicks/summary'.format(short_link), headers=make_headers(token))
    response.raise_for_status()
    clicks_count_data = response.json()
    return clicks_count_data['total_clicks']


def remove_http_symbols(short_link):
    return short_link[short_link.find('bit.ly'):]


if __name__ == "__main__":
    load_dotenv()
    parser = argparse.ArgumentParser(description=DESCRIPTION_TEXT)
    parser.add_argument('url', help=HELP_TEXT)
    args = parser.parse_args()

    try:
        url = args.url
        if url.startswith('http://bit.ly') or url.startswith('https://bit.ly'):
            print('Количество переходов:', count_clicks(os.getenv("BITLY_TOKEN"), remove_http_symbols(url)))
        else:
            print('Битлинк', shorten_link(os.getenv("BITLY_TOKEN"), url))
    except requests.exceptions.HTTPError:
        print('Ошибка: Неверная ссылка.')