Skip to content
Advertisement

print/list only 5 entries from OS.Walk in python

My Goal – To list only 5 entries when using OS walk. So far I have only been able to get a list of everything that I find using OS.Walk or only list one entry. (By using the return function)

My code: import os import subprocess

def playmusic(name):
    for root, dirs, files in os.walk('E:\', followlinks=True):
        for file in files:
            if name in file:
                vlc='C:/Program Files/VideoLAN/VLC/vlc.exe'
                music=str(os.path.join(root,file))
                print(music)
                #subprocess.Popen([vlc, music])
                #return
    print("Finish")
    input()
try:
    s=raw_input("name: ")
    playmusic(s)
except Exception as e:
    print(e)
    print("Error")

The Results:

=== RESTART: C:UsersVGMPC2Documentstesting scriptssearch and print.py ===
name: test
E:NesJordan Vs Bird3 Point Contest.mp4
E:NesJordan Vs BirdSlam Dunk Contest.mp4
E:playlistAction&Adventuretest.xspf
E:playlistschedule test 2.xspf
E:SnesLufia IIThe Greatest Thieves.mp4
E:Symbolic playlistsNintendo Generation3 Point Contest.mp4
E:Symbolic playlistsNintendo GenerationSlam Dunk Contest.mp4
Finish

My code

If there is any way to only show 5 instead of the whole list that would be great! I tried using len() but I was having trouble figuring out how to use it with the os walk search. I would say the biggest thing is the music=str(os.path.join(root,file)) as that does the search I believe.

Any ideas would be appreciated. Thank you for your time,

Advertisement

Answer

Try this:

def playmusic(name):
    for root, dirs, files in os.walk('E:\', followlinks=True):
        for file in files[0:5]: #You show the first five only
            if name in file:
                vlc='C:/Program Files/VideoLAN/VLC/vlc.exe'
                music=str(os.path.join(root,file))
                print(music)
                #subprocess.Popen([vlc, music])
                #return
    print("Finish")
    input()
try:
    s=raw_input("name: ")
    playmusic(s)
except Exception as e:
    print(e)
    print("Error")

User contributions licensed under: CC BY-SA
4 People found this is helpful
Advertisement