Initial Commit. 3 scripts

Author: hetsonii

.gitignore

@@ -0,0 +1 @@
+certificates/*

README.md

@@ -0,0 +1,19 @@
+# CertifyD
+
+- Version 1:
+    - No GUI
+    - Calculates the coordinates mathematically
+
+- Version 2:
+    - GUI
+    - Opens a image window to get the coordinates
+    - Opens a window to get the template image and data. No defaults set
+
+- Version 3:
+    - Filename defaults set. (data/template.png and data/names.csv)
+    - colourful terminal logging 
+
+- Version 4:
+    - Under development
+
+    

data/names.csv

@@ -0,0 +1,6 @@
+,name,checked_in_at,ticket_name
+,ABCD EFGH,5899662.00,Standard
+,IJKL MNOP,5899662.00,Standard
+,QRST UVWX,5899662.00,Standard
+,YZAB CDEF,5899662.00,Standard
+

data/template.png

@@ -0,0 +1,7 @@
+chardet==5.2.0
+pillow==10.2.0
+reportlab==4.1.0
+setuptools==59.6.0
+colorama==0.4.6
+pyfiglet==1.0.2
+termcolor==2.4.0

requirements.txt

@@ -0,0 +1,73 @@
+import csv
+import os
+from PyPDF2 import PdfReader, PdfWriter
+from reportlab.lib.pagesizes import letter
+from reportlab.pdfgen import canvas
+from PIL import Image, ImageDraw, ImageFont
+
+def write_names_on_file(input_file, csv_path, output_folder):
+    name, ext = os.path.splitext(input_file)
+    if ext.lower() == '.pdf':
+        write_names_on_pdf(input_file, csv_path, output_folder)
+    elif ext.lower() in ['.jpg', '.jpeg', '.png', '.gif']:
+        write_names_on_image(input_file, csv_path, output_folder)
+    else:
+        print(f"Unsupported file type: {ext}")
+
+def get_middle_coordinates(width, height, text, font_size):
+    text_width = font_size * len(text) * 0.6  # Rough approximation of text width
+    x = (width - text_width) / 2
+    y = (height - font_size) / 2
+    return x, y
+
+def write_names_on_pdf(pdf_path, csv_path, output_folder):
+    with open(csv_path, 'r') as csv_file:
+        csv_reader = csv.DictReader(csv_file)
+        for row in csv_reader:
+            name = row['name']
+            output_pdf_path = os.path.join(output_folder, f"{name}.pdf")
+            reader = PdfReader(pdf_path)
+            writer = PdfWriter()
+
+            for page_num in range(len(reader.pages)):
+                page = reader.pages[page_num]
+                writer.add_page(page)
+
+                width, height = letter
+                x, y = get_middle_coordinates(width, height, name, 12)  # Adjust font size as needed
+                packet = canvas.Canvas("temp.pdf", pagesize=letter)
+                # packet.drawString(x-200, y-40, ' '.join(name.split()[:2]), fontName="Helvetica", fontSize=12)
+                packet.setFont("Helvetica", 12)  # Setting font to Helvetica with size 12
+                packet.drawString(x-200, y-40, ' '.join(name.split()[:2]))
+                packet.save()
+
+                overlay = PdfReader("temp.pdf")
+                page.merge_page(overlay.pages[0])
+
+            with open(output_pdf_path, 'wb') as output_pdf:
+                writer.write(output_pdf)
+
+            os.remove("temp.pdf")  # Remove temporary file
+
+def write_names_on_image(image_path, csv_path, output_folder):
+    with open(csv_path, 'r') as csv_file:
+        csv_reader = csv.DictReader(csv_file)
+        for row in csv_reader:
+            name = row['name']
+            output_image_path = os.path.join(output_folder, f"{name}{os.path.splitext(image_path)[1]}")
+            image = Image.open(image_path)
+            draw = ImageDraw.Draw(image)
+            width, height = image.size
+            x, y = get_middle_coordinates(width, height, ' '.join(name.split()[:2]), 40)  # Adjust font size as needed
+            draw.text((x-200, y-40), name.title(), fill="black", font=ImageFont.truetype("arial.ttf", size=30))
+            image.save(output_image_path)
+
+
+input_file = "data/template.png"
+csv_file = "data/names.csv"
+output_folder = "certificates"
+
+if not os.path.exists(output_folder):
+    os.makedirs(output_folder)
+
+write_names_on_file(input_file, csv_file, output_folder)

script_v1.py

@@ -0,0 +1,51 @@
+import csv
+import os
+from PIL import Image, ImageDraw, ImageFont
+import tkinter as tk
+from tkinter import filedialog
+
+# Global variables for coordinates
+x, y = None, None
+
+def get_user_coordinates(img):
+    global x, y
+
+    def on_click(event):
+        global x, y
+        x, y = event.x, event.y
+        root.destroy()
+
+    root = tk.Tk()
+    root.title("Select Coordinates")
+    root.geometry("+100+100")
+    root.bind("<Button-1>", on_click)
+
+    canvas = tk.Canvas(root, width=img.width, height=img.height)
+    canvas.pack()
+    photo_image = tk.PhotoImage(file=img.filename)
+    canvas.create_image(0, 0, anchor=tk.NW, image=photo_image)
+
+    root.mainloop()
+
+def write_names_on_image(image_path, csv_path, output_folder):
+    global x, y
+    get_user_coordinates(Image.open(image_path))  # Call once to get coordinates
+    with open(csv_path, 'r') as csv_file:
+        csv_reader = csv.DictReader(csv_file)
+        for row in csv_reader:
+            name = row['name']
+            output_image_path = os.path.join(output_folder, f"{name}{os.path.splitext(image_path)[1]}")
+            image = Image.open(image_path)
+            draw = ImageDraw.Draw(image)
+            draw.text((x, y), name.title(), fill="black", font=ImageFont.truetype("arial.ttf", size=30))
+            image.save(output_image_path)
+
+# Example usage:
+input_file = filedialog.askopenfilename(title="Select input image file")
+csv_file = filedialog.askopenfilename(title="Select CSV file")
+output_folder = "certificates"
+
+if not os.path.exists(output_folder):
+    os.makedirs(output_folder)
+
+write_names_on_image(input_file, csv_file, output_folder)

script_v2.py

@@ -0,0 +1,94 @@
+import csv
+import os
+from PIL import Image, ImageDraw, ImageFont
+import tkinter as tk
+from tkinter import filedialog
+from colorama import init, Fore
+from termcolor import colored
+import pyfiglet
+
+# Initialize colorama
+init()
+
+# Global variables for coordinates
+x, y = None, None
+
+# Icons for logs
+ICONS = {
+    'info': colored('[ ℹ ]', 'blue'),
+    'success': colored('[ ✔ ]', 'green'),
+    'warning': colored('[ ⚠ ]', 'yellow'),
+    'error': colored('[ ✖ ]', 'red'),
+}
+
+def get_user_coordinates(img):
+    global x, y
+
+    def on_click(event):
+        global x, y
+        x, y = event.x, event.y
+        root.destroy()
+
+    root = tk.Tk()
+    root.title("Select Coordinates")
+    root.geometry("+100+100")
+    root.bind("<Button-1>", on_click)
+
+    canvas = tk.Canvas(root, width=img.width, height=img.height)
+    canvas.pack()
+    photo_image = tk.PhotoImage(file=img.filename)
+    canvas.create_image(0, 0, anchor=tk.NW, image=photo_image)
+
+    root.mainloop()
+
+def write_names_on_image(image_path, csv_path, output_folder):
+    global x, y
+    get_user_coordinates(Image.open(input_file))  # Call once to get coordinates
+    confirmed = False
+    
+    while not confirmed:
+        with open(csv_path, 'r') as csv_file:
+            csv_reader = csv.DictReader(csv_file)
+            for row in csv_reader:
+                name = row['name']
+                output_image_path = os.path.join(output_folder, f"{name}{os.path.splitext(image_path)[1]}")
+                image = Image.open(image_path)
+                draw = ImageDraw.Draw(image)
+                draw.text((x, y), name.title(), fill="black", font=ImageFont.truetype("arial.ttf", size=30))
+                image.save(output_image_path)
+                print(f"{ICONS['success']} {colored(f'Name {name} placed on image {os.path.basename(output_image_path)}', 'green')}")
+
+        # Show preview and ask for confirmation
+        preview_image = Image.open(output_image_path)
+        preview_image.show()
+        confirmed = tk.messagebox.askyesno("Confirmation", "Is the location correct?")
+        
+        if not confirmed:
+            get_user_coordinates(Image.open(input_file))
+
+    print(f"{ICONS['info']} {colored('Task completed successfully!', 'blue')}")
+
+
+
+banner = pyfiglet.figlet_format("CertifyD", font="slant")
+print(banner)
+
+# Default files
+input_file = 'data/template.png'
+csv_file = 'data/names.csv'
+
+if not (os.path.exists(input_file) or os.path.exists(csv_file)):
+    print(f"{ICONS['info']} {colored('Default files template.png or names.csv not found. Please select files.', 'blue')}")
+    input_file = filedialog.askopenfilename(title="Select input image file")
+    csv_file = filedialog.askopenfilename(title="Select CSV file")
+
+
+output_folder = "certificates"
+
+if not os.path.exists(output_folder):
+    os.makedirs(output_folder)
+
+
+print(f"{ICONS['info']} {colored('Writing names on images...', 'blue')}")
+write_names_on_image(input_file, csv_file, output_folder)
+print(f"{ICONS['info']} {colored('Task completed successfully!', 'blue')}")