sakura/manifest/manifest.py

# coding=utf-8
'''
Created on Apr 23, 2018
Desc: Webp convertor
@author: Mashiro https://2heng.xin
'''
import os
import sys
import json
import requests
import base64
import hashlib
from Crypto.PublicKey import RSA
from PIL import Image

class Single(object):
  def __init__(self, file, manifest):
    self.file = file
    self.mani = manifest

  def hash(self):
    hasher = hashlib.md5()
    with open('gallary/' + self.file, 'rb') as afile:
      buf = afile.read()
      hasher.update(buf)
    self.hash = hasher.hexdigest()
    self.jpeg = 'jpeg/' + self.hash + '.jpeg'
    self.webp = 'webp/' + self.hash + '.webp'
    self.jpeg_th = 'jpeg/' + self.hash + '.th.jpeg'
    self.webp_th = 'webp/' + self.hash + '.th.webp'

  def optimize(self):
    im = Image.open('gallary/' + self.file).convert('RGB')
    im.save(self.jpeg, 'JPEG') # todo: TinyPNG API
    im.save(self.webp, 'WEBP')
    im.thumbnail((450, 300))
    im.save(self.jpeg_th, 'JPEG')  # todo: TinyPNG API
    im.save(self.webp_th, 'WEBP')

  def manifest(self):
    self.mani[self.hash] = {
      'source': self.file,
      'jpeg': [self.jpeg, self.jpeg_th],
      'webp': [self.webp, self.webp_th]
    }

  def main(self):
    self.hash()
    # if os.path.exists(self.jpeg) and os.path.exists(self.webp):
    self.optimize()
    self.manifest()
    return self.mani


class Upload2Wordpress:
  def __init__(self, username, password, url):
    self.username = username
    self.password = password
    self.url = url

  def upload(self, file, field):
    data_string = self.username + ':' + self.password
    token = base64.b64encode(data_string.encode()).decode('utf-8')
    headers = {
      'Authorization': 'Basic ' + token,
      "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/78.0.3904.97"
    }
    files = {field: open(file, mode="rb")}
    reply = requests.post(self.url, headers=headers, files=files)
    print(json.loads(reply.content)['message'])

  def upload_manifest(self):
    print('start uploading `manifest.json`...')
    self.upload('manifest.json', 'manifest')

  def upload_key(self):
    print('start uploading `private.key`...')
    self.upload('private.key', 'rsa')
    print('start uploading `public.key`...')
    self.upload('public.key', 'rsa')

  def main(self):
    self.upload_manifest()
    self.upload_key()


def gen_manifest_json():
  onlyfiles = [f for f in os.listdir('gallary') if os.path.isfile(os.path.join('gallary', f))]
  id = 1
  Manifest = {}
  for f in onlyfiles:
    worker = Single(f, Manifest)
    Manifest = worker.main()
    print(str(id) + '/' + str(len(onlyfiles)))
    id += 1
  with open('manifest.json', 'w+') as json_file:
    json.dump(Manifest, json_file)


def gen_key_pairs():
  key = RSA.generate(1024)
  pv_key_string = key.exportKey()
  with open("private.key", "w+") as prv_file:
    print("{}".format(pv_key_string.decode()), file=prv_file)
  pb_key_string = key.publickey().exportKey()
  with open("public.key", "w+") as pub_file:
    print("{}".format(pb_key_string.decode()), file=pub_file)


def main():
  gen_manifest_json()
  if not os.path.exists("public.key") or not os.path.exists("private.key"):
    print("start generating key pairs...")
    gen_key_pairs()
  username = input('Enter your username: ')
  password = input('Enter your password: ')
  url = input('Enter your rest api url: ')
  upload = Upload2Wordpress(username, password, url)
  upload.main()


if __name__ == '__main__':
  main()
  key = input('`manifest.json` saved. Press any key to quit.')
  quit()