Permutation of numbers with repeating.

perm_repeat.py

#!/usr/bin/python

# John (Troon) Ombagi
# jayombagi@gmail.com 

# PR(n, k) = n^k   ----> Permutation with repetition.

import itertools
import sys


def PR(n, k):

    for permutation in itertools.product(xrange(n), repeat=k):
        print(''.join(map(str, permutation)))


if len(sys.argv) == 3:

    n = int(sys.argv[1])
    k = int(sys.argv[2])

    if n >= k:
        PR(n, k)

    else:
        print("[-] n should be greater or equal to k.")
        print("    Example : perm_repeat.py 10 4")
else:
    print("[-] Please supply two arguments : n and k.")
    print("    Example : perm_repeat.py 10 5")