-
Notifications
You must be signed in to change notification settings - Fork 0
/
fingerprinter_cmd.py
31 lines (25 loc) · 874 Bytes
/
fingerprinter_cmd.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
# Include standard modules
import argparse
from db_handler.db_handler import DatabaseHandler
from mysql.connector.errors import ProgrammingError
from utils.catalog_utils import purge_temp_folder
try:
from fingerprinting import djv
except ProgrammingError:
db = DatabaseHandler()
db.create_db('dejavu')
from fingerprinting import djv
import gc
# Initiate the parser
parser = argparse.ArgumentParser()
parser.add_argument("-T", "--tempdir", help="local temporary directory")
args = parser.parse_args()
if __name__ == '__main__':
if args.tempdir:
local_audio_temp_dir = args.tempdir
print("start making fingerprints")
djv.get_fingerprints_from_directory(local_audio_temp_dir, ['wav'], 1)
purge_temp_folder(local_audio_temp_dir)
gc.collect()
else:
raise ValueError('No value was given for temp dir.')