cmd/kbps.py

'''
Find time, filesize, or bitrate, given two of the three.

For example:

kbps.py --time 1:00:00 --size 2g
kbps.py --time 1:00:00 --kbps 4660
kbps.py --size 2g --kpbps 4660
'''
import argparse
import sys

from voussoirkit import bytestring

def hms_s(hms):
    hms = hms.split(':')
    seconds = 0
    if len(hms) == 3:
        seconds += int(hms[0])*3600
        hms.pop(0)
    if len(hms) == 2:
        seconds += int(hms[0])*60
        hms.pop(0)
    if len(hms) == 1:
        seconds += int(hms[0])
    return seconds

def s_hms(s):
    (minutes, seconds) = divmod(s, 60)
    (hours, minutes) = divmod(minutes, 60)
    return '%02d:%02d:%02d' % (hours, minutes, seconds)

def kbps(time=None, size=None, kbps=None):
    if [time, size, kbps].count(None) != 1:
        raise ValueError('Incorrect number of unknowns')

    if size is None:
        seconds = hms_s(time)
        kibs = int(kbps) / 8
        size = kibs * 1024
        size *= seconds
        out = bytestring.bytestring(size)
        return out

    if time is None:
        size = bytestring.parsebytes(size)
        kilobits = size / 128
        time = kilobits / int(kbps)
        return s_hms(time)

    if kbps is None:
        seconds = hms_s(time)
        size = bytestring.parsebytes(size)
        kibs = size / 1024
        kilobits = kibs * 8
        kbps = kilobits / seconds
        kbps = '%d kbps' % int(round(kbps))
        return kbps

def example_argparse(args):
    print(kbps(time=args.time, size=args.size, kbps=args.kbps))

def main(argv):
    parser = argparse.ArgumentParser()

    parser.add_argument('-t', '--time', dest='time', default=None)
    parser.add_argument('-s', '--size', dest='size', default=None)
    parser.add_argument('-k', '--kbps', dest='kbps', default=None)
    parser.set_defaults(func=example_argparse)

    args = parser.parse_args(argv)
    return args.func(args)

if __name__ == '__main__':
    raise SystemExit(main(sys.argv[1:]))