diff --git a/dotfiles/i3/scripts/crypto b/dotfiles/i3/scripts/crypto index f4154a1..29f7671 100755 --- a/dotfiles/i3/scripts/crypto +++ b/dotfiles/i3/scripts/crypto @@ -4,22 +4,32 @@ import argparse import requests -parser = argparse.ArgumentParser() -parser.add_argument("coin") -parser.add_argument("-b", "--base", default="USD") -parser.add_argument("-r", "--round", type=int, default=2) -args = parser.parse_args() -coin = args.coin.upper() -base = args.base.upper() -round_nb = args.round +def parse_arguments(): + parser = argparse.ArgumentParser() + parser.add_argument("coin") + parser.add_argument("-b", "--base", default="USD") + parser.add_argument("-r", "--round", type=int, default=2) + args = parser.parse_args() -req = requests.get( - "https://api.coinranking.com/v1/public/coins?base={}&symbols={}".format(base, coin) -) -req_j = req.json() -data = req_j["data"] -base_sign = data["base"]["sign"] -coin_data = data["coins"][0] -price = float(coin_data["price"]) + return args -print("{} = {}{}".format(coin, base_sign, round(price, round_nb))) +def compute(args): + coin = args.coin.upper() + base = args.base.upper() + round_nb = args.round + + req = requests.get( + "https://api.coinranking.com/v1/public/coins?base={}&symbols={}".format( + base, coin + ) + ) + req_j = req.json() + data = req_j["data"] + base_sign = data["base"]["sign"] + coin_data = data["coins"][0] + price = float(coin_data["price"]) + + return "{} = {}{}".format(coin, base_sign, round(price, round_nb)) + + +print(compute(parse_arguments()))