nixpkgs/pkgs/development/python-modules/videocr/default.nix
2023-01-22 11:17:58 -08:00

48 lines
1.0 KiB
Nix

{ lib
, buildPythonPackage
, fetchPypi
, levenshtein
, pytesseract
, opencv4
, fuzzywuzzy
}:
buildPythonPackage rec {
pname = "videocr";
version = "0.1.6";
src = fetchPypi {
inherit pname version;
sha256 = "1clifwczvhvbaw2spgxkkyqsbqh21vyfw3rh094pxfmq89ylyj63";
};
propagatedBuildInputs = [
levenshtein
pytesseract
opencv4
fuzzywuzzy
];
postPatch = ''
substituteInPlace setup.py \
--replace "python-Levenshtein" "Levenshtein" \
--replace "opencv-python" "opencv"
substituteInPlace videocr/constants.py \
--replace "master" "main"
substituteInPlace videocr/video.py \
--replace '--tessdata-dir "{}"' '--tessdata-dir="{}"'
'';
# Project has no tests
doCheck = false;
pythonImportsCheck = [ "videocr" ];
meta = with lib; {
description = "Extract hardcoded subtitles from videos using machine learning";
homepage = "https://github.com/apm1467/videocr";
license = licenses.mit;
maintainers = with maintainers; [ ozkutuk ];
};
}