You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
1.8 KiB
55 lines
1.8 KiB
#!/usr/bin/env python
|
|
# encoding: utf-8
|
|
|
|
|
|
"""
|
|
@version: ??
|
|
@author: liangliangyy
|
|
@license: MIT Licence
|
|
@contact: liangliangyy@gmail.com
|
|
@site: https://www.lylinux.org/
|
|
@software: PyCharm
|
|
@file: ping_baidu.py
|
|
@time: 2017/1/17 下午15:29
|
|
"""
|
|
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
from blog.models import Article, Tag, Category
|
|
from DjangoBlog.spider_notify import SpiderNotify
|
|
from django.contrib.sites.models import Site
|
|
|
|
site = Site.objects.get_current().domain
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = 'notify baidu url'
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument('data_type', type=str, choices=['all', 'article', 'tag', 'category'],
|
|
help='article : all article,tag : all tag,category: all category,all: All of these')
|
|
|
|
def get_full_url(self, path):
|
|
url = "https://{site}{path}".format(site=site, path=path)
|
|
return url
|
|
|
|
def handle(self, *args, **options):
|
|
type = options['data_type']
|
|
self.stdout.write('start get %s' % type)
|
|
|
|
urls = []
|
|
if type == 'article' or type == 'all':
|
|
for article in Article.objects.filter(status='p'):
|
|
urls.append(article.get_full_url())
|
|
if type == 'tag' or type == 'all':
|
|
for tag in Tag.objects.all():
|
|
url = tag.get_absolute_url()
|
|
urls.append(self.get_full_url(url))
|
|
if type == 'category' or type == 'all':
|
|
for category in Category.objects.all():
|
|
url = category.get_absolute_url()
|
|
urls.append(self.get_full_url(url))
|
|
|
|
self.stdout.write(self.style.SUCCESS('start notify %d urls' % len(urls)))
|
|
SpiderNotify.baidu_notify(urls)
|
|
self.stdout.write(self.style.SUCCESS('finish notify'))
|