util.py

import re
import sys
import time
import datetime

VERBOSE=0

def clean_name(name):
	"Clean up an mp3 name for comparison"
	if name is None:
		return None
	name = re.sub(r"\(.*\)","",name)
	name = re.sub(r"\[.*\]","",name)
	name = re.sub(r"\{.*\}","",name)
	name = re.sub(r"[^A-Za-z0-9]","",name)
	return name.lower()

def comp_name(n1,n2):
	"Copmare two names for equality, after applying cleanups"
	return clean_name(n1) == clean_name(n2)

def _combinations(func,doneargs,todoargs):
	if todoargs==():
		yield func(*doneargs)
	elif type(todoargs[0]) not in [type(()),type([])]:
		for ret in _combinations(func,doneargs+(todoargs[0],),todoargs[1:]):
			yield ret
	else:
		for arg in todoargs[0]:
			for ret in _combinations(func,doneargs+(arg,),todoargs[1:]):
				yield ret
	return
		

def combinations(func,*args):
	"""This function takes a function, and some arguments, some of which may be
collections.  For each sequence, the function is call combinatorially"""
	for ret in _combinations(func,(),args):
		yield ret

def update_progress(msg):
	"Display some progress"
	if type(msg) == type(''):
		msg = msg.decode('utf8','ignore')
	if VERBOSE:
		print time.strftime("%H:%M:%S"),msg.encode("ascii","ignore")
	else:
		sys.stdout.write(time.strftime("%H:%M:%S ")+msg.encode("ascii","ignore")+"\x1b[K\r")
		sys.stdout.flush()

def update_verbose_progress(msg):
	"Display some progress"
	if type(msg) == type(''):
		msg = msg.decode('utf8','ignore')
	if VERBOSE:
		print msg.encode("ascii","ignore")
	else:
		sys.stdout.write(msg.encode("ascii","ignore")+"\x1b[K\n")
		sys.stdout.flush()

def output_list(l):
	"Give na list of integers, return a string with ranges collapsed"
	if not l:
		return "[]"
	l.sort()
	ret=[]
	start=l[0]
	end=l[0]
	for i in l[1:]:
		if end+1==i:
			end=i
			continue
		if start!=end:
			ret.append("%d-%d" % (start,end))
		else:
			ret.append("%d" % start)
		start=i
		end=i
	if start!=end:
		ret.append("%d-%d" % (start,end))
	else:
		ret.append("%d" % start)
	return "[%s]" % (",".join(ret))

def list_difference(src,remove):
	res=src[:]
	for i in remove:
		if i in res:
			res.remove(i)
	return res

report_entries = []
def report(message):
        ts = datetime.datetime.now().ctime()
        if (type(message) == type([])):
                for m in message:
                        report(m)
                return
        report_entries.append("%s: %s" % (ts, message))
        print message

def write_report(reportpath):
	global report_entries
	f = open(reportpath, "w")
	try:
	        for r in report_entries:
	                f.write(r.encode("utf8"))
	                f.write("\n")
	finally:
		f.close()
		report_entries = []