|
|
from django.core.management.base import BaseCommand
|
|
|
|
|
|
from djangoblog.spider_notify import SpiderNotify
|
|
|
from djangoblog.utils import get_current_site
|
|
|
from blog.models import Article, Tag, Category
|
|
|
|
|
|
site = get_current_site().domain #ZNY 获取当前站点的域名
|
|
|
|
|
|
|
|
|
class Command(BaseCommand): #ZNY 定义通知百度URL命令类，继承BaseCommand
|
|
|
help = 'notify baidu url' #ZNY 命令帮助信息：通知百度URL
|
|
|
|
|
|
def add_arguments(self, parser): #ZNY 添加命令行参数方法
|
|
|
parser.add_argument( #ZNY 添加数据类型参数
|
|
|
'data_type', #ZNY 参数名称
|
|
|
type=str, #ZNY 参数类型为字符串
|
|
|
choices=[ #ZNY 参数可选值
|
|
|
'all', #ZNY 所有类型
|
|
|
'article', #ZNY 仅文章
|
|
|
'tag', #ZNY 仅标签
|
|
|
'category'], #ZNY 仅分类
|
|
|
help='article : all article,tag : all tag,category: all category,all: All of these') #ZNY 参数帮助信息
|
|
|
|
|
|
def get_full_url(self, path): #ZNY 获取完整URL方法
|
|
|
url = "https://{site}{path}".format(site=site, path=path) #ZNY 构建完整的HTTPS URL
|
|
|
return url #ZNY 返回完整URL
|
|
|
|
|
|
def handle(self, *args, **options): #ZNY 命令处理主方法
|
|
|
type = options['data_type'] #ZNY 从参数获取数据类型
|
|
|
self.stdout.write('start get %s' % type) #ZNY 输出开始获取数据的提示
|
|
|
|
|
|
urls = [] #ZNY 初始化URL列表
|
|
|
if type == 'article' or type == 'all': #ZNY 如果类型是文章或全部
|
|
|
for article in Article.objects.filter(status='p'): #ZNY 遍历所有已发布的文章
|
|
|
urls.append(article.get_full_url()) #ZNY 将文章完整URL添加到列表
|
|
|
if type == 'tag' or type == 'all': #ZNY 如果类型是标签或全部
|
|
|
for tag in Tag.objects.all(): #ZNY 遍历所有标签
|
|
|
url = tag.get_absolute_url() #ZNY 获取标签的相对URL
|
|
|
urls.append(self.get_full_url(url)) #ZNY 转换为完整URL并添加到列表
|
|
|
if type == 'category' or type == 'all': #ZNY 如果类型是分类或全部
|
|
|
for category in Category.objects.all(): #ZNY 遍历所有分类
|
|
|
url = category.get_absolute_url() #ZNY 获取分类的相对URL
|
|
|
urls.append(self.get_full_url(url)) #ZNY 转换为完整URL并添加到列表
|
|
|
|
|
|
self.stdout.write( #ZNY 输出开始通知的提示
|
|
|
self.style.SUCCESS( #ZNY 使用成功样式
|
|
|
'start notify %d urls' % #ZNY 显示要通知的URL数量
|
|
|
len(urls)))
|
|
|
SpiderNotify.baidu_notify(urls) #ZNY 调用百度通知功能推送URL
|
|
|
self.stdout.write(self.style.SUCCESS('finish notify')) #ZNY 输出完成通知的提示
|
