def after_insert_listener(mapper, connection, target):
from app.main.scheduler import add_job
add_job(target.id, target.frequency)
from app.models.task_status import TaskStatus
task_status = TaskStatus.__table__
connection.execute(task_status.insert().values(
task_id=target.id, task_name=target.name))
def after_update_listener(mapper, connection, target):
from app.main.scheduler import add_job, remove_job
remove_job(target.id)
from app.models.task_status import TaskStatus
task_status = TaskStatus.__table__
connection.execute(task_status.update().values(
last_status='更新任务成功', last_run=datetime.now(),
task_status='run').where(TaskStatus.task_id == target.id
and TaskStatus.task_type == 'html'))
add_job(target.id, target.frequency)
logger.info('task_{}更新'.format(target.id))
def after_update_listener(mapper, connection, target):
from app.main.scheduler import add_job, remove_job
if target.task_status == 'run':
if target.task_type == 'html':
from app.models.task import Task
task = Task.__table__
select_res = connection.execute(
task.select().where(Task.id == target.task_id))
for t in select_res:
remove_job(target.task_id)
add_job(target.task_id, t[6])
elif target.task_type == 'rss':
from app.models.rss_task import RSSTask
rss_task = RSSTask.__table__
select_res = connection.execute(
rss_task.select().where(RSSTask.id == target.task_id))
for t in select_res:
remove_job(target.task_id, 'rss')
add_job(target.task_id, t[3], 'rss')
else:
remove_job(target.id)
def create_app(config_name):
from config import config, logger
app.config.from_object(config[config_name])
# 中文化
Babel(app)
# 注册flask-login
login.init_app(app)
# bootstrap
bootstrap.init_app(app)
@login.user_loader
def load_user(user_id):
return User.query.get(user_id)
# 注册蓝图
from app.main.views import bp as main_bp
app.register_blueprint(main_bp)
# 注册数据库
db.init_app(app)
# 注册flask-admin
admin.init_app(app,
index_view=AdminIndexView(template='admin/index.html',
url='/'))
# 注册APScheduler
scheduler.init_app(app)
scheduler.start()
# 视图
from app.models.task import Task
from app.models.rss_task import RSSTask
from app.models.mail_setting import MailSetting
from app.models.notification import Notification
from app.models.task_status import TaskStatus
from app.models.user import User
admin.add_view(TaskStatusView(TaskStatus, db.session, name='任务状态'))
admin.add_view(TaskView(Task, db.session, name='网页监控任务管理'))
admin.add_view(RSSTaskView(RSSTask, db.session, name='RSS监控任务管理'))
admin.add_view(NotificationView(Notification, db.session, name='通知方式管理'))
admin.add_view(MailSettingView(MailSetting, db.session, name='系统邮箱设置'))
admin.add_view(UserView(User, db.session, name='账号密码管理'))
with app.test_request_context():
db.create_all()
mail_setting = MailSetting.query.first()
# 插入默认邮箱配置
if mail_setting is None:
mail_setting = MailSetting()
db.session.add(mail_setting)
db.session.commit()
# 初始化账号密码
user = User.query.first()
if user is None:
user = User('admin', 'admin')
db.session.add(user)
db.session.commit()
# 插入默认通知方式
notis = Notification.query.all()
mail_exist = False
wechat_exist = False
if len(notis) != 0:
for noti in notis:
if noti.type == 'mail':
mail_exist = True
if noti.type == 'wechat':
wechat_exist = True
if not mail_exist:
mail_noti = Notification('mail')
db.session.add(mail_noti)
db.session.commit()
if not wechat_exist:
wechat_noti = Notification('wechat')
db.session.add(wechat_noti)
db.session.commit()
# 在系统重启时重启任务
from app.main.scheduler import add_job
task_statuss = TaskStatus.query.all()
count = 0
for task_status in task_statuss:
if task_status.task_status == 'run':
count += 1
task_id = task_status.task_id
if task_status.task_type == 'html':
task = Task.query.filter_by(id=task_id).first()
add_job(task.id, task.frequency)
logger.info('重启task_' + str(task_id))
elif task_status.task_type == 'rss':
task = RSSTask.query.filter_by(id=task_id).first()
add_job(task_id, task.frequency, 'rss')
logger.info('重启task_rss' + str(task_id))
logger.info('重启{}个任务'.format(count))
return app
def create_app(config_name):
from config import config, logger
app.config.from_object(config[config_name])
# 中文化
Babel(app)
# 注册flask-login
login.init_app(app)
# bootstrap
bootstrap.init_app(app)
@login.user_loader
def load_user(user_id):
return User.query.get(user_id)
# 注册蓝图
from app.main.views import bp as main_bp
app.register_blueprint(main_bp)
# 注册数据库
db.init_app(app)
# 注册flask-admin
admin.init_app(
app, index_view=AdminIndexView(template='admin/index.html', url='/'))
# 注册APScheduler
scheduler.init_app(app)
scheduler.start()
# 视图
from app.models.task import Task
from app.models.rss_task import RSSTask
from app.models.mail_setting import MailSetting
from app.models.notification import Notification
from app.models.task_status import TaskStatus
from app.models.user import User
admin.add_view(TaskStatusView(TaskStatus, db.session, name='任务状态'))
admin.add_view(TaskView(Task, db.session, name='网页监控任务管理'))
admin.add_view(RSSTaskView(RSSTask, db.session, name='RSS监控任务管理'))
admin.add_view(NotificationView(Notification, db.session, name='通知方式管理'))
admin.add_view(MailSettingView(MailSetting, db.session, name='系统邮箱设置'))
admin.add_view(UserView(User, db.session, name='账号密码管理'))
with app.test_request_context():
db.create_all()
mail_setting = MailSetting.query.first()
# 插入默认邮箱配置
if mail_setting is None:
mail_setting = MailSetting()
db.session.add(mail_setting)
db.session.commit()
# 初始化账号密码
user = User.query.first()
if user is None:
import random
import string
random_password = ''.join(
random.sample(string.ascii_letters + string.digits, 10))
logger.info('数据库初始化成功，初始密码为' + random_password)
user = User('admin', random_password)
db.session.add(user)
db.session.commit()
# 插入默认通知方式
notis = Notification.query.all()
mail_exist = False
wechat_exist = False
if len(notis) != 0:
for noti in notis:
if noti.type == 'mail':
mail_exist = True
if noti.type == 'wechat':
wechat_exist = True
if not mail_exist:
mail_noti = Notification('mail')
db.session.add(mail_noti)
db.session.commit()
if not wechat_exist:
wechat_noti = Notification('wechat')
db.session.add(wechat_noti)
db.session.commit()
ppid = psutil.Process(os.getppid())
ppid_name = ppid.name()
# 加这一步判断主要是因为，
# 在debug模式下，会启动另外一个线程来自动重载，
# 这样会导致在两个线程中都启动任务，造成重复
if ('python' in ppid_name and
config_name == 'development') or config_name != 'development':
# 在系统重启时重启任务
from app.main.scheduler import add_job
task_statuss = TaskStatus.query.all()
count = 0
for task_status in task_statuss:
if task_status.task_status == 'run':
count += 1
task_id = task_status.task_id
if task_status.task_type == 'html':
task = Task.query.filter_by(id=task_id).first()
add_job(task.id, task.frequency)
logger.info('重启task_' + str(task_id))
elif task_status.task_type == 'rss':
task = RSSTask.query.filter_by(id=task_id).first()
add_job(task_id, task.frequency, 'rss')
logger.info('重启task_rss' + str(task_id))
logger.info('重启{}个任务'.format(count))
return app
