Advertisement
l1von

asdsd

Dec 15th, 2020
20
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 2.10 KB | None | 0 0
  1. @bot.on.chat_message(text=["/мем", "/meme"])
  2. async def meme_gen(messaga: Message):
  3. text_length = len(message.text)
  4. chat_path = f"{dir_to_txt}{message.peer_id}.txt"
  5. picture_path = f"{dir_to_pic}{message.peer_id}.txt"
  6.  
  7. with open(f'meme.jpg') as f:
  8. with open(chat_path, encoding="utf8") as f:
  9. text_lines = len(f.readlines())
  10.  
  11. if text_lines >= 10 and pic_count >= 1:
  12. print("ГЕНЕРИРУЕМ ПО КОМАНДЕ")
  13. with open(chat_path, encoding="utf8") as file:
  14. texts = file.read().splitlines()
  15. random_text = random.choice(texts)
  16. random_bottom_text = random.choice(texts) + " " + random.choice(texts)
  17.  
  18. random_filename = f"randomimg_{random.randint(0, 10000000000000000000000000)}.jpg"
  19. with open(random_filename, "wb") as f:
  20. f.write(response.content)
  21.  
  22. random_text_len = len(random_text)
  23.  
  24. dem_filename = f"result_{random.randint(0, 10000000000000000000000000)}.jpg"
  25. if random_text_len <= 50:
  26. dem = demcreate(random_text, random_bottom_text)
  27. dem.makeImage(random_filename.rstrip(".jpg"))
  28. add_watermark_and_rename(result_filename=dem_filename)
  29. photo = await photo_uploader.upload_message_photo(dem_filename)
  30. await message(attachment=photo)
  31. os.remove(dem_filename)
  32. os.remove(random_filename)
  33. return
  34. else:
  35. random_bottom_text_short = random_text[:50]
  36. dem = demcreate(random_text, random_bottom_text_short)
  37. dem.makeImage("randomimg")
  38. add_watermark_and_rename(result_filename=dem_filename)
  39. photo = await photo_uploader.upload_message_photo(dem_filename)
  40. await message(attachment=photo)
  41. os.remove(dem_filename)
  42. os.remove(random_filename)
  43. return
  44. else:
  45. await message(
  46. "Недостаточно сообщений для генерации.\nНужное количество: 10 сообщений и 1 фотография"
  47. )
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement