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')
		uid = self.input('uid')

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

		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 + '&' + uid
		key = Demeter.sha1(key)
		if key != signature:
			self.out('验证失败')

		return key

class test_path(Load):
	@Web.setting
	def get(self):
		from wand.image import Image
		filename = Demeter.path + 'runtime/upload/d1c6fd2a4b2941e083471c2ae3d97a37.pdf'

		i = 0
		pdf = Image(filename=filename, resolution=50)
		jpg = pdf.convert('jpg')
		req_image = []
		for img in jpg.sequence:
			if i == 0:
				img_page = Image(image=img)
				req_image.append(img_page.make_blob('jpg'))
			i = i+1

		for img in req_image:
			ff = open(filename + '.jpg','wb')
			ff.write(img)
			ff.close()
				
		self.out('yes', Demeter.config)
		return
		import csv,codecs
		from datetime import *
		file_name =  'test.csv'
		day = str(date.today())
		day = day.split('-')
		file_path = day[0] + '/' + day[1] + '/' + day[2]
		file_path = File.mkdirs(os.path.join(Demeter.path, 'runtime','upload', file_path)) + '/' + file_name

		data = [{'data': {'rows': [['2017-11-01 00:02:27', 10.95], ['2017-11-01 00:07:27', 9.45], ['2017-11-01 00:12:27', 9.4], ['2017-11-01 00:17:27', 9.45], ['2017-11-01 00:22:27', 9.45], ['2017-11-01 00:27:27', 9.35], ['2017-11-01 00:32:27', 9.35], ['2017-11-01 00:37:27', 9.25], ['2017-11-01 00:42:27', 9.3], ['2017-11-01 00:47:27', 10.55], ['2017-11-01 00:52:27', 9.55], ['2017-11-01 00:57:31', 9.25]], 'value': '[10.95, 9.45, 9.4, 9.45, 9.45, 9.35, 9.35, 9.25, 9.3, 10.55, 9.55, 9.25]', 'time': '["2017-11-01 00:02:27", "2017-11-01 00:07:27", "2017-11-01 00:12:27", "2017-11-01 00:17:27", "2017-11-01 00:22:27", "2017-11-01 00:27:27", "2017-11-01 00:32:27", "2017-11-01 00:37:27", "2017-11-01 00:42:27", "2017-11-01 00:47:27", "2017-11-01 00:52:27", "2017-11-01 00:57:31"]'}, 'name': '\xe6\x98\x8e\xe7\xbb\x86'}]

		data1 = [{'data': {'rows': [['2017-11-01 00:07:27', 3.45], ['2017-11-01 00:12:27', 4.4], ['2017-11-01 00:17:27', 6.45], ['2017-11-01 00:22:27', 5.45], ['2017-11-01 00:27:27', 5.35], ['2017-11-01 00:32:27', 3.35], ['2017-11-01 00:37:27', 3.25], ['2017-11-01 00:42:27', 4.3], ['2017-11-01 00:47:27', 3.55], ['2017-11-01 00:52:27', 4.55], ['2017-11-01 00:57:31', 5.25]], 'value': '[10.95, 9.45, 9.4, 9.45, 9.45, 9.35, 9.35, 9.25, 9.3, 10.55, 9.55, 9.25]', 'time': '["2017-11-01 00:02:27", "2017-11-01 00:07:27", "2017-11-01 00:12:27", "2017-11-01 00:17:27", "2017-11-01 00:22:27", "2017-11-01 00:27:27", "2017-11-01 00:32:27", "2017-11-01 00:37:27", "2017-11-01 00:42:27", "2017-11-01 00:47:27", "2017-11-01 00:52:27", "2017-11-01 00:57:31"]'}, 'name': '\xe6\x98\x8e\xe7\xbb\x86'}]

		datad = {}
		for t,v in data[0]['data']['rows']:
			if t not in datad:
				datad[t] = [t,v]
			else:
				datad[t].append(v)

		newdata = sorted(datad.keys())
		newds = []
		for i in newdata:
			newds.append(datad[i])

		datad = newds
		header = ['日期', '温度', '湿度']
		datad.insert(0, header)

		date = '日期'
		datas = [[date, 'dfdf'],
				['Bob', 14],
				['Tom', 23],
				['Jerry', '18']]

		#self.set_header("Content-Type","text/csv,charset=UTF-8")
		#self.set_header('Content-Disposition', 'attachment; filename=' + file_name)

		with open(file_path, 'w') as f:
			f.write(codecs.BOM_UTF8)
			writer = csv.writer(f)
			for row in datad:
				writer.writerow(row)

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

		file_path = url + file_path.replace(Demeter.path + 'runtime', '')

		self.redirect(file_path)
		#self.out('yes', datad)


# 请求转换 /main/convert 接口必须后端获取，token不允许暴露
#http://192.168.33.10:8088/main/convert?signature=44e3cd684a9fe697792a235c8c57838211f5823a&key=mo1209&nonce=1529659172&file=http%3A%2F%2F192.168.33.10%2Fsystem%2Fdata%2Fupload%2Fdoc_con%2F5b2cbf1064fd8.doc&uid=1
class convert_path(Load):
	@Web.setting
	def get(self):

		file = self.input('file')
		# 源文件id
		file_id = self.input('file_id')
		# 源文件类型 1为收费 2为不收费
		file_type = self.input('file_type', 1)
		# 当前的更新类型，update=insert则为新入库 update=update则为更新，区别是insert要入队
		update = self.input('update', 'insert')

		# 文件流 后续再增加这个吧，要把这个文件流先生成一个本地文件，暂时不做
		#filedata = self.input('filedata')

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

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

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

		# 推入到redis队列
		if file['status'] == 1 and update == 'insert':
			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'],
			'file' : 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'],
			'pic' : url + file['url'] + '.jpg',
			'page' : file['page'],
			'ext' : file['ext'],
			'name' : file['name'],
			'key' : file['key'],
			'size' : file['size']
		}

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

# 授权用户可以访问html的接口 后端接口，token不允许暴露
class auth_path(Load):
	@Web.setting
	def get(self):
		file = self.input('file')
		uid = self.input('uid')
		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('未生成文件信息')

		if file:
			service.auth(self.data['site']['id'], uid, file['id'], 2)

		self.out('yes', {'msg':1})

# 读取html 带有权限控制 该接口为前端接口 需要有授权接口 后续实现
class view_path(Load):
	@Web.setting
	def get(self):
		import re

		key = self.input('key')
		nonce = self.input('nonce')
		signature = self.input('signature')
		file = self.input('file')
		uid = self.input('uid')
		page = self.input('page')
		path = self.input('path')
		host = self.input('host')

		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('未生成文件信息')

		if page:
			page = Demeter.path + 'runtime' + path + page
			content = File.readContent(page)
		else:
			user = service.getAuth(self.data['site']['id'], uid, file['id'])

			limit = -1
			if not user and file['file_type'] == 1:
				# 没有文件信息，则有限制
				limit = self.data['site']['page']
			
			path = file['path']
			view = file['html'] + '_p' + str(limit) + '.html'
			if File.exists(view):
				content = File.readContent(view)
				File.write(view, content)
			else:
				url = self.request.protocol + "://" + self.request.host
				static = url + '/static/'
				content = File.readContent(file['html'])
				content = content.replace('src="', 'src="' + static)

				if limit > 0:
					pattern = '<div id="pf([0-9][a-zA-Z]+|[a-zA-Z0]+|['+str(limit+1)+'-9]|([1-9][0-9]+))"(.*?)<\/div>'
					content = re.sub(pattern, '', content)

				path = path.replace(Demeter.path + 'runtime', '')
				url = url + '/main/view?path=' + str(path) + '&file=' + file['key'] + '&nonce=' + nonce + '&key=' + key + '&signature=' + signature + '&uid=' + uid + '&page='
				content = content.replace('data-page-url="', 'data-page-url="' + url)

				script = '<script src="https://cdn.bootcss.com/jquery/3.3.1/jquery.min.js"></script>'
				script = script + '<script>document.domain="'+host+'";$(function(){var e = pdf2htmlEX.defaultViewer;var height = e.container.clientHeight;for(var i in e.pages){height = height + e.pages[i].original_height};parent.setHeight(height)})</script>'
				content = content.replace('</body>', script + '</body>')

				style = '<style>#page-container{background-color: white;background-image: none;}#sidebar{background-color: white;background-image: none;}.pf{border:1px solid #bababa;box-shadow:none}</style>'

				content = content.replace('</head>', style + '<head>')

		self.write(content)

# 下载源文件
class down_path(Load):
	@Web.setting
	def get(self):
		key = self.input('key')
		nonce = self.input('nonce')
		signature = self.input('signature')
		file = self.input('file')
		uid = self.input('uid')

		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('未生成文件信息')

		user = service.getAuth(self.data['site']['id'], uid, file['id'])

		if not user and file['file_type'] == 1:
			# 没有文件信息，则不允许下载
			self.out('没有权限下载')

		content = File.readContent(file['local'])
		self.write(content)