convert/front/api/main.py

# -*- coding: utf-8 -*-
"""
    demeter web page
    name:work.py
    author:rabin
"""
from .__load__ import *

class Common(object):
	@staticmethod
	def signature(self):
		key = self.input('key')
		nonce = self.input('nonce')
		signature = self.input('signature')
		file = self.input('file')

		if not key:
			self.out('参数错误：key')
		if not nonce:
			self.out('参数错误：nonce')
		if not signature:
			self.out('参数错误：signature')
		if not file:
			self.out('参数错误：file')

		site = Demeter.model('site')
		site.key = key

		self.data['site'] = site.select(type='fetchone')

		if not self.data['site']:
			self.out('站点信息不存在')

		key = key + '&' + self.data['site']['token'] + '&' + nonce + '&' + file
		key = Demeter.md5(key)
		if key != signature:
			self.out('验证失败')


# 请求转换 /main/convert 接口必须后端获取，token不允许暴露
class convert_path(Load):
	@Web.setting
	def get(self):

		file = self.input('file')
		self.data = {}
		Common.signature(self)

		service = Demeter.service('convert')
		file = service.update(self.data['site']['id'], self.data['site']['key'], file)

		if not file:
			self.out('未生成文件信息')

		# 推入到redis队列
		if file['status'] == 1:
			redis = Demeter.redis()
			config = Demeter.config['redis']
			redis.rpush(config['name'], file['id'])

		del self.data['site']['token']

		url = self.request.protocol + "://" + self.request.host

		self.data['file'] = {
			'id' : file['id'],
			'status' : file['status'],
			'url' : url + file['url'],
			'ext' : file['ext'],
			'name' : file['name'],
			'key' : file['key']
		}

		self.out('yes', self.data)

# 获取信息 后端接口，token不允许暴露
class get_path(Load):
	@Web.setting
	def get(self):

		file = self.input('file')
		self.data = {}
		Common.signature(self)

		service = Demeter.service('convert')
		file = service.get(self.data['site']['id'], self.data['site']['key'], file);

		if not file:
			self.out('未生成文件信息')

		del self.data['site']['token']

		url = self.request.protocol + "://" + self.request.host

		self.data['file'] = {
			'id' : file['id'],
			'status' : file['status'],
			'url' : url + file['url'],
			'page' : file['page'],
			'ext' : file['ext'],
			'name' : file['name'],
			'key' : file['key']
		}

		self.out('yes', self.data)

# 授权用户可以访问html的接口 后端接口，token不允许暴露 后续实现
class auth_path(Load):
	@Web.setting
	def get(self):
		key = int(self.input('site', 1))
		user = self.input('user')
		token = self.input('token')
		file = self.input('file')
		site = Demeter.model('site')
		site.key = key
		data = {}
		data['site'] = site.select(type='fetchone')

		if not file:
			self.out('错误的文件信息')

		if not data['site']:
			self.out('站点信息不存在')

		if data['site']['token'] != token:
			self.out('验证失败')

		service = Demeter.service('convert')
		data['file'] = service.get(data['site']['id'], key,  file);

		# 授权之后生成一个key，用于前端权限验证

		self.out('yes', data)

# 读取html 带有权限控制 该接口为前端接口 需要有授权接口 后续实现
class view_path(Load):
	@Web.setting
	def get(self):
		key = int(self.input('site', 1))
		token = self.input('token')
		file = self.input('file')
		site = Demeter.model('site')
		site.key = key
		data = {}
		data['site'] = site.select(type='fetchone')

		if not file:
			self.out('错误的文件信息')

		if not data['site']:
			self.out('站点信息不存在')

		if data['site']['token'] != token:
			self.out('验证失败')

		service = Demeter.service('convert')
		data['file'] = service.get(data['site']['id'], key,  file);

		self.out('yes', data)