# coding: utf-8 import os from datetime import datetime from PIL import Image,ImageFilter,ImageFont,ImageDraw import matplotlib.pyplot as plt import math from PIL import Image, ImageDraw, ImageFont import textwrap import random post_img = "res_post.jpg" # 底图，确定尺寸 # 最佳实践 # lines = textwrap.wrap(text, width=40) # y_text = h # for line in lines: # width, height = font.getsize(line) # draw.text(((w - width) / 2, y_text), line, font=font, fill=FOREGROUND) # y_text += height # 将文字居中显示 def longTextCut(text, line_words): """cut the text to len / given_num lines""" total_len = len(text) num_line = math.ceil(total_len/line_words) num_line = max(num_line, 1) res = [] for i in range(num_line): start_inx = int(i*line_words) end_inx = int(line_words* (i+1)) sub_seq = text[start_inx : end_inx] res.append(sub_seq) lines = textwrap.wrap(text, width=line_words) return res return lines # lines = textwrap.wrap(text, width=10) # w文字变形，每行不超过多少字，方便阅读 def reshape_texts(lines, line_words): new_lines = [] for it in lines: sub_lines = it.split("\n") split_lines = [] for inx, sub_it in enumerate(sub_lines): if len(sub_it) > line_words: relen_text = longTextCut(sub_it, line_words) split_lines.extend(relen_text) else: split_lines.append(sub_it) new_lines.extend(split_lines) return new_lines # w文字变形，每行不超过多少字，方便阅读 def reshape_texts_end(lines, line_words): new_lines = [] for it in lines: sub_lines = it.split("。") # t_ = [len(i) for i in sub_lines] # y_ = [i > line_words for i in t_] split_lines = [] for inx, sub_it in enumerate(sub_lines): if len(sub_it) > line_words: relen_text = longTextCut(sub_it, line_words) split_lines.extend(relen_text) else: split_lines.append(sub_it) new_lines.extend(split_lines) return new_lines #高斯模糊类 class MyGaussianBlur(ImageFilter.Filter): name = "GaussianBlur" def __init__(self, radius=2, bounds=None): self.radius = radius self.bounds = bounds def filter(self, image): if self.bounds: clips = image.crop(self.bounds).gaussian_blur(self.radius) image.paste(clips, self.bounds) return image else: return image.gaussian_blur(self.radius) # self.background_url=get_bing_image() #卡片生成类 16:9 class myPost: def __init__(self, front_img=None, img = r'./broadcast/kaobianBottem.jpeg' ): self.frontground = Image.open(front_img) if front_img else None # bing img # if type(image) is str: # return image # elif type(image) is Image.Image: # return encode_pil_to_base64(image).decode() self.img= Image.open(img) if type(img) is str else Image.fromarray(img) #底稿，确定尺寸 self.icon=None # self.icon=Image.open(self.icon_url).convert("RGBA") #纵向间距 # self.spacing=spacing self.width, self.height=self.img.size #加载小图 def loadIcon(self, point=(50,50)): """point=(x,y)""" pass # self.img.paste(self.icon,point,self.icon) def get_width_height(self): return self.width,self.height def getLinesCount(self, sz, spacing): words = int(self.width / sz) - 2 rows = int( self.height / (sz + spacing)) return (words, rows) #高斯模糊图片作为背景 def drawBlur(self): backflur = self.frontground.resize((self.width,self.height), resample=3).filter(MyGaussianBlur(radius=35)) self.img.paste(backflur,(0, 0)) #添加前景图片 def drawFrontground(self): """添加一个前景图壁纸，高斯虚化 """ if not self.frontground: return x=0 y=int(self.height*2/3.3) # 在下半部分画这个图 srcwidth,srcheight=self.frontground.size height=int(srcheight * self.width / srcwidth) #重设图片尺寸 frontground = self.frontground.resize((self.width, height),Image.Resampling.LANCZOS) alpha_layer = Image.new('L', (self.width, height), 0) draw = ImageDraw.Draw(alpha_layer) sz = (self.width/2-100, 0, self.width-30, height) # print("the area is :", sz) draw.ellipse(sz, fill=150) self.img.paste(frontground,(x,y),alpha_layer) def get_res(self): return self.img def postTextLine(self, title="Text--test", font="./simsun.ttc", sz=50, x=40, y=90, color = "red"): """main title""" draw = ImageDraw.Draw(self.img) font = ImageFont.truetype(font=font, size=sz) draw.text((x,y), title, font=font, fill=color) def postBoxText(self, texts=["Text--test\n sub text"], font="./simsun.ttc", sz=50, x=40, y=390, color = "red", spacing=40, ali = "left", bac_color=None): """lines""" # 定义文本内容 draw = ImageDraw.Draw(self.img) font = ImageFont.truetype(font=font, size=sz) text = "\n".join(texts) bbox = draw.multiline_textbbox((0, 0), text, font=font, spacing=spacing) # 计算文本的宽度和高度 width = bbox[2] - bbox[0] width += int(width * 0.3) height = bbox[3] - bbox[1] height += int(height * 0.3) if bac_color is not None: draw.rectangle((x-int(width * 0.15), y-int(height * 0.15), x + width, y + height), fill=bac_color) # 计算文本的位置 # x = (self.width - width) / 2 # y = (remain_height - height) / 2 # 在剩下的高度中居中显示 # y = 0 + spacing * 2 # 从头开始展示 # 绘制文本 draw.multiline_text((x, y), text, fill=color, font=font, align=ali, spacing=spacing) # 放置文字 def postText(self, texts=["Text--test\n sub text"], font="./simsun.ttc", sz=50, x=40, y=390, color = "red", spacing=20): draw = ImageDraw.Draw(self.img) # Set the font, text contents, and line spacing font = ImageFont.truetype(font=font, size=sz) # 定义文本内容 text = "\n".join(texts) draw.multiline_text((x, y), text, fill=color, font=font, align="left", spacing=spacing) # 获取文本的边界框 # bbox = draw.multiline_textbbox((0, 0), text, font=font, spacing=spacing) # 绘制文本 # draw.text((x,title_y), title, font=title_front, fill="red") # draw.multiline_text((x, y), text, fill=color, font=font, align="left", spacing=spacing) def drawCard(self): if frontground: self.drawFrontground() self.postText(title, texts, sz = 40) def drawCardExample(self): self.drawFrontground() self.postTextLine(title="Text--test", font="./simsun.ttc", sz=size, x=40, y=90, color = "red") self.postTextLine(title="content of sub text", font="./simsun.ttc", sz=size, x=40, y=90, color = "green") def show(self): pass #保存到本地 def saveCard(self, path_name): save_name = path_name # save_name = str(datetime.today()).split()[0]+'.png' self.img.save(save_name) return save_name def generatePost(lines = [""], front_img=None, bac_img="./kaobianBottem.jpeg", fn_name = "test_v1.png"): """给定背景，前景，文字，size，生成单张图, 并环肥内存中的图对象""" postcard=myPost(front_img=front_img, img = bac_img) line_words, rows = postcard.getLinesCount(sz, spacing) texts = reshape_texts(lines, line_words) print("总行数",len(texts)) postcard.drawFrontground() postcard.postText(texts[0], texts[1:], sz = 40) save_name=postcard.saveCard(fn_name) # ---opend image generated_image = postcard.get_res() return generated_image if __name__ == '__main__': # 输入 -======================== title = "大王叫我来巡山并发布公告，解决就业问题山中有老虎猴子承办问" lines = ['四、报名及网上缴费','本次公开考试招聘采取网络报名方式进行，不组织现场报名。报名时间：2023年3月10日至3月17日24:00。', '报名网站：广元人事考试网（http://gypta.e21cn.com/）。'] # text = doc['fee_sj'] # lines = content_with_date(text) # 只需要有日期信息的字符串 # texts = "\n".join(lines) result_post_file = "post_v1.png" bac_img = "./kaobianBottem.jpeg" # front_img = "./bingpost/2023-05-13.png" bing_post = "bingpost/" imgs_ = os.listdir(bing_post) imgs_ = [i for i in imgs_ if i.endswith("png")] ch_img = random.choice(imgs_) front_img = os.path.join(bing_post, ch_img) postcard=myPost(front_img=front_img, img = bac_img) postcard.drawFrontground() # 放置标题 + 字体字号 sz=80 spacing=20 words, rows = postcard.getLinesCount(sz=sz, spacing=spacing) title_text = longTextCut(title, words) title_text = textwrap.wrap(title, width=words) print(title_text) postcard.postText(title_text, font="./simsun.ttc", sz = sz, x=40, y=150, color = "blue", spacing=20) sz=50 spacing=20 words, rows = postcard.getLinesCount(sz=sz, spacing=spacing) texts = longTextCut(lines[1], words) print(texts) postcard.postText(texts, font="./simsun.ttc", sz = sz, x=40, y=490, color = "green", spacing=spacing) texts = reshape_texts(lines, words) print("texts 长文本 总行数",len(texts)) postcard.postText([lines[0]], font="./simsun.ttc", sz = int(sz*1.5),x=40, y=790, color = "red", spacing=spacing) postcard.postText(texts, font="./simsun.ttc", sz = sz,x=40, y=990, color = "green", spacing=spacing) save_name=postcard.saveCard(result_post_file) print(save_name) # postcard.show()