i just come out and take a bubble, haha \O^O/
This commit is contained in:
urinx
Executable
+278
@@ -0,0 +1,278 @@
|
||||
#!/usr/bin/env python
|
||||
# coding: utf-8
|
||||
|
||||
#===================================================
|
||||
from wechat import WeChat
|
||||
from wechat.utils import *
|
||||
from wx_handler import WeChatMsgProcessor
|
||||
from wx_handler import Bot
|
||||
from db import SqliteDB
|
||||
from db import MysqlDB
|
||||
from config import ConfigManager
|
||||
from config import Constant
|
||||
from config import Log
|
||||
#---------------------------------------------------
|
||||
import flask
|
||||
import threading
|
||||
import traceback
|
||||
import os
|
||||
import logging
|
||||
import time
|
||||
#===================================================
|
||||
|
||||
|
||||
cm = ConfigManager()
|
||||
db = SqliteDB(cm.getpath('database'))
|
||||
# db = MysqlDB(cm.mysql())
|
||||
wechat_msg_processor = WeChatMsgProcessor(db)
|
||||
wechat = WeChat(cm.get('wechat', 'host'))
|
||||
wechat.db = db
|
||||
wechat.bot = Bot()
|
||||
wechat.msg_handler = wechat_msg_processor
|
||||
wechat_msg_processor.wechat = wechat
|
||||
|
||||
PORT = int(cm.get('setting', 'server_port'))
|
||||
app = flask.Flask(__name__)
|
||||
app.config['UPLOAD_FOLDER'] = cm.getpath('uploaddir')
|
||||
|
||||
logger = logging.getLogger('werkzeug')
|
||||
log_format_str = Constant.SERVER_LOG_FORMAT
|
||||
formatter = logging.Formatter(log_format_str)
|
||||
flask_log_handler = logging.FileHandler(cm.getpath('server_log_file'))
|
||||
flask_log_handler.setLevel(logging.INFO)
|
||||
flask_log_handler.setFormatter(formatter)
|
||||
logger.addHandler(flask_log_handler)
|
||||
app.logger.addHandler(flask_log_handler)
|
||||
|
||||
|
||||
@app.route('/')
|
||||
def index():
|
||||
with open(Constant.SERVER_PAGE_INDEX, 'r') as f:
|
||||
return f.read()
|
||||
|
||||
|
||||
@app.route('/qrcode')
|
||||
def qrcode():
|
||||
qdir = cm.getpath('qrcodedir')
|
||||
if not os.path.exists(qdir):
|
||||
os.makedirs(qdir)
|
||||
image_path = '%s/%s_%d.png' % (qdir, wechat.uuid, int(time.time()*100))
|
||||
s = wechat.wx_conf['API_qrcode'] + wechat.uuid
|
||||
str2qr_image(s, image_path)
|
||||
return flask.send_file(image_path, mimetype='image/png')
|
||||
|
||||
|
||||
@app.route("/group_list")
|
||||
def group_list():
|
||||
"""
|
||||
@brief list groups
|
||||
"""
|
||||
result = wechat.db.select(Constant.TABLE_GROUP_LIST())
|
||||
j = {'count': len(result), 'group': result}
|
||||
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||
|
||||
|
||||
@app.route('/group_member_list/<g_id>')
|
||||
def group_member_list(g_id):
|
||||
"""
|
||||
@brief list group member
|
||||
@param g_id String
|
||||
"""
|
||||
result = wechat.db.select(Constant.TABLE_GROUP_USER_LIST(), 'RoomID', g_id)
|
||||
j = {'count': len(result), 'member': result}
|
||||
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||
|
||||
|
||||
@app.route('/group_chat_log/<g_name>')
|
||||
def group_chat_log(g_name):
|
||||
"""
|
||||
@brief list group chat log
|
||||
@param g_name String
|
||||
"""
|
||||
result = wechat.db.select(Constant.TABLE_GROUP_MSG_LOG, 'RoomName', g_name)
|
||||
j = {'count': len(result), 'chats': result}
|
||||
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||
|
||||
|
||||
@app.route('/upload', methods=['GET', 'POST'])
|
||||
def upload_file():
|
||||
if flask.request.method == 'POST':
|
||||
def allowed_file(filename):
|
||||
return '.' in filename and \
|
||||
filename.rsplit('.', 1)[1] in Constant.SERVER_UPLOAD_ALLOWED_EXTENSIONS
|
||||
|
||||
j = {'ret': 1, 'msg': ''}
|
||||
|
||||
# check if the post request has the file part
|
||||
if 'file' not in flask.request.files:
|
||||
j['msg'] = 'No file part'
|
||||
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||
|
||||
# if user does not select file, browser also
|
||||
# submit a empty part without filename
|
||||
file = flask.request.files['file']
|
||||
if file.filename == '':
|
||||
j['msg'] = 'No selected file'
|
||||
elif file and allowed_file(file.filename):
|
||||
filename = generate_file_name(file.filename)
|
||||
file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
|
||||
file.save(file_path)
|
||||
j['ret'] = 0
|
||||
j['msg'] = filename
|
||||
else:
|
||||
j['msg'] = 'File type not support'
|
||||
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||
else:
|
||||
with open(Constant.SERVER_PAGE_UPLOAD, 'r') as f:
|
||||
return f.read()
|
||||
|
||||
|
||||
@app.route('/send_msg/<to>/<msg>')
|
||||
def send_msg(to, msg):
|
||||
"""
|
||||
@brief send message to user or gourp
|
||||
@param to: String, user id or group id
|
||||
@param msg: String, words
|
||||
"""
|
||||
j = {'ret': 0 if wechat.send_text(to, msg) else 1}
|
||||
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||
|
||||
|
||||
@app.route('/send_img/<to>/<img>')
|
||||
def send_img(to, img):
|
||||
"""
|
||||
@brief send image to user or gourp
|
||||
@param to: String, user id or group id
|
||||
@param img: String, image file name
|
||||
"""
|
||||
img_path = os.path.join(app.config['UPLOAD_FOLDER'], img)
|
||||
j = {'ret': 0 if wechat.send_img(to, img_path) else 1}
|
||||
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||
|
||||
|
||||
@app.route('/send_emot/<to>/<emot>')
|
||||
def send_emot(to, emot):
|
||||
"""
|
||||
@brief send emotion to user or gourp
|
||||
@param to: String, user id or group id
|
||||
@param emot: String, emotion file name
|
||||
"""
|
||||
emot_path = os.path.join(app.config['UPLOAD_FOLDER'], emot)
|
||||
j = {'ret': 0 if wechat.send_emot(to, emot_path) else 1}
|
||||
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||
|
||||
|
||||
@app.route('/send_file/<to>/<file>')
|
||||
def send_file(to, file):
|
||||
"""
|
||||
@brief send file to user or gourp
|
||||
@param to: String, user id or group id
|
||||
@param file: String, file name
|
||||
"""
|
||||
file_path = os.path.join(app.config['UPLOAD_FOLDER'], file)
|
||||
j = {'ret': 0 if wechat.send_file(to, file_path) else 1}
|
||||
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||
|
||||
|
||||
def mass_send(method, data, func):
|
||||
j = {'ret': -1, 'unsend_list':[]}
|
||||
if method == 'POST' and data:
|
||||
to_list = data['to_list']
|
||||
msg = data['msg']
|
||||
media_type = data.get('media_type', '')
|
||||
|
||||
if media_type in ['img', 'emot']:
|
||||
file_path = os.path.join(app.config['UPLOAD_FOLDER'], msg)
|
||||
response = wechat.webwxuploadmedia(file_path)
|
||||
if response is not None:
|
||||
msg = response['MediaId']
|
||||
elif media_type == 'file':
|
||||
file_path = os.path.join(app.config['UPLOAD_FOLDER'], msg)
|
||||
data = {
|
||||
'appid': Constant.API_WXAPPID,
|
||||
'title': msg,
|
||||
'totallen': '',
|
||||
'attachid': '',
|
||||
'type': wechat.wx_conf['APPMSGTYPE_ATTACH'],
|
||||
'fileext': msg.split('.')[-1],
|
||||
}
|
||||
response = wechat.webwxuploadmedia(file_path)
|
||||
if response is not None:
|
||||
data['totallen'] = response['StartPos']
|
||||
data['attachid'] = response['MediaId']
|
||||
else:
|
||||
Log.error('File upload error')
|
||||
msg = data
|
||||
|
||||
for groups in split_array(to_list, 20):
|
||||
for g in groups:
|
||||
r = func(g, msg)
|
||||
if not r:
|
||||
j['unsend_list'].append(g)
|
||||
time.sleep(1)
|
||||
|
||||
j['ret'] = len(j['unsend_list'])
|
||||
|
||||
return j
|
||||
|
||||
|
||||
@app.route('/mass_send_msg/', methods=["GET", "POST"])
|
||||
def mass_send_msg():
|
||||
"""
|
||||
@brief send text to mass users or gourps
|
||||
"""
|
||||
j = mass_send(flask.request.method, flask.request.json, wechat.send_text)
|
||||
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||
|
||||
|
||||
@app.route('/mass_send_img', methods=["GET", "POST"])
|
||||
def mass_send_img():
|
||||
"""
|
||||
@brief send iamge to mass users or gourps
|
||||
"""
|
||||
j = mass_send(flask.request.method, flask.request.json, wechat.webwxsendmsgimg)
|
||||
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||
|
||||
|
||||
@app.route('/mass_send_emot', methods=["GET", "POST"])
|
||||
def mass_send_emot():
|
||||
"""
|
||||
@brief send emoticon to mass users or gourps
|
||||
"""
|
||||
j = mass_send(flask.request.method, flask.request.json, wechat.webwxsendemoticon)
|
||||
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||
|
||||
|
||||
@app.route('/mass_send_file', methods=["GET", "POST"])
|
||||
def mass_send_file():
|
||||
"""
|
||||
@brief send file to mass users or gourps
|
||||
"""
|
||||
j = mass_send(flask.request.method, flask.request.json, wechat.webwxsendappmsg)
|
||||
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||
|
||||
|
||||
def run_server():
|
||||
app.run(port=PORT)
|
||||
|
||||
if cm.get('setting', 'server_mode') == 'True':
|
||||
serverProcess = threading.Thread(target=run_server)
|
||||
serverProcess.start()
|
||||
|
||||
while True:
|
||||
try:
|
||||
wechat.start()
|
||||
except KeyboardInterrupt:
|
||||
echo(Constant.LOG_MSG_QUIT)
|
||||
wechat.exit_code = 1
|
||||
else:
|
||||
Log.error(traceback.format_exc())
|
||||
finally:
|
||||
wechat.stop()
|
||||
|
||||
if wechat.exit_code == 0:
|
||||
echo(Constant.MAIN_RESTART)
|
||||
else:
|
||||
# kill process
|
||||
os.system(Constant.LOG_MSG_KILL_PROCESS % os.getpid())
|
||||
|
||||
Reference in New Issue
Block a user