Python:Deledpimage.py

#!/usr/bin/python
# -*- coding: utf-8  -*-
"""
Script to remove EDP images in non-article namespaces. This script is currently used on the Chinese wikipedia.
 
Way:
* [[Image:logo.jpg]] --> [[:Image:logo.jpg]]
* [[:Image:logo.jpg]] pass
* Image:logo.jpg in gallery --> [[:Image:logo.jpg]] in gallery end
* logo.jpg(like used in template) --> hide(used <!--logo.jpg-->)
"""
#
# (C) Shizhao, 2008
#
# Distributed under the terms of the MIT license.
#
__version__ = '$Id: deledpimage.py,v 1.2 2008-03-14T21:59:00 Shizhao $'
#
 
"""
commond:
python deledpimage.py
 
删除显示在用户页上的合理使用图像
 
"""
import re, time
import wikipedia,catlib
 
site = wikipedia.getSite()
 
cat = {
    'en': u'Category:Non-free image copyright tags',
    'zh': u'Category:合理使用图像模板',
    }
 
content = {
    'en': u'This Non-free image NOT used in non-article namespaces, see[[Wikipedia:Non-free content#Policy]]',
    'zh': u'不是使用在条目中的非自由版权图像，根据[[Wikipedia:合理使用]]，不能在非条目名字空间展示：\n',
    }
 
msg = {
    'en': u'Robot: Fix EDP image use: [[%s]]',
    'zh': u'Bot修正EDP图像用法：[[%s]]',
    }
 
lcontent=wikipedia.translate(site, content)
category=wikipedia.translate(site, cat)
putmsg=wikipedia.translate(site, msg)
 
#from non-free copyright tag category get all EDPtemplate
templatecat=catlib.Category(site, category)
templatelist = templatecat.articlesList()
 
#from References of EDP template get all non-free images
for tempalte in templatelist:
    images = [page for page in tempalte.getReferences() if page.isImage()]
 
    for image in images :
        imagetitle=image.title()
        imagepage=wikipedia.ImagePage(site,imagetitle)
 
#from imagepage get all usingPages of non-articles
        pimages=[puseimage for puseimage in imagepage.usingPages() if puseimage.namespace()<>0]
        for pimage in pimages:
            ns=pimage.namespace()
            pimagetitle=pimage.title()
            c=u'\nfond an used the image [[%s]] in [[%s]]: ' % (imagetitle, pimagetitle)
 
            text=pimage.get()
 
            if imagetitle not in text:
 
# Not [[Image:]]  namespace
                if imagetitle[6:] in text:
 
                    imagetext = re.search(imagetitle[6:]+'(.*?)(|)',text,re.I).group(0)
                    text = re.sub(imagetitle[6:]+'(.*?)(|)', '<!--'+lcontent+imagetext+' \n-->', text, re.I)
                    wikipedia.output(c+u'remove!!!\nSleep 10 s......')
                    pimage.put(text, putmsg % imagetitle)
                    time.sleep(10)
 
#used [[Image:wiki.png]]  image
            else:
 
                if '[['+imagetitle in text:
                    try:
                        imagetext=re.search('\[\['+imagetitle+'(.*?)\]\]',text,re.I).group(0)
#Image in userpage, imagepage,and all talkpage , [[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('\[\['+imagetitle+'(.*?)\]\]', '<!--'+lcontent+'\n-->'+'[['+':'+imagetitle+']]',text, re.I)
                            wikipedia.output(c+u'FIX!\nSleep 10 s......')
                            pimage.put(text, putmsg % imagetitle)
                            time.sleep(10)
 
#Image in template, categorypage,  remove
                        elif ns==10 or ns==14:
                            text = re.sub('\[\['+imagetitle+'(.*?)(|)\]\]', '<!--'+lcontent+imagetext+'\n-->',text, re.I)
                            wikipedia.output(c+u'Remove!!!\nSleep 10 s......')
                            pimage.put(text, putmsg % imagetitle)
                            time.sleep(10)
                    except:
                        print 'Error'
                        pass
 
#                elif '[[:'+imagetitle in text:
#                    wikipedia.output(c+u'EDP is OK!')
 
#Image in <gallery></gallery>
                else:
                    try:
                        imagetext=re.search(imagetitle+'(.*?)\n',text,re.I).group(0)
                        text = re.sub(imagetitle+'(.*?)\n', '',text, re.I)
                        text=re.sub('</gallery>\n', '</gallery>\n'+'<!--'+lcontent+imagetext+'\n-->\n'+'[[:'+imagetitle+']]\n',text, re.I)
                        wikipedia.output(c+u'FIX <gallery>!\nSleep 10 s......')
                        pimage.put(text, putmsg % imagetitle)
                        time.sleep(10)
                    except:
                        print 'Error'
                        pass