diff --git a/README.md b/README.md
index d62d83f..9699efa 100644
--- a/README.md
+++ b/README.md
@@ -1,7 +1,3 @@
-
- 
-
-
# wechat-admin
> 微信管理系统
diff --git a/views/api.py b/views/api.py
index 3c41dcc..c87f67c 100644
--- a/views/api.py
+++ b/views/api.py
@@ -1,5 +1,6 @@
# coding=utf-8
import os
+import glob
from flask import Flask, request
from flask.views import MethodView
from flask_sqlalchemy import get_debug_queries
@@ -19,6 +20,12 @@
from models.messaging import Message, Notification
PER_PAGE = 20
+here = os.path.abspath(os.path.dirname(__file__))
+
+try:
+ FileNotFoundError
+except NameError:
+ FileNotFoundError = OSError
class ApiFlask(Flask):
@@ -94,6 +101,11 @@ def login():
@json_api.route('/logout', methods=['post'])
def logout():
_wx_ctx_stack.pop()
+ for f in glob.glob('{}/*.pkl'.format(here)):
+ try:
+ os.remove(f)
+ except FileNotFoundError:
+ pass
return {'msg': ''}