#!/usr/bin/python
# -*- coding: utf-8 -*-
"""
commond:
python deledpimage.py -category:category:name
删除显示在用户页上的合理使用图像
"""
import re, time
import wikipedia
site = wikipedia.getSite()
summary = u'remove EDP image from user namespace'
template=''
ns=2
name=[]
#设置参数
for arg in wikipedia.handleArgs():
if arg.startswith('-template:'):
template = arg[10:]
# else:
# template = wikipedia.input(u'plese input EDP images template: ')
elif arg.startswith('-ns:'):
ns=arg[4:]
# else:
# ns=wikipedia.input(u'plese input check namespace: ')
name.append(arg)
#print name
template=name[0][10:]
ns=int(name[1][4:])
#template='logo'
#ns=2
#得到使用该模板的图像
s=wikipedia.Page(site, 'Template:%s' % template)
#print template, ns
images = [page for page in s.getReferences() if page.isImage()]
for image in images:
imagepage=wikipedia.ImagePage(site,image.title())
#使用该图像的页面
pimages=[puseimage for puseimage in imagepage.usingPages() if puseimage.namespace()==ns]
for pimage in pimages:
c=u'\nfond an used the image [[%s]] in [[%s]]: ' % (image.title(), pimage.title())
p = wikipedia.Page(site, pimage.title())
text=p.get()
title=image.title()
# if title in text:
# wikipedia.output(u'YES, YES, YES')
# else:
# wikipedia.output(u'NO, NO, NO')
#例如： title = Image:wiki.png
if title not in text:
# 没有名字空间的图像，常在模板中使用
if title[6:] in text:
text = re.sub(title[6:]+'(.*?)\n', '\n',text, re.I)
pimage.put(text, u'Bot移去EDP图像 [[%s]] in [[%s]]' % (image.title(), pimage.title()))
wikipedia.output(c+u'remove!!!')
else:
#图像已经不在页面上，或者图像使用在页面的模板中
wikipedia.output(c+u'have remove.')
#图像使用[[Image:wiki.png]]语法
else:
if '[['+title in text:
#在用户页，图像页或各对话页使用的图像，将[[Image:wiki.png]]变成[[:Image:wiki.png]]
if ns==1 or ns==6 or ns==2 or ns==3 or ns==5 or ns==7 or ns==9 or ns==11 or ns==13 or ns==15 or ns==17 or ns==101:
text = re.sub('\[\['+title+'(.*?)\]\]', '[['+':'+title+']]',text, re.I)
pimage.put(text, u'Bot修正EDP图像用法: [[:%s]]' % title)
wikipedia.output(c+u'fix!')
#在模板，分类页找到的的图像，删除
elif ns==10 or ns==14:
text = re.sub('\[\['+title+'(.*?)\]\]', '',text, re.I)
pimage.put(text, u'Bot移去EDP图像 [[%s]] in [[%s]]' % (image.title(), pimage.title()))
wikipedia.output(c+u'remove!!!')
#找到[[:Image:wiki.png]]使用的图像
elif '[[:'+title in text:
wikipedia.output(c+u'EDP is OK!')
#在<gallery></gallery>中使用的图像
else:
text = re.sub(title+'(.*?)</gallery>', '</gallery>',text, re.I)
text=re.sub('</gallery>\n', '</gallery>\n'+'[[:'+title+']]\n',text, re.I)
pimage.put(text, u'Bot修正EDP图像<gallery>用法: [[:%s]]' % title)
wikipedia.output(c+u'FIX <gallery>')
