Merge pull request #194961 from Luflosi/add/python3Packages.invisible-watermark

pkgs/development/python-modules/invisible-watermark/default.nix

@@ -0,0 +1,49 @@
+{ lib
+, buildPythonPackage
+, pythonOlder
+, fetchFromGitHub
+, opencv4
+, pytorch
+, onnx
+, onnxruntime
+, pillow
+, pywavelets
+, numpy
+}:
+
+buildPythonPackage rec {
+  pname = "invisible-watermark";
+  version = "0.1.5";
+  disabled = pythonOlder "3.6";
+
+  src = fetchFromGitHub {
+    owner = "ShieldMnt";
+    repo = "invisible-watermark";
+    rev = version;
+    sha256 = "sha256-NGDPEETuM7rYbo8kXYoRWLJWpa/lWLKEvaaiDzSWYZ4=";
+  };
+
+  propagatedBuildInputs = [
+    opencv4
+    pytorch
+    onnx
+    onnxruntime
+    pillow
+    pywavelets
+    numpy
+  ];
+
+  postPatch = ''
+    substituteInPlace setup.py \
+      --replace 'opencv-python>=4.1.0.25' 'opencv'
+  '';
+
+  pythonImportsCheck = [ "imwatermark" ];
+
+  meta = with lib; {
+    description = "A library for creating and decoding invisible image watermarks";
+    homepage = "https://github.com/ShieldMnt/invisible-watermark";
+    license = licenses.mit;
+    maintainers = with maintainers; [ Luflosi ];
+  };
+}

pkgs/top-level/python-packages.nix

@@ -4565,6 +4565,8 @@ self: super: with self; {
 
   intreehooks = callPackage ../development/python-modules/intreehooks { };
 
+  invisible-watermark = callPackage ../development/python-modules/invisible-watermark { };
+
   invocations = callPackage ../development/python-modules/invocations { };
 
   invoke = callPackage ../development/python-modules/invoke { };