def post(self, category): if not self.has_permission: return if not self.request.files or 'myfile' not in self.request.files: self.write({"status": "error", "message": "对不起,请选择文件"}) return file_type_list = [] if category == 'music': file_type_list = ['audio/mpeg', 'audio/x-wav', 'audio/mp3'] if not file_type_list: return send_file = self.request.files['myfile'][0] if send_file['content_type'] not in file_type_list: if category == 'music': self.write({ "status": "error", "message": "对不起,仅支持 mp3, wav 格式的音乐文件" }) return if category == 'music': if len(send_file['body']) > 20 * 1024 * 1024: self.write({"status": "error", "message": "对不起,请上传20M以下的音乐文件"}) return user = self.current_user if category == 'music': upload_path = os.path.join(config.upload_path, 'music', get_year(), get_month()) else: return mkdir_p(upload_path) timestamp = str(int(time.time())) + gen_random_str() + '_' + str( user.id) image_format = send_file['filename'].split('.').pop().lower() filename = timestamp + '.' + image_format file_path = os.path.join(upload_path, filename) with open(file_path, 'wb') as f: f.write(send_file['body']) path = '/' + get_relative_path(file_path) if not self.is_ajax: return return self.write({ 'path': path, 'status': "success", 'message': '上传成功', 'category': category, 'content_type': send_file['content_type'], })
def post(self, category): if not self.has_permission: return if not self.request.files or 'myfile' not in self.request.files: self.write({"status": "error", "message": "对不起,请选择文件"}) return file_type_list = [] if category == 'music': file_type_list = ['audio/mpeg', 'audio/x-wav', 'audio/mp3'] if not file_type_list: return send_file = self.request.files['myfile'][0] if send_file['content_type'] not in file_type_list: if category == 'music': self.write({"status": "error", "message": "对不起,仅支持 mp3, wav 格式的音乐文件"}) return if category == 'music': if len(send_file['body']) > 20 * 1024 * 1024: self.write({"status": "error", "message": "对不起,请上传20M以下的音乐文件"}) return user = self.current_user if category == 'music': upload_path = os.path.join(config.upload_path, 'music', get_year(), get_month()) else: return mkdir_p(upload_path) timestamp = str(int(time.time())) + gen_random_str() + '_' + str(user.id) image_format = send_file['filename'].split('.').pop().lower() filename = timestamp + '.' + image_format file_path = os.path.join(upload_path, filename) with open(file_path, 'wb') as f: f.write(send_file['body']) path = '/' + get_relative_path(file_path) if not self.is_ajax: return return self.write({ 'path': path, 'status': "success", 'message': '上传成功', 'category': category, 'content_type': send_file['content_type'], })
class ImgUploadHandler(BaseHandler): @orm.db_session @tornado.web.authenticated @require_admin def post(self, node_id): category = self.get_argument('category', None) node = Node.get(id=node_id) if not node: return self.redirect_next_url() if not self.request.files or 'myimage' not in self.request.files: self.write({"status": "error", "message": "对不起,请选择图片"}) return image_type_list = [ 'image/gif', 'image/jpeg', 'image/pjpeg', 'image/png', 'image/bmp', 'image/x-png' ] icon_type_list = [ 'image/gif', 'image/jpeg', 'image/pjpeg', 'image/png', 'image/bmp', 'image/x-png', 'image/ico .ico', 'image/x-icon' ] send_file = self.request.files['myimage'][0] if category != 'icon' and send_file[ 'content_type'] not in image_type_list: self.write({ "status": "error", "message": "对不起,仅支持 jpg, jpeg, bmp, gif, png\ 格式的图片" }) return if category == 'icon' and send_file[ 'content_type'] not in icon_type_list: self.write({ "status": "error", "message": "对不起,仅支持 ico, jpg, jpeg, bmp, gif, png\ 格式的图片" }) return if len(send_file['body']) > 6 * 1024 * 1024: self.write({"status": "error", "message": "对不起,请上传6M以下的图片"}) return tmp_file = tempfile.NamedTemporaryFile(delete=True) tmp_file.write(send_file['body']) tmp_file.seek(0) try: image_one = Image.open(tmp_file.name) except IOError, error: logging.info(error) logging.info('+' * 30 + '\n') logging.info(self.request.headers) tmp_file.close() self.write({"status": "error", "message": "对不起,此文件不是图片"}) return width = image_one.size[0] height = image_one.size[1] if width < 20 or height < 20 or width > 30000 or height > 30000: tmp_file.close() self.write({ "status": "error", "message": "对不起,请上传长宽在20px~30000px之间的图片!" }) return user = self.current_user upload_path = sys.path[0] + "/static/upload/" + get_year() + '/' +\ get_month() + "/" if not os.path.exists(upload_path): try: os.system('mkdir -p %s' % upload_path) except: pass timestamp = str(int(time.time())) + \ ''.join(random.sample('ZYXWVUTSRQPONMLKJIHGFEDCBAzyxwvutsrqponmlkjihgfedcba', 6)) + '_' + str(user.id) image_format = send_file['filename'].split('.').pop().lower() tmp_name = upload_path + timestamp + '.' + image_format image_one.save(tmp_name) tmp_file.close() path = '/' +\ '/'.join(tmp_name.split('/')[tmp_name.split('/').index("static"):]) del_path = None if category == 'head': del_path = node.head_img node.head_img = path msg = '节点头部背景设置成功' data = {'path': path, 'category': 'head'} elif category == 'icon': image_two = image_one.resize((75, 75), Image.ANTIALIAS) tmp_name2 = upload_path + timestamp + 'x75.' + image_format image_two.save(tmp_name2) try: os.system('rm -f %s%s' % (sys.path[0], path)) except: pass path = '/' +\ '/'.join(tmp_name2.split('/')[tmp_name2.split('/').index("static"):]) del_path = node.icon_img node.icon_img = path msg = '节点图标设置成功' data = {'path': path, 'category': 'icon'} elif category == 'background': del_path = node.background_img node.background_img = path msg = '节点背景设置成功' data = {'path': path, 'category': 'background'} else: msg = '图片上传成功' data = {'path': path} if del_path: try: os.system('rm -f %s%s' % (sys.path[0], del_path)) except: pass return self.send_success_result(msg=msg, **data)
def post(self): if not self.has_permission: return user = self.current_user if not user: return self.redirect_next_url() if self.request.files == {} or 'myimage' not in self.request.files: self.write({"status": "error", "message": "对不起,请选择图片"}) return image_type_list = [ 'image/gif', 'image/jpeg', 'image/pjpeg', 'image/png', 'image/bmp', 'image/x-png' ] send_file = self.request.files['myimage'][0] if send_file['content_type'] not in image_type_list: self.write({ "status": "error", "message": "对不起,仅支持 jpg, jpeg, bmp, gif, png\ 格式的图片" }) return if len(send_file['body']) > 6 * 1024 * 1024: self.write({"status": "error", "message": "对不起,请上传6M以下的图片"}) return tmp_file = tempfile.NamedTemporaryFile(delete=True) tmp_file.write(send_file['body']) tmp_file.seek(0) try: image_one = Image.open(tmp_file.name) except IOError as error: logging.info(error) logging.info('+' * 30 + '\n') logging.info(self.request.headers) tmp_file.close() self.write({"status": "error", "message": "对不起,此文件不是图片"}) return width = image_one.size[0] height = image_one.size[1] if width < 80 or height < 80 or width > 30000 or height > 30000: tmp_file.close() self.write({ "status": "error", "message": "对不起,请上传长宽在80px~30000px之间的图片!" }) return user = self.current_user upload_path = os.path.join(config.upload_path, get_year(), get_month()) mkdir_p(upload_path) timestamp = str(int(time.time())) + gen_random_str() + '_' + str( user.id) image_format = send_file['filename'].split('.').pop().lower() filename = timestamp + '.' + image_format tmp_path = os.path.join(upload_path, filename) image_one.save(tmp_path) tmp_file.close() path = '/' + get_relative_path(tmp_path) category = self.get_argument('category', None) del_path = None if category == 'head': del_path = user.head_img user.head_img = path data = {'path': path, 'category': 'head'} elif category == 'background': del_path = user.background_img user.background_img = path data = {'path': path, 'category': 'background'} else: data = {'path': path, 'category': 'other'} if del_path: remove_file(get_asset_path(del_path)) return self.send_success_result(**data)
def post(self, node_id): category = self.get_argument("category", None) node = Node.get(id=node_id) if not node: return self.redirect_next_url() if not self.request.files or "myimage" not in self.request.files: self.write({"status": "error", "message": "对不起,请选择图片"}) return image_type_list = ["image/gif", "image/jpeg", "image/pjpeg", "image/png", "image/bmp", "image/x-png"] icon_type_list = [ "image/gif", "image/jpeg", "image/pjpeg", "image/png", "image/bmp", "image/x-png", "image/ico .ico", "image/x-icon", ] send_file = self.request.files["myimage"][0] if category != "icon" and send_file["content_type"] not in image_type_list: self.write( { "status": "error", "message": "对不起,仅支持 jpg, jpeg, bmp, gif, png\ 格式的图片", } ) return if category == "icon" and send_file["content_type"] not in icon_type_list: self.write( { "status": "error", "message": "对不起,仅支持 ico, jpg, jpeg, bmp, gif, png\ 格式的图片", } ) return if len(send_file["body"]) > 6 * 1024 * 1024: self.write({"status": "error", "message": "对不起,请上传6M以下的图片"}) return tmp_file = tempfile.NamedTemporaryFile(delete=True) tmp_file.write(send_file["body"]) tmp_file.seek(0) try: image_one = Image.open(tmp_file.name) except IOError as error: logging.info(error) logging.info("+" * 30 + "\n") logging.info(self.request.headers) tmp_file.close() self.write({"status": "error", "message": "对不起,此文件不是图片"}) return width = image_one.size[0] height = image_one.size[1] if width < 20 or height < 20 or width > 30000 or height > 30000: tmp_file.close() self.write({"status": "error", "message": "对不起,请上传长宽在20px~30000px之间的图片!"}) return user = self.current_user upload_path = sys.path[0] + "/static/upload/" + get_year() + "/" + get_month() + "/" if not os.path.exists(upload_path): try: os.system("mkdir -p %s" % upload_path) except: pass timestamp = ( str(int(time.time())) + "".join(random.sample("ZYXWVUTSRQPONMLKJIHGFEDCBAzyxwvutsrqponmlkjihgfedcba", 6)) + "_" + str(user.id) ) image_format = send_file["filename"].split(".").pop().lower() tmp_name = upload_path + timestamp + "." + image_format image_one.save(tmp_name) tmp_file.close() path = "/" + "/".join(tmp_name.split("/")[tmp_name.split("/").index("static") :]) del_path = None if category == "head": del_path = node.head_img node.head_img = path msg = "节点头部背景设置成功" data = {"path": path, "category": "head"} elif category == "icon": image_two = image_one.resize((75, 75), Image.ANTIALIAS) tmp_name2 = upload_path + timestamp + "x75." + image_format image_two.save(tmp_name2) try: os.system("rm -f %s%s" % (sys.path[0], path)) except: pass path = "/" + "/".join(tmp_name2.split("/")[tmp_name2.split("/").index("static") :]) del_path = node.icon_img node.icon_img = path msg = "节点图标设置成功" data = {"path": path, "category": "icon"} elif category == "background": del_path = node.background_img node.background_img = path msg = "节点背景设置成功" data = {"path": path, "category": "background"} else: msg = "图片上传成功" data = {"path": path} if del_path: try: os.system("rm -f %s%s" % (sys.path[0], del_path)) except: pass return self.send_success_result(msg=msg, **data)
def post(self): if not self.has_permission: return user = self.current_user if not user: return self.redirect_next_url() if self.request.files == {} or 'myimage' not in self.request.files: self.write({"status": "error", "message": "对不起,请选择图片"}) return image_type_list = ['image/gif', 'image/jpeg', 'image/pjpeg', 'image/png', 'image/bmp', 'image/x-png'] send_file = self.request.files['myimage'][0] if send_file['content_type'] not in image_type_list: self.write({"status": "error", "message": "对不起,仅支持 jpg, jpeg, bmp, gif, png\ 格式的图片"}) return if len(send_file['body']) > 6 * 1024 * 1024: self.write({"status": "error", "message": "对不起,请上传6M以下的图片"}) return tmp_file = tempfile.NamedTemporaryFile(delete=True) tmp_file.write(send_file['body']) tmp_file.seek(0) try: image_one = Image.open(tmp_file.name) except IOError as error: logging.info(error) logging.info('+' * 30 + '\n') logging.info(self.request.headers) tmp_file.close() self.write({"status": "error", "message": "对不起,此文件不是图片"}) return width = image_one.size[0] height = image_one.size[1] if width < 80 or height < 80 or width > 30000 or height > 30000: tmp_file.close() self.write({"status": "error", "message": "对不起,请上传长宽在80px~30000px之间的图片!"}) return user = self.current_user upload_path = os.path.join(config.upload_path, get_year(), get_month()) mkdir_p(upload_path) timestamp = str(int(time.time())) + gen_random_str() + '_' + str(user.id) image_format = send_file['filename'].split('.').pop().lower() filename = timestamp + '.' + image_format tmp_path = os.path.join(upload_path, filename) image_one.save(tmp_path) tmp_file.close() path = '/' + get_relative_path(tmp_path) category = self.get_argument('category', None) del_path = None if category == 'head': del_path = user.head_img user.head_img = path data = {'path': path, 'category': 'head'} elif category == 'background': del_path = user.background_img user.background_img = path data = {'path': path, 'category': 'background'} else: data = {'path': path, 'category': 'other'} if del_path: remove_file(get_asset_path(del_path)) return self.send_success_result(**data)