mirror of
https://github.com/correl/dejavu.git
synced 2024-11-23 19:19:53 +00:00
98 lines
2.3 KiB
Python
Executable file
98 lines
2.3 KiB
Python
Executable file
#!/usr/bin/python
|
|
|
|
import sys
|
|
import json
|
|
import warnings
|
|
|
|
from dejavu import Dejavu
|
|
from dejavu.recognize import FileRecognizer
|
|
from dejavu.recognize import MicrophoneRecognizer
|
|
from dejavu.recognize import FileRecognizer
|
|
|
|
warnings.filterwarnings("ignore")
|
|
|
|
def init():
|
|
# load config from a JSON file (or anything outputting a python dictionary)
|
|
with open("dejavu.cnf") as f:
|
|
config = json.load(f)
|
|
|
|
# create a Dejavu instance
|
|
return Dejavu(config)
|
|
|
|
def showHelp():
|
|
print ""
|
|
print "------------------------------------------------"
|
|
print "DejaVu audio fingerprinting and recognition tool"
|
|
print "------------------------------------------------"
|
|
print ""
|
|
print "Usage: dejavu.py [command] [arguments]"
|
|
print ""
|
|
print "Available commands:"
|
|
print ""
|
|
print " Fingerprint a file"
|
|
print " dejavu.py fingerprint /path/to/file.extension"
|
|
print ""
|
|
print " Fingerprint all files in a directory"
|
|
print " dejavu.py fingerprint /path/to/directory extension"
|
|
print ""
|
|
print " Recognize what is playing through the microphone"
|
|
print " dejavu.py recognize mic number_of_seconds"
|
|
print ""
|
|
print " Recognize a file by listening to it"
|
|
print " dejavu.py recognize file /path/to/file"
|
|
print ""
|
|
print " Display this help screen"
|
|
print " dejavu.py help"
|
|
print ""
|
|
exit
|
|
|
|
if len(sys.argv) > 1:
|
|
command = sys.argv[1]
|
|
else:
|
|
showHelp()
|
|
|
|
if command == 'fingerprint': # Fingerprint all files in a directory
|
|
|
|
djv = init()
|
|
|
|
|
|
if len(sys.argv) == 4:
|
|
|
|
directory = sys.argv[2]
|
|
extension = sys.argv[3]
|
|
print "Fingerprinting all .%s files in the %s directory" % (extension, directory)
|
|
|
|
djv.fingerprint_directory(directory, ["." + extension], 4)
|
|
|
|
else:
|
|
|
|
filepath = sys.argv[2]
|
|
djv.fingerprint_file(filepath)
|
|
|
|
elif command == 'recognize': # Recognize audio
|
|
|
|
source = sys.argv[2]
|
|
song = None
|
|
|
|
if source in ['mic', 'microphone']:
|
|
|
|
seconds = int(sys.argv[3])
|
|
djv = init()
|
|
song = djv.recognize(MicrophoneRecognizer, seconds=seconds)
|
|
|
|
elif source == 'file':
|
|
|
|
djv = init()
|
|
sourceFile = sys.argv[3]
|
|
song = djv.recognize(FileRecognizer, sourceFile)
|
|
|
|
else:
|
|
|
|
showHelp()
|
|
|
|
print song
|
|
|
|
else:
|
|
|
|
showHelp()
|
|
|