randomiser

#!/usr/bin/env python3

import os
import random
import sys

from FileClass import File

# Length of generated filenames, in characters
FILENAMELENGTH = 8

# To help pylint with scoping
def main():
    if 2 <= len(sys.argv):
        if len(sys.argv) == 2 and os.path.isdir(sys.argv[1]):
            root = sys.argv[1]
        else:
            print('''Usage: {0} [directory]
            
Rename all files in <directory> to have random names but keep extensions.
If not given, <directory> will be the current directory. Will always ignore
files named `{0}` and directories. All renames are checked and retried if files
are not identical.'''.format(sys.argv[0]), file=sys.stderr)
            sys.exit(1)
    else:
        root = '.'

    files = list()
    for d, _, fs in os.walk(root):
        for f in fs:
            path = os.path.join(d, f)
            if path == sys.argv[0]:
                continue
            files.append(File(path))

    for f in files:
        while True:
            try:
                randnum = random.getrandbits(FILENAMELENGTH * 4)
                randname = '{{:0{}x}}'.format(FILENAMELENGTH).format(randnum)
                f.SafeSelfCopy(randname)
            except FileExistsError:
                pass
            else:
                break # Next file

    for f in files:
        f.Remove()

if __name__ == '__main__':
    main()