klemek
153 lines
5.8 KiB
Python
153 lines
5.8 KiB
Python
import os
|
|
import logging
|
|
from typing import List
|
|
import PIL
|
|
from os import path
|
|
from meme_otron import img_factory
|
|
from meme_otron import meme_db
|
|
from meme_otron import utils
|
|
from meme_otron import meme_otron
|
|
|
|
logging.basicConfig(format="[%(asctime)s][%(levelname)s][%(module)s] %(message)s", level=logging.WARNING)
|
|
|
|
img_factory.load_fonts()
|
|
meme_db.load_memes()
|
|
|
|
templates_dir = utils.relative_path(__file__, "templates")
|
|
preview_dir = utils.relative_path(__file__, "preview")
|
|
doc_template_file = utils.relative_path(__file__, "README-template.md")
|
|
doc_file = utils.relative_path(__file__, "README.md")
|
|
|
|
COLUMNS = 3
|
|
IMG_HEIGHT = 400
|
|
|
|
|
|
def main():
|
|
make_empty(templates_dir)
|
|
make_empty(preview_dir)
|
|
|
|
with open(doc_template_file, mode='r') as f:
|
|
content = "".join(f.readlines())
|
|
|
|
full_list = sorted(meme_db.LIST)
|
|
template_list = [meme_id for meme_id in full_list if len(meme_db.get_meme(meme_id).texts) > 0]
|
|
reaction_list = [meme_id for meme_id in full_list if meme_id not in template_list]
|
|
|
|
content = produce_template_list(content, "LIST1", template_list)
|
|
content = produce_template_list(content, "LIST2", reaction_list)
|
|
|
|
content = produce_example(content, "EXAMPLE1", "example1.jpg", "",
|
|
"brain3",
|
|
"Making memes using an image editor",
|
|
"Making memes using a Python script",
|
|
"Making memes using a Discord bot")
|
|
|
|
content = produce_example(content, "EXAMPLE2", "example2.jpg",
|
|
"The 5th text is not set and the 3rd is explicitly set to empty",
|
|
"see_that_guy",
|
|
"See that guy over there?",
|
|
"He uses an image editor to make memes",
|
|
"",
|
|
"meme-otron's dev")
|
|
|
|
content = produce_example(content, "EXAMPLE3", "example3.jpg",
|
|
"Note how texts make paragraphs",
|
|
"text",
|
|
"*Meme has a 'made with meme-otron' watermark*",
|
|
"reddit: ...",
|
|
"9gag: ...",
|
|
"meme-otron's dev:",
|
|
"-",
|
|
"culture",
|
|
"meme otron")
|
|
|
|
content = produce_example(content, "EXAMPLE4", "example4.jpg",
|
|
"",
|
|
"image",
|
|
"https://i.imgur.com/DNLFUuK.png",
|
|
"-",
|
|
"text",
|
|
"meme-otron's dev close to finishing the idea",
|
|
"-",
|
|
"holup")
|
|
|
|
with open(doc_file, mode='w') as f:
|
|
f.write(content)
|
|
|
|
|
|
def make_empty(target_dir: str):
|
|
if path.exists(target_dir):
|
|
for file in os.listdir(target_dir):
|
|
if path.isfile(path.join(target_dir, file)):
|
|
os.unlink(path.join(target_dir, file))
|
|
else:
|
|
os.mkdir(target_dir)
|
|
|
|
|
|
def produce_template_list(content: str, tag: str, id_list: List[str]):
|
|
if len(id_list) == 0:
|
|
return content
|
|
doc_content = "|" * (COLUMNS + 1) \
|
|
+ "\n|" + ":---:|" * COLUMNS
|
|
info_line = None
|
|
img_line = None
|
|
i = None
|
|
for i, meme_id in enumerate(id_list):
|
|
meme = meme_db.get_meme(meme_id)
|
|
img = img_factory.build_from_template(meme.template, meme.texts, debug=True)
|
|
if img is not None:
|
|
base = True
|
|
if len(meme.texts) > 0:
|
|
base = False
|
|
image_path = path.join(templates_dir, meme.template)
|
|
img.save(image_path)
|
|
size = (round(img.size[0] * IMG_HEIGHT / img.size[1]), IMG_HEIGHT)
|
|
img2 = img.resize(size, resample=PIL.Image.LANCZOS)
|
|
img2.save(path.join(preview_dir, meme.template))
|
|
if i % COLUMNS == 0:
|
|
if info_line is not None and img_line is not None:
|
|
doc_content += info_line + img_line
|
|
info_line = "\n|"
|
|
img_line = "\n|"
|
|
info_line += f"**{meme_id}**"
|
|
if len(meme.aliases) > 0:
|
|
info_line += f"<br>alt: {', '.join(meme.aliases)}"
|
|
if meme.info is not None:
|
|
info_line += f"<br><a href='{meme.info}' target='_blank'>more info</a>"
|
|
info_line += "|"
|
|
if base:
|
|
img_line += f"<a href='../templates/{meme.template}' target='_blank'>"
|
|
else:
|
|
img_line += f"<a href='./templates/{meme.template}' target='_blank'>"
|
|
img_line += f"<img alt='enlarge' src='./preview/{meme.template}'/>" \
|
|
f"</a>|"
|
|
print(i, meme_id)
|
|
info_line += "|" * (COLUMNS - (i % COLUMNS))
|
|
img_line += "|" * (COLUMNS - (i % COLUMNS))
|
|
doc_content += info_line + img_line
|
|
return inject_content(doc_content, content, tag)
|
|
|
|
|
|
def produce_example(content: str, tag: str, file_name: str, note: str, *args: str):
|
|
doc_content = f"> {note}\n\n" \
|
|
"```\n" + \
|
|
" \n".join(['"' + a + '"' if ' ' in a or len(a) == 0 else a for a in args]) + \
|
|
"\n```\n\n" \
|
|
f""
|
|
img, err = meme_otron.compute(*args)
|
|
if img is not None:
|
|
img.save(utils.relative_path(__file__, file_name))
|
|
return inject_content(doc_content, content, tag)
|
|
|
|
|
|
def inject_content(new_content, content, tag):
|
|
start_str = f"<!--{tag}-START-->"
|
|
end_str = f"<!--{tag}-END-->"
|
|
i0 = content.index(start_str)
|
|
i1 = content.index(end_str) + len(end_str)
|
|
return content[:i0] + start_str + "\n" + new_content + "\n" + end_str + content[i1:]
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|