Python 小把戏之图片转字符串

#! /usr/bin/python3
# -*- coding: UTF-8 -*-

from PIL import Image
import argparse
import os
import time
import ctypes
 


#命令行输入参数处理
# description参数可以用于插入描述脚本用途的信息，可以为空
parser = argparse.ArgumentParser(description="your script description")

parser.add_argument('-f', '--file')  			          #输入的第一个参数 图片名   
parser.add_argument('-o', '--output')                     #输入的第二个命令行参数
parser.add_argument('--width', type = int, default = 100)  #输出字符画宽 默认 100
parser.add_argument('--height', type = int, default = 100) #输出字符画高 默认 100

# 将变量以标签-值的字典形式存入args字典
args = parser.parse_args()

IMG = args.file    
WIDTH = args.width
HEIGHT = args.height
OUTPUT = args.output

ascii_char = list("$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:,\"^`'. ")
img_num = 0   #记录处理的图片数量
# 将256灰度映射到70个字符上
def get_char(r,g,b,alpha = 256):
    if alpha == 0:                                   #透明度
        return ' '	
    length = len(ascii_char)                         
    gray = int(0.2126 * r + 0.7152 * g + 0.0722 * b) #RGB色彩转化为灰度

    unit = (256.0 + 1)/length
    return ascii_char[int(gray/unit)]                #灰度转化成字符

#批量文件操作
def batch_rename(path):
	global img_num
	#判断路径是否是目录 是否是文件
	if (not os.path.isdir(path) and not os.path.isfile(path)):
		return False
		
	if os.path.isfile(path):
		#分割出目录与文件名
		file_path = os.path.split(path)
		#分割出文件名与文件的扩展名
		lists = file_path[1].split('.')
		#取出后缀名
		file_ext = lists[-1]
		img_ext = ['bmp', 'jpeg', 'gif', 'psd', 'png', 'jpg']
		if file_ext in img_ext:                            #确保文件是图片
			im = Image.open(path)                          #打开图片文件
			im.show()                                      #显示图片
			im = im.resize((WIDTH,HEIGHT), Image.NEAREST)  #更改图片大小
			txt = ""
			img_num += 1
			for i in range(HEIGHT):
				for j in range(WIDTH):
					txt += get_char(*im.getpixel((j,i)))   #获取对应的字符
					print(get_char(*im.getpixel((j,i))), end = ' ')  
				txt += '\n'
				print(' ')
			with open(lists[0] + '.txt','w') as f:         #写入txt
				f.write(txt)
		
			
	elif os.path.isdir(path):
		for item in os.listdir(path):
			batch_rename(os.path.join(path, item))
if __name__ == '__main__':	
	if IMG:               #如果传入 -f 文件名参数  将该文件转化为字符串
		img_dir = IMG
	else:                 #如果没有传入 -f 文件名参数  默认将当前文件夹下所有图片都转化为字符串
		img_dir = '.'
	start = time.time()
	
	 #字符画输出到文件
	batch_rename(img_dir)
	print('总共处理了 %s 张图片， 耗时 %4.2f' % (img_num, time.time() - start ))