untappd.py

#!/usr/bin/env python3
import json
import datetime
import argparse


def avg_abv(abv_list):
    avg_abv = sum(abv_list) / len(abv_list)
    return round(avg_abv, 2)


def uniques(data):
    uniques = {x['bid']: x for x in data}.values()
    duplicates = len(data) - len(uniques)
    noabv = [x for x in uniques if x['beer_abv'] == '0']
    nost = [x for x in uniques if x['serving_type'] == '']

    print(f'\n{len(data)} total check-ins with {len(uniques)} unique beer id\'s, '
          f'{duplicates} duplicates check-ins, ')
    print(f'{len(noabv)} had no abv data and {len(nost)} had no serving type.\n')
    return uniques


def serving_type(data):
    st = set()
    for i in data:
        if i['serving_type'] != '':
            st.add(i['serving_type'])

    longest = len(max(st, key=len))
    for i in st:
        count = 0
        for d in data:
            if d['serving_type'] == i:
                count += 1
        print(f'{i: <{longest}} : {count}')


def per_year(data):
    year = datetime.date.today().year
    for y in range(2010, year + 1):
        checkin = 0
        abv_list = []
        for d in data:
            if str(y) in d['created_at'] and float(d['beer_abv']) > 0:
                abv_list.append(float(d['beer_abv']))
                checkin += 1
        print(f'{y}      : {max(abv_list)}% max '
              f'| {avg_abv(abv_list)}% avg '
              f'| {min(abv_list)}% min '
              f'| {checkin} check-ins')


def all_years(data):
    checkin = 0
    abv_list = []
    for d in data:
        if float(d['beer_abv']) > 0:
            abv_list.append(float(d['beer_abv']))
            checkin += 1
    print(f'All Years : {max(abv_list)}% max '
          f'| {avg_abv(abv_list)}% avg | {min(abv_list)}% min '
          f'| {checkin} check-ins')


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('file', help='name of the untappd json export file')
    args = parser.parse_args()

    with open(args.file) as f:
        data = json.load(f)
        f.close

    print(f'Working on {args.file}')

    per_year(data)
    all_years(data)
    uniques(data)
    serving_type(data)