Add a script to make podcast playlist in the way I like

Another crappy personal script that may or may not be useful to anyone
else
main
Buddy Sandidge 12 years ago
parent 57ffcd4f23
commit 1e5b2c44f9

@ -0,0 +1,104 @@
#!/usr/bin/env python
"""
This is a program that generates playlists based on the size of the
largest file/directory. It searches directories, getting their size
and lists the first file in the largest direcotry.
"""
import os
import sys
class File:
"""A file has a size and a name"""
def __init__(self, path):
self.name = path
self.size = os.path.getsize(self.name)
def get_size(self):
"""Get the size of the current file"""
return self.size
def get_name(self):
"""Get the name of the current file"""
return self.name
class Directory:
"""A directory is a list that can contain files and other directories."""
def __init__(self, path):
self.path = path
self.files = []
files = os.listdir(path)
files.sort()
for fil in files:
if type(fil).__name__ == "str":
self.files.append(File("%s/%s" % (path, fil)))
def get_dir_name(self):
"""Get the name of the current directory."""
return self.path
def get_dir_size(self):
"""Get the size of the current directory."""
ret = 0
for fil in self.files:
ret += fil.get_size()
return ret
def get_head_file(self):
"""Get the 'head' file in the directory."""
return self.files[0].get_name()
def num_files(self):
"""Get the number of files in the directory."""
return len(self.files)
def remove_head_file(self):
"""Remove the head file from being counted."""
self.files = self.files[1:]
def sort_dir(direct):
"""Sort the directories by size."""
return direct.get_dir_size()
def main():
"""The main script."""
try:
base_dir = sys.argv[1]
except IndexError:
print "Must enter path of directory to search. Example: %s ./foo" % \
sys.argv[0]
os.abort()
if os.path.exists(base_dir) == False:
print "Could not find path: '%s'" % base_dir
os.abort()
if base_dir[-1] == "/":
base_dir = base_dir[:-1]
base_dir_listing = os.listdir(base_dir)
directories = []
for direct in base_dir_listing:
rel_dir = "%s/%s" % (base_dir, direct)
if os.path.islink(rel_dir) == False and os.path.isdir(rel_dir):
directories.append(Directory(rel_dir))
while directories:
directories.sort(key=sort_dir)
directories.reverse()
headfile = directories[0].get_head_file()
#headfile = headfile.replace('podcasts/', 'podcast/', 1)
#print headfile
sys.stdout.write("%s\n" % headfile)
directories[0].remove_head_file()
if directories[0].num_files() == 0:
directories = directories[1:]
if __name__ == "__main__":
main()
Loading…
Cancel
Save