#!/usr/bin/env python3

import os
import random
import subprocess
import argparse

parser = argparse.ArgumentParser(description='Število Pujs Pep, ki jih želiš zaigrat')
parser.add_argument('stevilo', metavar='N', type=int, nargs='+',
                    help='Število filmčkov za zaigrat')
args = parser.parse_args()


def main(num):
    list_of_files = [x for x in os.listdir(os.getcwd()) if x.endswith('avi')]
    videos = random.sample(list_of_files, args.stevilo[0])
    # filename = list_of_files[random.randint(0,len(list_of_files)-1)]

    for vid in videos:
        vid = subprocess.run(['omxplayer', vid])

if __name__ == "__main__":
    main(args.stevilo[0])