def import_directory(db_handle, path, label_id, include_subdirectories=True): path += '/' dirs = filter(os.path.isdir, [path + r for r in os.listdir(path)]) files = filter(os.path.isfile, [path + r for r in os.listdir(path)]) for f in files: found_dup = False for p in db_handle.get_photos_of_size(f): if filecmp.cmp(datastore.get(p[0]), f): print("Duplicate Found: ", f, p[0]) db_handle.label_photo(p[0], label_id) found_dup = True break if not found_dup: pid = db_handle.add_photo(f) datastore.add(f, pid) db_handle.label_photo(pid, label_id) if include_subdirectories: for d in dirs: name = d.replace(path, '') child_label = db_handle.create_label(name, label_id) import_directory(db_handle, d, child_label)
def img(id_): with open(datastore.get(id_), 'rb') as f: resp = f.read() return Response(resp, mimetype="image/jpeg")
def test_get(): assert (get("testVal"))