nerdfont-patcher.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os
import sys
import re
import multiprocessing
from time import time
Script = "Nerdfont-patcher.py"

try:
    import psMat
except ImportError:
    sys.exit(Script + ": FontForge module is probably not installed. [See: http://designwithfontforge.com/en-US/Installing_Fontforge.html]")
try:
    import configparser
except ImportError:
    sys.exit(Script + ": configparser module is probably not installed. Try `pip install configparser` or equivalent")
try:
    import fontforge
except ImportError:
    sys.exit(
        Script + (
            ": FontForge module could not be loaded. Try installing fontforge python bindings "
            "[e.g. on Linux Debian or Ubuntu: `sudo apt install fontforge python-fontforge`]"
        )
    )

def folder_exists(folder):
    ''' Bool: Checks if a folder exists '''
    return os.path.isdir(folder)


def patcher_exists():
    ''' Bool: Checks if the NerdFonts patcher script exists '''
    return os.path.isfile("./font-patcher")


def download_patcher():
    ''' Downloads the Nerd Fonts Patcher script '''
    print("Downloading NerdFonts Patcher Script")
    command = "curl -s -L https://raw.githubusercontent.com/ryanoasis/nerd-fonts/master/font-patcher --output font-patcher"
    os.system(command)


def download_src_fonts():
    ''' Downloads Glyph source files from the Nerd Fonts Repo '''
    with open("font_names.txt", "r") as f:
        print("Downloading glyph source fonts\n")
        for font_name in f.readlines():

            # Removes \r\n or crlf
            font_name = font_name.rstrip()

            # Skips the comments in the font_names.txt
            if "#" in font_name:
                continue

            # User feedback
            print("Downloading {}".format(font_name))
            command = "curl -s -L 'https://github.com/ryanoasis/nerd-fonts/blob/master/src/glyphs/*?raw=true' --output 'src/glyphs/*'".replace(
                "*", font_name)
            os.system(command)


def split(xlist, cpu_count):
    ''' Splits list of length n in x number of lists '''
    k, m = divmod(len(xlist), cpu_count)
    if list((xlist[i * k + min(i, m):(i + 1) * k + min(i + 1, m)] for i in range(cpu_count))) == [[], []]:
        return []
    return (xlist[i * k + min(i, m):(i + 1) * k + min(i + 1, m)] for i in range(cpu_count))


def cpu_n(processes):
    ''' Determines the number of cpu processes to use given the number of n '''
    if (len(processes) % 2 == 0) and (len(processes) > 2):
        return 4
    elif (not len(processes) % 2 == 0) and (len(processes) == 1):
        return 1
    elif (not len(processes) % 2 == 0) and (len(processes) >= 5):
        return 4
    else:
        return 2


def fix_filenames_in(folder):
    ''' (ง '̀͜ '́ )ง Removes spaces from filenames within a folder '''
    # NOTE: Shorter signature from Core Kramer StackOverflow
    # https://stackoverflow.com/questions/64468635/is-there-a-better-way-to-replace-multiple-spaces-in-file-names
    files = os.listdir(folder)

    for file_name in files:
        new_name = ''.join([re.sub(r'\s+', '-', i) for i in file_name])
        old_filename = folder + os.sep + file_name
        new_filename = folder + os.sep + new_name
        os.rename(old_filename, new_filename)


def blitzer():
    pass


def patch(fonts, name):
    ''' Font patching time! (•̀o•́)ง Go Drink a coffee. '''
    # Loop over each item and patch the glyphs
    for i, _ in enumerate(fonts):

        # This will patch everything. Powerline, Weather icons, FontAwesome... EVERYTHING!
        # It takes a while to patch fonts so get a coffee while waiting. Oh! and the '-w 'switch is for
        # Windows compatibility.. Something about limiting the number of characters in the font name.
        # More Info here:
        # https://github.com/ryanoasis/nerd-fonts#option-8-patch-your-own-font
        print("\nPatching {}\n".format(fonts[i]))
        command = "fontforge -quiet -script ." + os.sep + "font-patcher -q -w -c --no-progressbars {} -ext ttf -out Patched".format(fonts[i]) + os.sep + "{}".format(name)
        os.system(command)



if __name__ == '__main__':
    # Get number of arguments
    arg_len = len(sys.argv)

    # We need two args
    if arg_len > 1:
        processes = []
        font_folder, name = sys.argv[1:]

        # Create the src folder and download all the font glyphs from the NerdFonts repo on first run
        if not folder_exists("." + os.sep + "src" + os.sep + "glyphs"):
            os.makedirs("." + os.sep + "src" + os.sep + "glyphs")
            download_src_fonts()

        # Create a folder based on the name param for the fonts
        if not folder_exists('.' + os.sep + 'Patched' + os.sep + '{}'.format(name)):
            os.makedirs('.' + os.sep + 'Patched' + os.sep + '{}'.format(name))

        # Download the NerdFonts Patcher on first run
        if not patcher_exists():
            download_patcher()

        # Scan the folder passed in via script arg
        fix_filenames_in(font_folder)
        fonts = os.listdir(font_folder)

        # Get fonts only
        fonts = [font_folder + os.sep + item for item in fonts if '.ttf' in item or '.otf' in item]

        # Get num processes to use
        cpus = cpu_n(fonts)

        # Split list into smaller lists
        groups = list(split(fonts, cpus))

        # Real num of CPUs to use
        cpu_num = len(groups)

        # Multiprocessing
        ts = time()
        for i in range(cpu_num):
            process = multiprocessing.Process(target=patch, args=(groups[i],name))
            processes.append(process)
            process.start()

        for proc in processes:
            proc.join()
        te = time()-ts

        # timer
        os.system('clear')
        print("(⌐■_■) Done!")
        print("Took {:.2f} seconds".format(te))
        print('Fonts are in the Patched folder')
        # patch(fonts, name)
    else:
        print("\n")
        print("+------------------------------------------------------+")
        print("| ¯\_(ツ)_/¯            Whoopss!              (ง '̀͜ '́ )ง  |")
        print("+------------------------------------------------------+\n")
        print("Almost did it..! I need two arguments.\n\n1) The Folder where the fonts are, and\n2) The name of the font to Patch\n")
        print("Like:\n > {} ./Fonts/Agave Agave\n\n".format(sys.argv[0]))