Script to convert only webp to jpg and rename

[Nephiro]

Since I don't like the webp format I created a script to convert the images from webp to jpg. It also has a renaming function.
I know that there is the setting in the downloader to convert the images at the end, but I want to convert only webp, that option convert all the images, even png (or jpg in case you select png) that I'm ok with it

Works like this:

• With the variable "SITES_LIST" you activate the rename function (if you don't want it then write "SITES_LIST = []"), if you wanna add a site you need to check the name of the downloader you are using, for example downloading from exHentai the downloader name to add is "ehen".

• If you want the rename function it will rename the files at the end of the process with this format. If the total pages to download are below 100 then it will have 2 digits (ex: 01.jpg, 02.jpg, etc). if between 100 and 999 will have 3 digits (ex: 001.jpg, 002.jpg, etc), etc.
  The rename function has another feature. Some posts in site like exHentai have the author already in the title (but not all), so this script will search if the artist is already in the name and will fix it, even the folder name. If instead the author is missing, it will remove the [N\A] from the folder name.
  I think I already covered most of the sites with this problem, but if you wanna add or remove you can do it easily like I said before.
  Also there is the SPECIAL_AUTHORS_LIST variable, where you can fix some author name. Mostly because the ones already wrote in the title are not correct (like wrong lower characters, etc), if you notice problems with the post you are downloading, just add the "fixed" author name in the list and you are done. Beware that if the author name has different space characters, like for example "Arisayoshi" and "Arisa Yoshi", then it will not work and you have to fix it manually

Here the script:

#coding: utf8
#title: Convert webp to jpg and rename
#author: Nephiro

from utils import Hook, get_print, clean_title
from PIL import Image
import os


SPECIAL_AUTHORS_LIST = ['Melissa N', 'MeowWithMe']
SITES_LIST = ['kemono.party', 'hitomi', 'ehen', 'nhentai']

@Hook.task_finished('convert')
def hook_convert(cw):
    if cw.type in SITES_LIST:
        index = 0
        folderNameChanged = False
        totNumDecimals = len(str(len(cw.names)))
        pdfFile = os.path.join(cw.dir, os.path.basename(cw.dir) + '.pdf')
        artist = '[{}]'.format(cw.artist.title())
        newArtist = artist
        
        for i, specialAuthor in enumerate(SPECIAL_AUTHORS_LIST):
            if specialAuthor.lower() in cw.artist.lower():
                newArtist = '[{}]'.format(specialAuthor)
                break
        
        folderName = cw.title
        folderPath = cw.dir
        
        if folderName.count(artist) > 1 or newArtist != artist or '[N／A]' in folderName:
            if '[N／A]' in folderName:
                folderName = folderName.replace('[N／A]', '').strip()
            else:
                folderName = folderName.replace(artist, '')
                folderName = folderName.replace(newArtist, '')
                folderName = newArtist + ' ' + folderName.strip()
            
            cw.setTitle(folderName)
            folderPath = os.path.realpath(os.path.join(os.path.dirname(cw.dir), folderName))
            
            if folderPath != os.path.realpath(cw.dir):
                folderNameChanged = True
            
            if not os.path.exists(folderPath):
                os.makedirs(folderPath)
        
        if totNumDecimals < 2:
            totNumDecimals = 2
        
        for i, file in enumerate(cw.names):
            if os.path.exists(file):
                index += 1
                name = str(index)
                numDecimals = len(name)
                ext = os.path.splitext(file)[1]
                
                if numDecimals < totNumDecimals:
                    for x in range(totNumDecimals - numDecimals):
                        name = '0' + name
                
                filePath = os.path.realpath(os.path.join(folderPath, name))
                cw.setNameAt(i, filePath)
                
                if 'webp' in ext:
                    im = Image.open(file)
                    rgb_im = im.convert('RGB')
                    rgb_im.save(filePath + '.jpg', quality=95)
                    os.remove(file)
                else:
                    os.replace(file, filePath + ext)
        
        if os.path.exists(pdfFile):
            pdfName = folderName.replace(newArtist, '').strip()
            os.replace(pdfFile, os.path.realpath(os.path.join(folderPath, pdfName + '.pdf')))
        
        if folderNameChanged:
            os.rmdir(cw.dir)
            cw.dir = folderPath
    else:
        for file in cw.names:
            name, ext = os.path.splitext(file)
            
            if 'webp' in ext:
                im = Image.open(file)
                rgb_im = im.convert('RGB')
                rgb_im.save(name + '.jpg', quality=95)
                os.remove(file)

messageBox('Script added')

[NightLancer]

For those who don't like having wrong extensions but doesn't want to convert files I found another solution: #4791 (comment)
It's semi-automatic and maybe not so comfortable as this one, but it doesn't reconvert images and there's no need to specify authors/sites, as it only renames files extensions.

[Nephiro]

You don't need to specify any author or sites to make the conversion works if you read my post ;) You can leave those empty, you can even remove everything inside the "if cw.type in SITES_LIST" and use only the code in the else and will convert all webp files to natural jpg

[NightLancer]

@Nephiro I've read your post, as well as looked through code and saw ext = os.path.splitext(file)[1], which will give "jpg" in case of v2ph as hitomi doesn't double-check real file extension and relies on the information from site. So, for anyone who encounters same problem there is only viable option with real type checking.

[Nephiro]

Oh ok, now I see the problem, sorry. However you can always do like this. Copy the code in the "else" and replace it in the "if cw.type in SITES_LIST". Add that site (or site in general you wanna check), and move the "im = Image.open(file)" before the "if 'webp' in ext" and change it to "if im.format.lower() == 'webp'".
It should work like this because it should give you the real format of the file image