happy_number.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os


def happy_number(val):
    nums = []
    nums.append(val)
    while True:
        new_num = sum([pow(int(c), 2) for c in str(val)])
        if new_num == 1:
            return True
        elif new_num in nums:
            return False
        nums.append(new_num)
        val = new_num


def main():
    print(happy_number(int(os.sys.argv[1])))

if __name__ == '__main__':
    main()