1

python webp.py -i <input_file>

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66

from utils.resize import resize
from utils.add_watermark import add_watermark
from utils.get_prompt import get_prompt
import argparse
import glob
import os


def resize_and_addwatermark(inputFile, outputFile):
    image = resize(inputFile)
    result = add_watermark(image)
    result.save(outputFile, 'webp', quality=80)


def images_handler(inputFolder, outputFolder, image_name='girl'):
   # get png/jpg files and index them
    images = glob.glob(inputFolder + "/*.png")
    images.extend(glob.glob(inputFolder + "/*.jpg"))
    images.extend(glob.glob(inputFolder + "/*.jpeg"))

    name_index = 0
    has_prompt = False

    for image in images:
        if os.path.isfile(image):
            has_prompt = generate_prompt(image, has_prompt, outputFolder)
            print('doing', image)
            file_name = image_name + str(name_index) + ".webp"
            name_index += 1
            output_file = os.path.join(outputFolder, file_name)
            resize_and_addwatermark(image, output_file)
            print('done', image, 'to', output_file)
    print('done all')


def generate_prompt(image, has_prompt, outputFolder):
    if not has_prompt:
        prompt = get_prompt(image)
        if prompt:
            has_prompt = True
            prompt_file = os.path.join(outputFolder, 'prompt.txt')
            with open(prompt_file, 'w', encoding='utf8') as f:
                f.write(prompt)
                print('done', prompt_file)
    return has_prompt


def get_args():
    parser = argparse.ArgumentParser(
        description='Resize image and add watermark')
    parser.add_argument('-i', '--input', help='input folder', required=True)
    parser.add_argument('-n', '--name', help='name of images', default='girl_')
    args = parser.parse_args()
    return args


if __name__ == '__main__':
    args = get_args()
    inputFolder = args.input
    outputFolder = inputFolder + "\output"
    name = args.name
    if not os.path.exists(outputFolder):
        os.makedirs(outputFolder)

    images_handler(inputFolder, outputFolder, name)

1

python batch_webp.py -i <input_dir>

1

python py2wp.py -f <input_images_folder> -tags <tags> -cat <categories> -t <title> -s <slug>

  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109

from wordpress_xmlrpc import Client, WordPressPost
from wordpress_xmlrpc.methods.posts import NewPost
from wordpress_xmlrpc.methods import media
from wordpress_xmlrpc.compat import xmlrpc_client

import argparse
import os

# replace with your own wordpress site url and credentials
wp = Client('http://your_wp_site.com/xmlrpc.php', 'admin', 'password')

def python_to_wp(folder, title, tags, category, slug=None):
    print('Creating post: ' + title)

    post = WordPressPost()
    post.title = title
    if slug:
        post.slug = slug
    post.content = ''
    prompt_txt_content = get_prompt_txt_content(folder)
    post.terms_names = {
        'post_tag': tags.split(','),
        'category': category.split(',')
    }

    post.excerpt = prompt_txt_content

    post.post_status = 'publish'

    images = get_folder_images(folder)

    has_cover = False

    for image in images:
        # upload image
        image_with_path = os.path.join(folder, image)
        image_uploaded = upload_image(image_with_path)
        image_url = image_uploaded['url']
        image_id = image_uploaded['id']
        image_name = image_uploaded['file']

        # add image to post content
        post.content += f'''<figure class="wp-block-image size-full"><img decoding="async" loading="lazy" src="{image_url}" alt="" ></figure>'''
        if not has_cover:
            post.thumbnail = image_id
            if image_name == 'cover.webp':
                has_cover = True

    wp.call(NewPost(post))
    print('Post created: ' + title)


def get_folder_images(folder):
    images = [f for f in os.listdir(
        folder) if os.path.isfile(os.path.join(folder, f))]
    images = [f for f in images if f.endswith('.webp')]
    return images


def upload_image(image):
    # get image name
    image_name = os.path.basename(image)

    # get image data
    data = {
        'name': image_name,
        'type': 'image/webp',
    }

    # open image
    with open(image, 'rb') as img:
        data['bits'] = xmlrpc_client.Binary(img.read())

    # upload image
    response = wp.call(media.UploadFile(data))

    return response


def get_prompt_txt_content(folder):
    prompt_file = os.path.join(folder, 'prompt.txt')
    if os.path.isfile(prompt_file):
        with open(prompt_file, 'r', encoding='utf8') as f:
            content = f.read()
            return content
    else:
        return ''


def get_args():
    parser = argparse.ArgumentParser()
    parser.add_argument('-f', '--folder', help='Choose Folder', required=True)
    parser.add_argument('-t', '--title', help='Post Title', required=True)
    parser.add_argument('-s', '--slug', help='Post Slug')
    parser.add_argument('-tags', '--tags', help='Post Tags', required=True)
    parser.add_argument('-cat', '--category',
                        default='妹子图', help='Post Category')
    args = parser.parse_args()
    return args


def main():
    args = get_args()
    python_to_wp(**vars(args))


if __name__ == '__main__':
    main()