comit with last rembo10 commit

headphones/encode.py

@@ -1,11 +1,15 @@
 import os
 import headphones
-import argparse
 import shutil
 import time
 
 from subprocess import call
 
+try:
+    import argparse
+except ImportError:
+    import lib.argparse as argparse
+
 def encode(albumPath):
 
 	tempDirEncode=os.path.join(albumPath,"temp")
@@ -21,7 +25,7 @@ def encode(albumPath):
 		
 	for r,d,f in os.walk(albumPath):
 		for music in f:
-			if any(music.endswith('.' + x) for x in headphones.MEDIA_FORMATS):
+			if any(music.endswith('.' + x) for x in ["flac", "m4a", "wav"]):
 				musicFiles.append(os.path.join(r, music))
 				musicTempFiles.append(os.path.join(tempDirEncode, music))