def get_base64_diff_value(s1, s2):
    base64chars = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/'
    res = 0
    for i in xrange(len(s2)):     #xrange() 函数用法与 range 完全相同，所不同的是生成的不是一个数组，而是一个生成器。
        if s1[i] != s2[i]:
            return abs(base64chars.index(s1[i]) - base64chars.index(s2[i])) #index() 方法查找指定值的首次出现。
    return res


def solve_stego():
    with open('ComeOn!.txt', 'rb') as f:
        file_lines = f.readlines()
        bin_str = ''
        for line in file_lines:
            steg_line = line.replace('\n', '')
            norm_line = line.replace('\n', '').decode('base64').encode('base64').replace('\n', '')
            diff = get_base64_diff_value(steg_line, norm_line)
            print diff
            pads_num = steg_line.count('=')
            if diff:
                bin_str += bin(diff)[2:].zfill(pads_num * 2)       #zfill() 方法返回指定长度的字符串，原字符串右对齐，前面填充0。
            else:
                bin_str += '0' * pads_num * 2
            print goflag(bin_str)


def goflag(bin_str):    #bin()函数用于获取数字的二进制值，接受数字并返回等效的二进制字符。
    res_str = ''
    for i in xrange(0, len(bin_str), 8):    
      #xrange(start, stop[, step])start: 计数从 start 开始。默认是从 0 开始。例如 xrange(5) 等价于 xrange(0， 5)stop: 计数到 stop 			结束，但不包括 stop。例如：xrange(0， 5) 是 [0, 1, 2, 3, 4] 没有 5。step：步长，默认为1。例如：xrange(0， 5) 等价于 xrange(0, 				5, 1)
        res_str += chr(int(bin_str[i:i + 8], 2))
    return res_str


if __name__ == '__main__':
    solve_stego()

# coding=utf-8
import os
#将无后缀的文件加上.jpg
dir_list = os.listdir('./')
#print(dir_list)
for file in dir_list:
    if '.' not in file:
# print(file)
        os.rename(file, file+'.jpg')

# coding=utf-8
# python2

import os
from PIL import Image

#图片压缩后的大小
width_i = 134
height_i = 130

#每行每列显示图片数量
row_max = 6
line_max = 6

all_path = list()
num = 0
pic_max = line_max * row_max

#文件夹路径
dir_name = r"./"


#获取文件夹下文件名的列表
dir_list = os.listdir('./')

#将文件列表重新排序（按文件原始数据末尾的数字）
for i in range(36,0,-1):
    # print(i)
    for file in dir_list:
        if '.jpg' in file:
            f=open(file ,'rb')
            n1 = f.read()
            n2 = n1[-2:]
            # print()
            if i > 9:
                if str(i) in n2:
                    # all_path.append(os.path.join(file))
                    all_path.insert(0,os.path.join(file))
                    # print(all_path)
                    break
            else:
                if 'd9' in n1.encode('hex')[-4:] and str(i) in n2:
                    # print(file)
                    all_path.insert(0,os.path.join(file))
                    break
print(all_path)


toImage = Image.new('RGBA',(width_i*line_max,height_i*row_max))

#拼接图片
for i in range(row_max):
    for j in range(line_max):
        # 每次打开图片绝对路路径列表的第一张图片
        pic_fole_head = Image.open(all_path[num])
        # 获取图片的尺寸
        wihth,height = pic_fole_head.size
        # 按照指定的尺寸，给图片重新赋值，<PIL.Image.Image image mode=RGB size=200x200 at 0x127B7978>
        tmppic = pic_fole_head.resize((width_i, height_i))
        # 计算每个图片的左上角的坐标点(0, 0)，(0, 200)，(0, 400)，(200, 0)，(200, 200)。。。。(400, 400)
        loc = (int(j % line_max * width_i), int(i % line_max * height_i))
        print(loc)
        # print("第{}张图的存放位置".format(num),loc)
        toImage.paste(tmppic, loc)
        num = num + 1

        if num >= len(all_path):
            break
    if num >= pic_max:
        break

# print(toImage.size)
toImage.save('merged.png')