main.py

#!/usr/bin/python3
# -----------------------------------------------------------------
# simple script for edit and creating pdf files.
# and converting txt files to pdf files.
#
#
#
# Author:N84.
#
# Create Date:Fri Mar 18 11:28:38 2022.
# ///
# ///
# ///
# -----------------------------------------------------------------

from os import system
from os import name as OS_NAME
from os import getcwd
from os import listdir
from fpdf import FPDF
from sys import argv


def clear():
    """wipe the terminal screen."""

    if OS_NAME == "posix":
        # for *nix machines.
        system("clear")

    elif OS_NAME == "windows":
        system("cls")

    else:
        # for other os in the world.
        # system("your-command")
        pass

    return None


clear()


def get_args():
    """return all the arguments that pass,
    to the program except the program name."""

    return argv[1:]


def separate2lines(string: str, max_char: int = 64):
    """separate a string into multi line string, with max_length of line,
    equal to max_char.
    we will use this function to align pdf lines."""

    # first make sure to remove all linefeeds.
    text = string.replace("\n", "")

    lines = []
    temp_string, counter = "", 0

    for char in text:

        if counter == max_char-1:
            lines.append(temp_string + char + "\n")
            temp_string = ""
            counter = 0
            continue

        temp_string += char
        counter += 1

    lines.append(temp_string)

    return lines


def file_exist(file_name: str):
    """checkout if the given file name is exist,
    in the working dir or not."""

    return file_name in listdir(getcwd())


def txt2pdf(file_name: str = None):
    """convert any text  => *.txt, to pdf"""

    # guard condition.
    if not file_exist(file_name):
        return

    if file_name is None:
        raise Exception("FilePathError: please enter the file path.")

    with open(f"{getcwd()}/{file_name}", "r") as file:
        text = file.read()

    # create pdf object.
    pdf = FPDF()

    pdf.add_page()

    pdf.set_font("Arial", size=12)

    text_lines = separate2lines(text, 90)
    # print(text_lines)

    for index, line in enumerate(text_lines):
        pdf.cell(0, 10, txt=line, ln=1, align="L")

    # make sure to split the file from the dot and get the name only.
    # in simple words remove the file extension.
    pdf.output(f"./{file_name.split('.')[0]}.pdf")

    pdf.close()

    return None


def main():

    # get all the args=>file-names.
    files = get_args()

    for file in files:
        txt2pdf(file)


if __name__ == '__main__':
    main()