def save(self, *args, **kwargs):
mid, title, extra = iscc.meta_id(self.title, self.extra)
if self.ident:
new_ident = [mid] + list(self.ident.split('-')[1:])
self.ident = '-'.join(new_ident)
if self.file:
new_upload = isinstance(self.file.file, UploadedFile)
if new_upload:
# Generate ISCC
filename, file_extension = splitext(self.file.name)
ext = file_extension.lower().lstrip('.')
data = self.file.open('rb').read()
if ext in self.TEXT_EXTENSIONS:
if ext == 'docx':
text = docx2txt.process(BytesIO(data))
print(text)
else:
text = self.file.open().read()
cid = iscc.content_id_text(text)
elif ext in self.IMAGE_EXTENSIONS:
cid = iscc.content_id_image(BytesIO(data))
did = iscc.data_id(data)
iid, self.tophash = iscc.instance_id(data)
iscc_code = '-'.join((mid, cid, did, iid))
self.ident = iscc_code
super(MediaContent, self).save(*args, **kwargs)
def test_content_id_mixed():
cid_t_1 = iscc.content_id_text("Some Text")
cid_t_2 = iscc.content_id_text("Another Text")
cid_m = iscc.content_id_mixed([cid_t_1])
assert cid_m == "CM3k9pp7JS7nP"
cid_m = iscc.content_id_mixed([cid_t_1, cid_t_2])
assert cid_m == "CM3kHkNRGvnhB"
cid_i = iscc.content_id_image("lenna.jpg")
cid_m = iscc.content_id_mixed([cid_t_1, cid_t_2, cid_i])
assert cid_m == "CM3hswzATv9d3"
def run(self):
file_ending = self.file_path.split('.')[-1]
if file_ending in ['jpg', 'png', 'jpeg', 'gif']:
self.parent.content_id = iscc.content_id_image(self.file_path)
elif file_ending == 'txt':
with open(self.file_path, encoding='utf-8') as infile:
self.parent.content_id = iscc.content_id_text(infile.read())
elif file_ending == 'docx':
self.parent.content_id = iscc.content_id_text(
docx2txt.process(self.file_path))
self.parent.instance_id, self.parent.instance_hash = iscc.instance_id(
self.file_path)
self.parent.data_id = iscc.data_id(self.file_path)
if self.parent.meta_id:
self.parent.show_conflicts()
def test_content_id_image():
cid_i = iscc.content_id_image("lenna.jpg")
assert len(cid_i) == 13
assert cid_i == "CYmLoqBRgV32u"
data = BytesIO(open("lenna.jpg", "rb").read())
cid_i = iscc.content_id_image(data, partial=True)
assert len(cid_i) == 13
assert cid_i == "CimLoqBRgV32u"
img1 = Image.open("lenna.jpg")
img2 = img1.filter(ImageFilter.GaussianBlur(10))
img3 = ImageEnhance.Brightness(img1).enhance(1.4)
img4 = ImageEnhance.Contrast(img1).enhance(1.2)
cid1 = iscc.content_id_image(img1)
cid2 = iscc.content_id_image(img2)
cid3 = iscc.content_id_image(img3)
cid4 = iscc.content_id_image(img4)
assert iscc.distance(cid1, cid2) == 0
assert iscc.distance(cid1, cid3) == 2
assert iscc.distance(cid1, cid4) == 0
def batch(path, recursive, guess, debug):
"""Create ISCC Codes for all files in PATH.
Example:
$ iscc batch ~/Documents
"""
if debug:
log.add(sys.stdout)
results = []
for f in get_files(path, recursive=recursive):
filesize = os.path.getsize(f)
if not filesize:
msg = "Cannot proccess empty file: {}".format(f)
log.warning(msg)
continue
media_type = mime_clean(mime_guess(f))
if media_type not in SUPPORTED_MIME_TYPES:
fname = basename(f)
msg = "Unsupported file {} with mime type: {},,,,".format(
fname, media_type)
log.warning(msg)
continue
if media_type == "application/x-mobipocket-ebook":
try:
tempdir, epub_filepath = mobi.extract(f)
tika_result = parser.from_file(epub_filepath)
shutil.rmtree(tempdir)
except Exception as e:
msg = "Error with mobi extraction %s"
log.error(msg)
continue
else:
tika_result = parser.from_file(f)
title = get_title(tika_result, guess=guess, uri=f)
mid, norm_title, _ = iscc.meta_id(title)
gmt = mime_to_gmt(media_type, file_path=f)
if gmt == GMT.IMAGE:
try:
cid = iscc.content_id_image(f)
except Exception as e:
msg = "Clould not proccess image: {} ({})".format(f, e)
log.error(msg)
continue
elif gmt == GMT.TEXT:
text = tika_result["content"]
if not text:
msg = "Could not extract text from {}".format(basename(f))
log.warning(msg)
continue
cid = iscc.content_id_text(tika_result["content"])
elif gmt == GMT.AUDIO:
if not fpcalc.is_installed():
fpcalc.install()
features = audio_id.get_chroma_vector(f)
cid = audio_id.content_id_audio(features)
elif gmt == GMT.VIDEO:
features = video_id.get_frame_vectors(abspath(f))
cid = video_id.content_id_video(features)
else:
log.error("Could not generate ISCC")
continue
did = iscc.data_id(f)
iid, tophash = iscc.instance_id(f)
iscc_code_cs = ",".join((mid, cid, did, iid))
click.echo("{iscc_code},{tophash},{fname},{gmt},{title}".format(
iscc_code=iscc_code_cs,
tophash=tophash,
fname=basename(f),
gmt=gmt,
title=norm_title,
))
iscc_code = "-".join((mid, cid, did, iid))
results.append(
dict(
iscc=iscc_code,
norm_title=norm_title,
tophash=tophash,
gmt=gmt,
file_name=basename(f),
))
return results
def gen(file, guess, title, extra, verbose):
"""Generate ISCC Code for FILE."""
filesize = os.path.getsize(file.name)
if not filesize:
raise click.BadParameter("Cannot proccess empty file: {}".format(
file.name))
media_type = mime_clean(mime_guess(file.name))
if media_type not in SUPPORTED_MIME_TYPES:
click.echo("Unsupported media type {}.".format(media_type))
click.echo(
"Please request support at https://github.com/iscc/iscc-cli/issues"
)
if media_type == "application/x-mobipocket-ebook":
tempdir, epub_filepath = mobi.extract(file.name)
tika_result = parser.from_file(epub_filepath)
shutil.rmtree(tempdir)
else:
tika_result = parser.from_file(file.name)
if not title:
title = get_title(tika_result, guess=guess, uri=file.name)
if not extra:
extra = ""
mid, norm_title, _ = iscc.meta_id(title, extra)
gmt = mime_to_gmt(media_type, file_path=file.name)
if gmt == GMT.IMAGE:
cid = iscc.content_id_image(file.name)
elif gmt == GMT.TEXT:
text = tika_result["content"]
if not text:
click.echo("Could not extract text from {}".format(file.name))
return
cid = iscc.content_id_text(tika_result["content"])
elif gmt == GMT.AUDIO:
if not fpcalc.is_installed():
fpcalc.install()
features = audio_id.get_chroma_vector(file.name)
cid = audio_id.content_id_audio(features)
elif gmt == GMT.VIDEO:
features = video_id.get_frame_vectors(abspath(file.name))
cid = video_id.content_id_video(features)
else:
click.echo("Could not generate ISCC")
return
did = iscc.data_id(file.name)
iid, tophash = iscc.instance_id(file.name)
if not norm_title:
iscc_code = "-".join((cid, did, iid))
else:
iscc_code = "-".join((mid, cid, did, iid))
click.echo("ISCC:{}".format(iscc_code))
if verbose:
if norm_title:
click.echo("Norm Title: %s" % norm_title)
click.echo("Tophash: %s" % tophash)
click.echo("Filepath: %s" % file.name)
click.echo("GMT: %s" % gmt)
return dict(iscc=iscc_code,
norm_title=norm_title,
tophash=tophash,
gmt=gmt)
def web(url, guess, title, extra, verbose):
"""Generate ISCC Code from URL."""
extra = extra or ""
try:
resp = requests.get(url, headers=HEADERS, stream=True)
except Exception as e:
raise click.BadArgumentUsage(e)
data = BytesIO(resp.content)
media_type = clean_mime(detector.from_buffer(data))
if media_type not in SUPPORTED_MIME_TYPES:
click.echo("Unsupported media type {}".format(media_type))
click.echo(
"Please request support at https://github.com/iscc/iscc-cli/issues"
)
return
if media_type == "application/x-mobipocket-ebook":
data.seek(0)
tempdir, filepath = mobi.extract(data)
tika_result = parser.from_file(filepath)
shutil.rmtree(tempdir)
else:
data.seek(0)
tika_result = parser.from_buffer(data)
if not title:
title = get_title(tika_result, guess=guess, uri=url)
mid, norm_title, _ = iscc.meta_id(title, extra)
gmt = mime_to_gmt(media_type)
if gmt == GMT.IMAGE:
data.seek(0)
cid = iscc.content_id_image(data)
elif gmt == GMT.TEXT:
text = tika_result["content"]
if not text:
click.echo("Could not extract text")
return
cid = iscc.content_id_text(tika_result["content"])
elif gmt == GMT.AUDIO:
if not fpcalc.is_installed():
fpcalc.install()
data.seek(0)
features = audio_id.get_chroma_vector(data)
cid = audio_id.content_id_audio(features)
elif gmt == GMT.VIDEO:
local_path = download_file(url, sanitize=True)
features = video_id.get_frame_vectors(local_path)
cid = video_id.content_id_video(features)
os.remove(local_path)
data.seek(0)
did = iscc.data_id(data)
data.seek(0)
iid, tophash = iscc.instance_id(data)
if not norm_title:
iscc_code = "-".join((cid, did, iid))
else:
iscc_code = "-".join((mid, cid, did, iid))
click.echo("ISCC:{}".format(iscc_code))
if verbose:
if norm_title:
click.echo("Norm Title: %s" % norm_title)
click.echo("Tophash: %s" % tophash)
click.echo("Filepath: %s" % url)
click.echo("GMT: %s" % gmt)
return dict(iscc=iscc_code,
norm_title=norm_title,
tophash=tophash,
gmt=gmt)
def from_file(file: UploadFile = File(...),
title: str = Form(""),
extra: str = Form("")):
"""Generate Full ISCC Code from Media File with optional explicit metadata."""
media_type = detector.from_buffer(file.file)
if media_type not in SUPPORTED_MIME_TYPES:
raise HTTPException(
HTTP_415_UNSUPPORTED_MEDIA_TYPE,
"Unsupported media type '{}'. Please request support at "
"https://github.com/iscc/iscc-service/issues.".format(media_type),
)
if media_type == "application/x-mobipocket-ebook":
file.file.seek(0)
tempdir, filepath = mobi.extract(file.file)
tika_result = parser.from_file(filepath)
shutil.rmtree(tempdir)
else:
file.file.seek(0)
tika_result = parser.from_buffer(file.file)
if not title:
title = get_title(tika_result, guess=True)
mid, norm_title, norm_extra = iscc.meta_id(title, extra)
gmt = mime_to_gmt(media_type)
if gmt == GMT.IMAGE:
file.file.seek(0)
cid = iscc.content_id_image(file.file)
elif gmt == GMT.TEXT:
text = tika_result["content"]
if not text:
raise HTTPException(HTTP_422_UNPROCESSABLE_ENTITY,
"Could not extract text")
cid = iscc.content_id_text(tika_result["content"])
elif gmt == GMT.AUDIO:
file.file.seek(0)
features = audio_id.get_chroma_vector(file.file)
cid = audio_id.content_id_audio(features)
elif gmt == GMT.VIDEO:
file.file.seek(0)
_, ext = splitext(file.filename)
fn = "{}{}".format(uuid.uuid4(), ext)
tmp_path = join(APP_DIR, fn)
with open(tmp_path, "wb") as buffer:
shutil.copyfileobj(file.file, buffer)
features = video_id.get_frame_vectors(tmp_path)
cid = video_id.content_id_video(features)
os.remove(tmp_path)
file.file.seek(0)
did = iscc.data_id(file.file)
file.file.seek(0)
iid, tophash = iscc.instance_id(file.file)
if not norm_title:
iscc_code = "-".join((cid, did, iid))
else:
iscc_code = "-".join((mid, cid, did, iid))
components = iscc_split(iscc_code)
result = dict(
iscc=iscc_code,
tophash=tophash,
gmt=gmt,
bits=[code_to_bits(c) for c in components],
)
if norm_title:
result["title"] = title
result["title_trimmed"] = norm_title
if norm_extra:
result["extra"] = extra
result["extra_trimmed"] = norm_extra
file.file.close()
return result