我需要这样做,并且我认为作为生成的图像,像素将被干净地着色而没有抖动。
import requests
from PIL import Image
image_url = "https://avatars.githubusercontent.com/u/85325807?v=4"
img_data = requests.get(image_url).content
with open("avatar.jpg", "wb") as handler:
handler.write(img_data)
image = Image.open("avatar.jpg")
colour_count = len(set(image.getdata()))
print(image.size, "colours:", colour_count)
if colour_count < 10:
print("this is a gravatar")
这里发生了什么?
我们下载图像,然后将其作为PIL图像加载。然后我们制作像素颜色集,以便只获取唯一的颜色。如果只有2种唯一的颜色,则它是一个gravatar,如果有更多,则是一张照片。(我将其设置为10,以给自己留出一些余地,因为我不想要假阴性。)
您可以使用此技术构建图像的URL。
https://github.com/twbs.png
或者
https://github.com/npm.png?size=200
我如何在现实生活中使用它
在一门大一课程中,我想检查用户是否已更新其照片,以便教学团队可以轻松地将代码库与人员匹配。
def has_real_photo(repo_path):
repo = git.cmd.Git(repo_path)
origin_url = get_origin_url(repo)
owner = origin_url.split("/")[3]
image_url = f"https://github.com/{owner}.png?size=40"
img_data = requests.get(image_url).content
with open("avatar.jpg", "wb") as handler:
handler.write(img_data)
image = Image.open("avatar.jpg")
colour_count = len(set(image.getdata()))
if colour_count > 10:
block_image = blocky_photo(image)
print(block_image)
return True
else:
block_image = blocky_photo(image)
print(
f"Your GitHub profile picture only has {colour_count} colours.\n"
"This makes me think it's the default avatar.\n"
"Not like this:\n",
block_image,
"""Like this:
╭───────────╮
│ !!!!!!! │
│ / \ │
│ │ O O │ │
│<│ v │>│
│ \ ─── / │
│ \____/ │
╰───────────╯\n"""
"Go to https://github.com/settings/profile and upload a photo of your face.\n"
"This really helps us understand who's who and be more useful in tutorials.",
)
return False
def blocky_photo(image):
colour_map_list = list(
zip(
list(set(image.getdata())),
["█", "░", "▒", "▓", "X", "#", "%", "/", ":", "*"],
)
)
colour_map = {x[0]: x[1] for x in colour_map_list}
image = image.resize((20, 10), Image.NEAREST)
pixels = list(image.getdata())
width, height = image.size
block_image = ""
for i in range(len(pixels)):
block_image += colour_map[pixels[i]]
if (i + 1) % (width) == 0:
block_image += "\n "
return block_image
他们有一组测试,可以自行运行,并打印:
Your GitHub profile picture only has 2 colours.
This makes me think it's the default avatar.
Not like this:
████████████████████
█████░░░░░░░░░░█████
█████░░░░░░░░░░█████
█████░░░████░░░█████
█████░░░████░░░█████
█████░░░████░░░█████
██░░░░░░░░░░░░░░░░██
██░░░░░░░░░░░░░░░░██
██░░░░░░████░░░░░░██
████████████████████
Like this:
╭───────────╮
│ !!!!!!! │
│ / \ │
│ │ O O │ │
│<│ v │>│
│ \ ─── / │
│ \____/ │
╰───────────╯
Go to https://github.com/settings/profile and upload a photo of your face.
This really helps us understand who's who and be more useful in tutorials.
✘ You've got a photo for your GitHub account