# -*- coding: utf-8 -*-
# Created on 2018/3/20
import base64
import random
import os
import sys
import time

from PIL import Image, ImageFont, ImageDraw

reload(sys)
sys.setdefaultencoding('utf8')

BASE_PATH = "E:\\MyWork\\qingwa\\5\\" # 底圖所在路徑
TMP_PATH = "E:\\MyWork\\qingwa\\5\\tmp\\" # 生成圖片緩存路徑
font_size = 35


# 216 194 119

class MyCar:

 def __init__(self, name):
  self.name = name
  self.name_append = "的氣質適合開"
  if not os.path.exists(TMP_PATH):
   os.mkdir(TMP_PATH)
  self.end_path = TMP_PATH + str(int(time.time())) + str(random.randint(100, 999)) + ".png" # 圖片處理完之后保存的文件名
  self.data = "" # base64數據初始化
  self.pic_handle()
  self.base_64()

 def base_64(self):
  """
  將圖片讀成base64的格式，返回給移動端渲染
  :return:
  """
  res = open(self.end_path, 'rb')
  base64_data = base64.b64encode(res.read())
  res.close()
  d = {
   'image': 'data:image/jpg;base64,' + base64_data
  }
  self.data = d

 def pic_handle(self):
  # 底圖路徑
  img_path = BASE_PATH + str(random.randint(1, 8)) + ".jpg"
  # 底圖的操作對象
  font_img = Image.open(img_path).convert("RGBA")
  # 即將在該底圖上寫字
  draw = ImageDraw.Draw(font_img)
  # 畫筆
  name_font = ImageFont.truetype("wryh.TTF", size=35)
  # 即將寫的字
  name = self.name + self.name_append
  # 底圖的寬高
  w, h = font_img.size
  # 寫在底圖上的區域，計算字符串的長度，讓它寬度居中（高度居中 同理）
  # name_loaction分別指寬高，圖片左上角為(0,0)坐標
  # 寫字，fill為字體顏色，RGB值
  # try except 避免字符串編碼的問題(unicode編碼 再次轉換會報錯)
  try:
   name_location = (((w - len(unicode(name, "UTF-8")) * font_size) / 2), 76)
   draw.text(name_location, unicode(name, "UTF-8"), fill=(216, 194, 119), font=name_font)
  except BaseException as e:
   print e.message
   name_location = (((w - len(name) * font_size) / 2), 76)
   draw.text(name_location, name, fill=(216, 194, 119), font=name_font)
  print self.end_path
  # 保存處理好的圖片
  font_img.save(self.end_path)
  # 顯示圖片
  font_img.show()

 def end_data(self):
  return self.data


if __name__ == '__main__':
 my_car = MyCar("測試")
 return_data = my_car.end_data()
 print type(return_data)