beats.py

"""Module to calculate beats(Swatch Internet Time).
"""

from datetime import datetime


def internettime(hours, minutes, seconds, tzone, centibeats=False):
    """Returns time(Swatch Internet Time) in beats.
    """
    itime = ((seconds + (minutes * 60) + ((hours + tzone + 1) * 3600)) / 86.4) % 1000
    beats = int(itime)
    cbeats = str(itime).split(".")[1][0:3]
    if centibeats:
        return "@{0}.{1}".format(beats, cbeats)
    return "@{0}".format(beats)


def now(centibeats=False):
    """Gets the current time in beats.
    """
    utc_now = datetime.utcnow()
    beats = internettime(
        utc_now.hour, utc_now.minute, utc_now.second, 0, centibeats
    )
    return beats


def main():
    """The main function.
    """
    parser = argparse.ArgumentParser(description="")
    parser.add_argument(
        "-c",
        "--centibeats",
        action="store_true",
        help="Display centibeats",
        default=False,
    )
    parser.add_argument("-t", "--time", help="Time to convert (HH:MM:SS)")
    parser.add_argument(
        "-z",
        "--timezone",
        metavar="TZ",
        type=float,
        help="Timezone in hours, default: UTC+01",
        default=1,
    )
    parser.add_argument("-v", "--version", action="store_true")
    args = parser.parse_args()

    if args.version:
        print(".beat v{0}".format(pkg_resources.get_distribution("beat").version))
        return
    if args.time:
        hours, minutes, seconds = args.time.split(":")
        print(
            internettime(
                int(hours), int(minutes), int(seconds), args.timezone, args.centibeats
            )
        )
    else:
        beats = now(args.centibeats)
        print(beats)


if __name__ == "__main__":
	main()