Merge pull request larymak#88 from odinmay/main

Add CLI-Photo-Watermark script, Update readme.md

Author: larymak

CLI-Photo-Watermark/README.md

@@ -0,0 +1,11 @@
+# Usage #
+ ---
+
+- This script is for watermarking an entire directory of images ( .jpgs | .pngs )
+- Run the script in the directory that you want to watermark like this:
+    ``python main.py -w [watermark] -p [position] ``    
+
+- watermark - The text you want to watermark on the images
+- position - ul : upper left , ur - upper right , ll - lower left , lr - lower right
+
+- This will make a folder called 'Watermarked' and copy the new watermarked images into it, preserving original images

CLI-Photo-Watermark/main.py

@@ -0,0 +1,76 @@
+import argparse
+import cv2
+import os
+
+# Watermark Configuration
+font = cv2.FONT_HERSHEY_COMPLEX
+color = (255, 255, 255)
+thickness = 4
+
+# Setting up the argument parser for CMD Line interface
+ap = argparse.ArgumentParser()
+ap.add_argument('-f', '--file', required=False,
+                help='Path to target file')
+ap.add_argument('-w', '--watermark', required=True,
+                help='Text you would like to watermark image with | (Enclose in quotes if there are spaces)')
+ap.add_argument('-d', '--directory', required=False,
+                help='Processes every image in the CWD')
+ap.add_argument('-p', '--position', required=True,
+                help='Options are "ul"(upper left) "ur"(upper right) "ll"(lower left) "lr"(lower right)')
+args = ap.parse_args()
+print(args)
+
+def process_image(filename, watermark, pos):
+    """
+    :param filename: str
+        the path of the photo, built from cwd
+    :param watermark: str
+        the text you want watermarked on the image
+    :param pos: str
+        the position of the watermark  ex. "ll" (lower left) | "ur" (upper right)
+    :return: None
+        a new folder name "Watermarked" will be made in CWD with finished images
+    """
+
+    working_image = cv2.imread(filename)
+    text_length = len(watermark)
+    if working_image.shape[0] >= 4000:
+        avg_char = 120
+        text_width = text_length * avg_char
+        fontScale = 6
+        image_ul = (0, 150)
+        image_ur = (working_image.shape[1] - text_width, 150)
+        image_ll = (0, working_image.shape[0] - 50)
+        image_lr = (working_image.shape[1] - text_width, working_image.shape[0] - 50)
+    else:
+        avg_char = 80
+        text_width = text_length * avg_char
+        fontScale = 3
+        image_ul = (0, 100)
+        image_ur = (working_image.shape[1] - text_width, 100)
+        image_ll = (0, working_image.shape[0] - 50)
+        image_lr = (working_image.shape[1] - text_width, working_image.shape[0] - 50)
+
+    if pos == 'ul':
+        new_image = cv2.putText(working_image, args.watermark, image_ul, font, fontScale, color, thickness, cv2.LINE_AA)
+
+    if pos == 'ur':
+        new_image = cv2.putText(working_image, args.watermark, image_ur, font, fontScale, color, thickness, cv2.LINE_AA)
+
+    if pos == 'll':
+        new_image = cv2.putText(working_image, args.watermark, image_ll, font, fontScale, color, thickness, cv2.LINE_AA)
+
+    if pos == 'lr':
+        new_image = cv2.putText(working_image, args.watermark, image_lr, font, fontScale, color, thickness, cv2.LINE_AA)
+
+    if not os.path.exists(os.getcwd() + '\\Watermarked'):
+        os.mkdir(os.getcwd() + '\\Watermarked')
+
+    path = os.getcwd() + '\\' + 'Watermarked' + '\\' + file
+    cv2.imwrite(path, new_image)
+
+
+# Call function on all files in CWD ending with .png or .jpg
+for file in os.listdir(os.getcwd()):
+    if file.endswith('.jpg') or file.endswith('.png'):
+        process_image(os.getcwd() + '\\' + file, args.watermark, args.position)

CLI-Photo-Watermark/requirements.txt

@@ -0,0 +1 @@
+opencv-python~=4.5.1.48

CLI-Photo-Watermark/testphoto.jpg

@@ -87,4 +87,5 @@ The contribution guidelines are as per the guide [HERE](https://github.com/larym
 | 44    | [Sudoku Solver](https://github.com/Mannuel25/Python-project-Scripts/tree/main/SudokuSolver)             | [Ruben Grande Muñoz](https://github.com/RgrMz)         |
 | 45    | [Duplicate File Remover](https://github.com/mas-designs/Python-project-Scripts/tree/main/Remove%20Duplicate%20Files%20in%20Folder)             | [Michael Stadler](https://github.com/mas-designs)         |
 | 46    |  [Image Divider](https://github.com/larymak/Python-project-Scripts/tree/main/ImageDivider)              | [Rajarshi Banerjee](https://github.com/GSAUC3)          |)
-| 47    |  [Morse Code Converter](https://github.com/HarshitRV/Python-project-Scripts/tree/main/Morse-Code-Converter)              | [HarshitRV](https://github.com/HarshitRV)          |)
+| 47    |  [Morse Code Converter](https://github.com/HarshitRV/Python-project-Scripts/tree/main/Morse-Code-Converter)              | [HarshitRV](https://github.com/HarshitRV)          |)
+| 48    | [CLI Photo Watermark](https://github.com/odinmay/Python-project-Scripts/tree/main/CLI-Photo-Watermark)                                           | [Odin May](https://github.com/odinmay)