juni/aikaterna-cogs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Formatting

Browse Source
This commit is contained in:
aikaterna
2017-09-13 20:30:08 -07:00
parent b5457460d9
commit 5e599aa2e3
1 changed files with 123 additions and 129 deletions
Download Patch File Download Diff File Expand all files Collapse all files

252
imgwelcome/imgwelcome.py
View File

@@ -3,7 +3,7 @@
# https://github.com/AznStevy/Maybe-Useful-Cogs
# imgwelcomeset_upload is based on code in orels' drawing.py
# https://github.com/orels1/ORELS-Cogs
# Parts of createWelcomeImage and on_member_join are from the Welcomer bot:
# Parts of _create_welcome and on_member_join are from the Welcomer bot:
# https://discordbots.org/bot/330416853971107840
import asyncio
import aiohttp
@@ -23,9 +23,9 @@ from PIL import Image, ImageFont, ImageOps, ImageDraw
default_settings = {"ANNOUNCE": False,
"BACKGROUND": "data/imgwelcome/transparent.png",
"BORDER": [255,255,255,230], "CHANNEL": None,
"OUTLINE":[0,0,0,255], "SERVERTEXT": [255,255,255,230],
"TEXT": [255,255,255,230]}
"BORDER": [255, 255, 255, 230], "CHANNEL": None,
"OUTLINE": [0, 0, 0, 255], "SERVERTEXT": [255, 255, 255, 230],
"TEXT": [255, 255, 255, 230]}
class ImgWelcome:
@@ -34,118 +34,118 @@ class ImgWelcome:
def __init__(self, bot):
self.bot = bot
self.settings = dataIO.load_json('data/imgwelcome/settings.json')
self.version = "0.1.2"
self.version = "0.1.2b"
async def save_settings(self):
dataIO.save_json('data/imgwelcome/settings.json', self.settings)
async def createWelcomeImage(self, member, url):
async def _create_welcome(self, member, url):
server = member.server
defaultFont = ImageFont.truetype("data/imgwelcome/fonts/UniSansHeavy.otf",50)
smallFont = ImageFont.truetype("data/imgwelcome/fonts/UniSansHeavy.otf",20)
italicFont = ImageFont.truetype("data/imgwelcome/fonts/UniSansHeavy.otf",30)
italicFontsmall = ImageFont.truetype("data/imgwelcome/fonts/UniSansHeavy.otf",22)
italicFontsupersmall = ImageFont.truetype("data/imgwelcome/fonts/UniSansHeavy.otf",18)
italicFontmicroscopic = ImageFont.truetype("data/imgwelcome/fonts/UniSansHeavy.otf",12)
Background = Image.open(self.settings[server.id]["BACKGROUND"]).convert('RGBA')
NoProfilePicture = Image.open("data/imgwelcome/noimage.png")
welcome_font = ImageFont.truetype("data/imgwelcome/fonts/UniSansHeavy.otf", 50)
server_font = ImageFont.truetype("data/imgwelcome/fonts/UniSansHeavy.otf", 20)
name_font = ImageFont.truetype("data/imgwelcome/fonts/UniSansHeavy.otf", 30)
name_font_medium = ImageFont.truetype("data/imgwelcome/fonts/UniSansHeavy.otf", 22)
name_font_small = ImageFont.truetype("data/imgwelcome/fonts/UniSansHeavy.otf", 18)
name_font_smallest = ImageFont.truetype("data/imgwelcome/fonts/UniSansHeavy.otf", 12)
background = Image.open(self.settings[server.id]["BACKGROUND"]).convert('RGBA')
no_profile_picture = Image.open("data/imgwelcome/noimage.png")
global WelcomePicture
WelcomePicture = Image.new("RGBA",(500,150))
WelcomePicture = ImageOps.fit(Background,(500,150),centering=(0.5,0.5))
WelcomePicture.paste(Background)
WelcomePicture = WelcomePicture.resize((500,150), Image.NEAREST)
global welcome_picture
welcome_picture = Image.new("RGBA", (500, 150))
welcome_picture = ImageOps.fit(background, (500, 150), centering=(0.5, 0.5))
welcome_picture.paste(background)
welcome_picture = welcome_picture.resize((500, 150), Image.NEAREST)
# Load profile picture and make template
ProfileArea = Image.new("L",(512,512),0)
draw = ImageDraw.Draw(ProfileArea)
draw.ellipse(((0,0),(512,512)),fill=255)
ProfileArea = ProfileArea.resize((128,128), Image.ANTIALIAS)
profile_area = Image.new("L", (512, 512), 0)
draw = ImageDraw.Draw(profile_area)
draw.ellipse(((0, 0), (512, 512)), fill=255)
profile_area = profile_area.resize((128, 128), Image.ANTIALIAS)
try:
url = url.replace('webp?size=1024', 'png')
url = url.replace('gif?size=1024', 'png')
await self.getProfile(url)
ProfilePicture = Image.open('data/imgwelcome/profilepic.png')
await self._get_profile(url)
profile_picture = Image.open('data/imgwelcome/profilepic.png')
except:
ProfilePicture = NoProfilePicture
ProfileAreaOutput = ImageOps.fit(ProfilePicture,(128,128),centering=(0,0))
ProfileAreaOutput.putalpha(ProfileArea)
profile_picture = no_profile_picture
profile_area_output = ImageOps.fit(profile_picture, (128, 128), centering=(0, 0))
profile_area_output.putalpha(profile_area)
bordercolor = tuple(self.settings[member.server.id]["BORDER"])
fontcolor = tuple(self.settings[member.server.id]["TEXT"])
servercolor = tuple(self.settings[member.server.id]["SERVERTEXT"])
textoutline = tuple(self.settings[server.id]["OUTLINE"])
mask = Image.new('L', (512,512), 0)
mask = Image.new('L', (512, 512), 0)
draw_thumb = ImageDraw.Draw(mask)
draw_thumb.ellipse((0,0) + (512,512), fill=255, outline=0)
circle = Image.new("RGBA", (512,512))
draw_thumb.ellipse((0, 0) + (512, 512), fill=255, outline=0)
circle = Image.new("RGBA", (512, 512))
draw_circle = ImageDraw.Draw(circle)
draw_circle.ellipse([0,0,512,512], fill=(bordercolor[0], bordercolor[1], bordercolor[2],180), outline = (255,255,255,250))
circle = circle.resize((136,136), Image.ANTIALIAS)
circle_mask = mask.resize((136,136), Image.ANTIALIAS)
drawtwo = ImageDraw.Draw(WelcomePicture)
CircleAreaOutput = ImageOps.fit(circle,(136,136),centering=(0,0))
WelcomePicture.paste(circle, (7,7), circle_mask)
WelcomePicture.paste(ProfileAreaOutput,(11,11),ProfileAreaOutput)
draw_circle.ellipse([0, 0, 512, 512], fill=(bordercolor[0], bordercolor[1], bordercolor[2], 180), outline=(255, 255, 255, 250))
circle = circle.resize((136, 136), Image.ANTIALIAS)
circle_mask = mask.resize((136, 136), Image.ANTIALIAS)
drawtwo = ImageDraw.Draw(welcome_picture)
welcome_picture.paste(circle, (7, 7), circle_mask)
welcome_picture.paste(profile_area_output, (11, 11), profile_area_output)
# Draw welcome text
uname = (str(member.name) + "#" + str(member.discriminator))
drawtwo.text((149,16),"Welcome",font=defaultFont, fill=(textoutline))
drawtwo.text((151,16),"Welcome",font=defaultFont, fill=(textoutline))
drawtwo.text((150,15),"Welcome",font=defaultFont, fill=(textoutline))
drawtwo.text((150,17),"Welcome",font=defaultFont, fill=(textoutline))
drawtwo.text((150,16),"Welcome",font=defaultFont, fill=(fontcolor))
drawtwo.text((149, 16), "Welcome", font=welcome_font, fill=(textoutline))
drawtwo.text((151, 16), "Welcome", font=welcome_font, fill=(textoutline))
drawtwo.text((150, 15), "Welcome", font=welcome_font, fill=(textoutline))
drawtwo.text((150, 17), "Welcome", font=welcome_font, fill=(textoutline))
drawtwo.text((150, 16), "Welcome", font=welcome_font, fill=(fontcolor))
if len(uname) <= 17:
drawtwo.text((151,63),uname,font=italicFont, fill=(textoutline))
drawtwo.text((153,63),uname,font=italicFont, fill=(textoutline))
drawtwo.text((152,62),uname,font=italicFont, fill=(textoutline))
drawtwo.text((152,64),uname,font=italicFont, fill=(textoutline))
drawtwo.text((152,63),uname,font=italicFont, fill=(fontcolor))
drawtwo.text((151, 63), uname, font=name_font, fill=(textoutline))
drawtwo.text((153, 63), uname, font=name_font, fill=(textoutline))
drawtwo.text((152, 62), uname, font=name_font, fill=(textoutline))
drawtwo.text((152, 64), uname, font=name_font, fill=(textoutline))
drawtwo.text((152, 63), uname, font=name_font, fill=(fontcolor))
if len(uname) > 17:
if len(uname) <= 23:
drawtwo.text((151,66),uname,font=italicFontsmall, fill=(textoutline))
drawtwo.text((153,66),uname,font=italicFontsmall, fill=(textoutline))
drawtwo.text((152,65),uname,font=italicFontsmall, fill=(textoutline))
drawtwo.text((152,67),uname,font=italicFontsmall, fill=(textoutline))
drawtwo.text((152,66),uname,font=italicFontsmall, fill=(fontcolor))
drawtwo.text((151, 66), uname, font=name_font_medium, fill=(textoutline))
drawtwo.text((153, 66), uname, font=name_font_medium, fill=(textoutline))
drawtwo.text((152, 65), uname, font=name_font_medium, fill=(textoutline))
drawtwo.text((152, 67), uname, font=name_font_medium, fill=(textoutline))
drawtwo.text((152, 66), uname, font=name_font_medium, fill=(fontcolor))
if len(uname) >= 24:
if len(uname) <= 32:
drawtwo.text((151,70),uname,font=italicFontsupersmall, fill=(textoutline))
drawtwo.text((153,70),uname,font=italicFontsupersmall, fill=(textoutline))
drawtwo.text((152,69),uname,font=italicFontsupersmall, fill=(textoutline))
drawtwo.text((152,71),uname,font=italicFontsupersmall, fill=(textoutline))
drawtwo.text((152,70),uname,font=italicFontsupersmall, fill=(fontcolor))
drawtwo.text((151, 70), uname, font=name_font_small, fill=(textoutline))
drawtwo.text((153, 70), uname, font=name_font_small, fill=(textoutline))
drawtwo.text((152, 69), uname, font=name_font_small, fill=(textoutline))
drawtwo.text((152, 71), uname, font=name_font_small, fill=(textoutline))
drawtwo.text((152, 70), uname, font=name_font_small, fill=(fontcolor))
if len(uname) >= 33:
drawtwo.text((151,73),uname,font=italicFontmicroscopic, fill=(textoutline))
drawtwo.text((153,73),uname,font=italicFontmicroscopic, fill=(textoutline))
drawtwo.text((152,72),uname,font=italicFontmicroscopic, fill=(textoutline))
drawtwo.text((152,74),uname,font=italicFontmicroscopic, fill=(textoutline))
drawtwo.text((152,73),uname,font=italicFontmicroscopic, fill=(fontcolor))
drawtwo.text((151, 73), uname, font=name_font_smallest, fill=(textoutline))
drawtwo.text((153, 73), uname, font=name_font_smallest, fill=(textoutline))
drawtwo.text((152, 72), uname, font=name_font_smallest, fill=(textoutline))
drawtwo.text((152, 74), uname, font=name_font_smallest, fill=(textoutline))
drawtwo.text((152, 73), uname, font=name_font_smallest, fill=(fontcolor))
member_number = len(member.server.members)
drawtwo.text((151,96),"You are the " + str(member_number) + self.getSuffix(member_number) + " member",font=smallFont, fill=(textoutline))
drawtwo.text((153,96),"You are the " + str(member_number) + self.getSuffix(member_number) + " member",font=smallFont, fill=(textoutline))
drawtwo.text((152,95),"You are the " + str(member_number) + self.getSuffix(member_number) + " member",font=smallFont, fill=(textoutline))
drawtwo.text((152,97),"You are the " + str(member_number) + self.getSuffix(member_number) + " member",font=smallFont, fill=(textoutline))
drawtwo.text((152,96),"You are the " + str(member_number) + self.getSuffix(member_number) + " member",font=smallFont, fill=(servercolor))
drawtwo.text((151,116),"of " + str(member.server.name) + "!",font=smallFont, fill=(textoutline))
drawtwo.text((153,116),"of " + str(member.server.name) + "!",font=smallFont, fill=(textoutline))
drawtwo.text((152,115),"of " + str(member.server.name) + "!",font=smallFont, fill=(textoutline))
drawtwo.text((152,117),"of " + str(member.server.name) + "!",font=smallFont, fill=(textoutline))
drawtwo.text((152,116),"of " + str(member.server.name) + "!",font=smallFont, fill=(servercolor))
# Export
ImageObject = BytesIO()
WelcomePicture.save(ImageObject, format="PNG")
ImageObject.seek(0)
return ImageObject
drawtwo.text((151, 96), "You are the " + str(member_number) + self._get_suffix(member_number) + " member", font=server_font, fill=(textoutline))
drawtwo.text((153, 96), "You are the " + str(member_number) + self._get_suffix(member_number) + " member", font=server_font, fill=(textoutline))
drawtwo.text((152, 95), "You are the " + str(member_number) + self._get_suffix(member_number) + " member", font=server_font, fill=(textoutline))
drawtwo.text((152, 97), "You are the " + str(member_number) + self._get_suffix(member_number) + " member", font=server_font, fill=(textoutline))
drawtwo.text((152, 96), "You are the " + str(member_number) + self._get_suffix(member_number) + " member", font=server_font, fill=(servercolor))
drawtwo.text((151, 116), "of " + str(member.server.name) + "!", font=server_font, fill=(textoutline))
drawtwo.text((153, 116), "of " + str(member.server.name) + "!", font=server_font, fill=(textoutline))
drawtwo.text((152, 115), "of " + str(member.server.name) + "!", font=server_font, fill=(textoutline))
drawtwo.text((152, 117), "of " + str(member.server.name) + "!", font=server_font, fill=(textoutline))
drawtwo.text((152, 116), "of " + str(member.server.name) + "!", font=server_font, fill=(servercolor))
image_object = BytesIO()
welcome_picture.save(image_object, format="PNG")
image_object.seek(0)
return image_object
def _rgb_to_hex(self, rgb):
rgb = tuple(rgb[:3])
return '#%02x%02x%02x' % rgb
def _hex_to_rgb(self, hex_num: str, a:int):
def _hex_to_rgb(self, hex_num: str, a: int):
h = hex_num.lstrip('#')
# if only 3 characters are given
@@ -153,24 +153,24 @@ class ImgWelcome:
expand = ''.join([x*2 for x in str(h)])
h = expand
colors = [int(h[i:i+2], 16) for i in (0, 2 ,4)]
colors = [int(h[i:i+2], 16) for i in (0, 2, 4)]
colors.append(a)
return tuple(colors)
def _is_hex(self, color:str):
def _is_hex(self, color: str):
if color is not None and len(color) != 4 and len(color) != 7:
return False
reg_ex = r'^#(?:[0-9a-fA-F]{3}){1,2}$'
return re.search(reg_ex, str(color))
async def getProfile(self, url):
async def _get_profile(self, url):
async with aiohttp.get(url) as r:
image = await r.content.read()
with open('data/imgwelcome/profilepic.png','wb') as f:
with open('data/imgwelcome/profilepic.png', 'wb') as f:
f.write(image)
def getSuffix(self, num):
def _get_suffix(self, num):
num = str(num)
last = num[len(num)-1:len(num)]
if last == "1":
@@ -182,7 +182,7 @@ class ImgWelcome:
else:
return "th"
async def data_check(self, ctx):
async def _data_check(self, ctx):
server = ctx.message.server
if server.id not in self.settings:
self.settings[server.id] = deepcopy(default_settings)
@@ -194,7 +194,7 @@ class ImgWelcome:
await self.save_settings()
if "OUTLINE" not in self.settings[server.id].keys():
self.settings[server.id]["OUTLINE"] = [0,0,0,255]
self.settings[server.id]["OUTLINE"] = [0, 0, 0, 255]
await self.save_settings()
@checks.admin_or_permissions(manage_server=True)
@@ -208,14 +208,14 @@ class ImgWelcome:
@imgwelcome.command(pass_context=True, name="border", no_pm=True)
async def imgwelcome_border(self, ctx, bordercolor=None):
"""Set the profile image border color.
Use hex codes for colors and clear for transparent."""
Use hex codes for colors and ‘clear’ for transparent."""
server = ctx.message.server
await self.data_check(ctx)
await self._data_check(ctx)
default_a = 230
valid = True
if bordercolor == "clear":
self.settings[server.id]["BORDER"] = [0,0,0,0]
self.settings[server.id]["BORDER"] = [0, 0, 0, 0]
elif self._is_hex(bordercolor):
self.settings[server.id]["BORDER"] = self._hex_to_rgb(bordercolor, default_a)
else:
@@ -223,14 +223,14 @@ class ImgWelcome:
valid = False
if valid:
await self.bot.say('The profile color has been set.')
await self.bot.say('The profile border color has been set.')
await self.save_settings()
@imgwelcome.command(pass_context=True, name="text", no_pm=True)
async def imgwelcome_text(self, ctx, textcolor:str, servercolor:str):
async def imgwelcome_text(self, ctx, textcolor: str, servercolor: str):
"""Set text colors. Use hex code for colors."""
server = ctx.message.server
await self.data_check(ctx)
await self._data_check(ctx)
default_a = 230
valid = True
@@ -257,7 +257,7 @@ class ImgWelcome:
if not server.me.permissions_in(channel).send_messages:
await self.bot.say("No permissions to speak in that channel.")
return
await self.data_check(ctx)
await self._data_check(ctx)
self.settings[server.id]["CHANNEL"] = channel.id
await self.save_settings()
await self.bot.send_message(channel, "This channel will be used for welcome messages.")
@@ -266,7 +266,7 @@ class ImgWelcome:
async def imgwelcome_clear(self, ctx):
"""Set the background to transparent."""
server = ctx.message.server
await self.data_check(ctx)
await self._data_check(ctx)
self.settings[server.id]['BACKGROUND'] = 'data/imgwelcome/transparent.png'
await self.save_settings()
await self.bot.say('Welcome image background is now transparent.')
@@ -275,13 +275,13 @@ class ImgWelcome:
async def imgwelcome_outline(self, ctx, outline=None):
"""Set the text outline. White or black."""
server = ctx.message.server
await self.data_check(ctx)
await self._data_check(ctx)
valid = True
if outline == "white":
self.settings[server.id]["OUTLINE"] = [255,255,255,255]
self.settings[server.id]["OUTLINE"] = [255, 255, 255, 255]
await self.save_settings()
elif outline == "black":
self.settings[server.id]["OUTLINE"] = [0,0,0,255]
self.settings[server.id]["OUTLINE"] = [0, 0, 0, 255]
await self.save_settings()
else:
await self.bot.say('Outline color is invalid. Use clear or black.')
@@ -290,25 +290,24 @@ class ImgWelcome:
if valid:
await self.bot.say('The text outline has been set.')
@imgwelcome.command(name="preview", pass_context=True)
async def imagewelcome_preview(self, ctx, member:discord.Member=None):
@imgwelcome.command(name="preview", pass_context=True, no_pm=True)
async def imagewelcome_preview(self, ctx, member: discord.Member=None):
"""Show a welcome image with the current settings."""
server = ctx.message.server
author = ctx.message.author
channel = ctx.message.channel
if member is None:
member = ctx.message.author
await self.data_check(ctx)
channelobj = self.bot.get_channel(channel.id)
await self.bot.send_typing(channelobj)
ImageObject = await self.createWelcomeImage(member, member.avatar_url)
await self.bot.send_file(channelobj,ImageObject,filename="welcome.png")
await self._data_check(ctx)
channel_object = self.bot.get_channel(channel.id)
await self.bot.send_typing(channel_object)
image_object = await self._create_welcome(member, member.avatar_url)
await self.bot.send_file(channel_object, image_object, filename="welcome.png")
@imgwelcome.command(pass_context=True, name="toggle", no_pm=True)
async def imgwelcome_toggle(self, ctx):
"""Toggle welcome messages on the server."""
server = ctx.message.server
await self.data_check(ctx)
await self._data_check(ctx)
self.settings[server.id]["ANNOUNCE"] = not self.settings[server.id]["ANNOUNCE"]
if self.settings[server.id]["ANNOUNCE"]:
await self.bot.say("Now welcoming new users.")
@@ -320,13 +319,11 @@ class ImgWelcome:
async def imgwelcome_upload(self, ctx, default=None):
"""Upload a background through Discord. 500px x 150px.
This must be an image file and not a url."""
# request image from user
server = ctx.message.server
await self.data_check(ctx)
await self._data_check(ctx)
await self.bot.say("Please send the file to use as a background. File must be 500px x 150px.")
answer = await self.bot.wait_for_message(timeout=30, author=ctx.message.author)
# get the image from message
try:
bg_url = answer.attachments[0]["url"]
success = True
@@ -337,8 +334,6 @@ class ImgWelcome:
serverimage = Image
if success:
# download the image
try:
async with aiohttp.get(bg_url) as r:
image = await r.content.read()
@@ -353,17 +348,15 @@ class ImgWelcome:
except Exception as e:
success = False
print(e)
if success:
# check dimensions
if serverimage.size == (500,150):
if serverimage.size == (500, 150):
self.settings[server.id]['BACKGROUND'] = "data/imgwelcome/" + ctx.message.server.id + "/serverbg.png"
await self.save_settings()
else:
await self.bot.say("Image needs to be 500x150.")
return
backgroundimg = ('data/imgwelcome/{}/serverbg.png'.format(server.id))
self.settings[server.id]['BACKGROUND'] = (backgroundimg)
background_img = ('data/imgwelcome/{}/serverbg.png'.format(server.id))
self.settings[server.id]['BACKGROUND'] = (background_img)
await self.save_settings()
await self.bot.say('Welcome image for this server set to uploaded file.')
else:
@@ -384,19 +377,20 @@ class ImgWelcome:
if not self.settings[server.id]["ANNOUNCE"]:
return
channelid = self.settings[server.id]["CHANNEL"]
channelobj = self.bot.get_channel(channelid)
await self.bot.send_typing(channelobj)
ImageObject = await self.createWelcomeImage(member, member.avatar_url)
await self.bot.send_file(channelobj,ImageObject,filename="welcome.png")
channel_object = self.bot.get_channel(channelid)
await self.bot.send_typing(channel_object)
image_object = await self._create_welcome(member, member.avatar_url)
await self.bot.send_file(channel_object, image_object, filename="welcome.png")
if (len(member.server.members) % 100) == 0 or (len(member.server.members) == 1337):
msg = "\N{PARTY POPPER} Thanks <@" + member.id + ">, you're the ***" + str(len(member.server.members)) + "*** th user on this server! \N{PARTY POPPER}"
await self.bot.send_message(channelobj,msg)
joinDate = datetime.datetime.strptime(str(member.created_at),"%Y-%m-%d %H:%M:%S.%f")
currentDate = datetime.datetime.now(datetime.timezone.utc)
currentDate = currentDate.replace(tzinfo=None)
timeSinceJoining = currentDate - joinDate
if timeSinceJoining.days < 7:
await self.bot.send_message(channelobj,"\N{WARNING SIGN} This account was created less than a week ago (" + str(timeSinceJoining.days) + " days ago)")
await self.bot.send_message(channel_object, msg)
date_join = datetime.datetime.strptime(str(member.created_at), "%Y-%m-%d %H:%M:%S.%f")
date_now = datetime.datetime.now(datetime.timezone.utc)
date_now = date_now.replace(tzinfo=None)
since_join = date_now - date_join
if since_join.days < 7:
await self.bot.send_message(channel_object, "\N{WARNING SIGN} This account was created less than a week ago (" + str(since_join.days) + " days ago)")
def check_folders():
if not os.path.exists('data/imgwelcome/'):