Aller au contenu

Utilisateur:Phe/Scripts/create dl.py

La bibliothèque libre.
# -*- coding: utf-8 -*-

import botpywi
import query_ext
import sys
import re
import wikipedia
import locale

AL_BY_VOLUME = 1
AL_ALPHA = 2

def transform_link(match, template, blacklist):
    if not match.group(2) in blacklist:
        text = match.group(1) + u' {{' + template + u'|' + match.group(2).upper() + u'|{{sc|' + match.group(2) + u'}}}}'
    else:
        text = match.group(1) + u' {{sc|' + match.group(2) + u'}}'
    if len(match.groups()) > 3 and match.group(3):
        for it in re.finditer(u', {{sc\|([^}]*?)}}', match.group(3)):
            text += u', {{' + template + u'|' + it.group(1).upper() + u'|{{sc|' + it.group(1) + u'}}}}'
    return text

# u"\\1 {{Tr6L|\\2|{{sc|\\2}}}}"
def transform_trevoux_link(match):
    blacklist = set( [
            u'Le P. Mabillon', u'Maty', u'Ducange', u'Durand', u'Pasquier',
            u'Lymnæus'
            ] )

    return transform_link(match, u'Tr6L', blacklist)

def transform_michaud_link(match):
    return transform_link(match, u'Mich2L', set())

Trevoux_1771_data = {
    u'prefix dl' : u'Utilisateur:Phe/Dictionnaire et encyclopédie/Trévoux',
    u'prefix article' : u'Utilisateur:Phe/Dictionnaire et encyclopédie/Trévoux',
    u'need dl prefix code' : True,
    u'need article list' : AL_BY_VOLUME | AL_ALPHA,
    u'suffix article list alpha' : u'Index alphabétique',
    u'cat article' : u'Articles du Dictionnaire de Trévoux, 6e édition',
    u'create link' : { u'regexp' : [ u"(Voyez|''Voyez'') {{sc\|([^}]*?)}}((, {{sc\|([^}]*?)}})*)", transform_trevoux_link ], u'skip regexp' : u'{{[Tt]r6l[ ]*\|' },
    u'error page' : u'Discussion utilisateur:Phe/Dictionnaire et encyclopédie/Trévoux',
}

Encyclo_diderot_1_data = {
    u'prefix dl' : u'L’Encyclopédie/1re édition',
    u'prefix article' : u'L’Encyclopédie/1re édition',
    u'need dl prefix code' : True,
    u'need article list' : AL_BY_VOLUME | AL_ALPHA,
    u'suffix article list alpha' : u'Index alphabétique',
    u'cat article' : u'L’Encyclopédie, 1re édition',
    u'error page' : u'Discussion:L’Encyclopédie/1re édition/Erreur',
    #u'cat article regexp' : [ (u'([Vv]ille|riviere|île\b)', u'Géographie'), (u'Norwege', u'Norvège') , (u'\bHist\.', u'Histoire'), (u'Géog\.', u'Géographie'), (u'Art méchaniq', u'Mécanique'), (u'\barbres\b', u'Botanique'), (u'Jurisp', u'Jurisprudence'), (u'Myth', u'Mythologie'), (u'Arith\.', u'Mathématiques'), (u'Comm\.', u'Commerce'), ('Theol\.', u'Théologie'), (u'Œconomie.', u'Économie'), (u'maladie', u'Médecine'), (u'Mathématiciens', u'Mathématiques'), (u'Botan', u'Botanique'), (u'Suede', u'Suède') ],
    #u'cat article regexp self' : [ u'(France|Allemagne|Italie|Suisse|Angleterre|Espagne|Afrique|Musique|Astronomie|Chimie|Marine|Physiologie|Philosophie)',  ],
    u'cat article module' : 'Diderot_cat',
    u'if not cated' : [ u'à revoir' ],
    u'signature' : [ ( u"\(''A''\)", u"Antoine-Gaspard Boucher d’Argis|Boucher d’Argis"), (u"\(''a''\)", u"Nicolas Lenglet Du Fresnoy|Du Fresnoy"), (u"\(''B''\)", u"Louis de Cahusac|Cahusac"), (u"\(''C''\)", u"Jean Pestré|Pestré"), (u"\(''c''\)", u"Louis Jean-Marie Daubenton|Louis Daubenton"), (u"\(''D''\)", u"Louis-Jacques Goussier|Goussier"), (u"\(''d''\)", u"Arnulphe d’Aumont|Aumont"), (u"\(''E''\)", u"Jean-Baptiste de La Chapelle|La Chapelle"), (u"\(''e''\)", u"Claude Bourgelat|Bourgelat"), (u"\(''F''\)", u"César Chesneau Dumarsais|Dumarsais"), (u"\(''f''\)", u"Jacques-François de Villiers|Villiers"), (u"\(''G''\)", u"Edme-François Mallet|Mallet"), (u"\(''g''\)", u"Paul-Joseph Barthez|Barthez"), (u"\(''H''\)", u"François-Vincent Toussaint|Toussaint"), (u"\(''h''\)", u"André Morellet|Morellet"), (u"\(''I''\)", u"Pierre Daubenton|Daubenton"), (u"\(''K''\)", u"Antoine Joseph Dezallier d’Argenville|Argenville"), (u"\(''L''\)", u"Pierre Tarin|Tarin"), (u"\(''M''\)", u"Paul-Jacques Malouin|Malouin"), (u"\(''m''\)", u"Jean-Joseph Menuret|Menuret"), (u"\(''N''\)", u"Urbain de Vandenesse|Vandenesse"), (u"\(''O''\)", u"Jean le Rond d’Alembert|d’Alembert"), (u"\(''P''\)", u"Jacques-François Blondel|Blondel"), (u"\(''Q''\)", u"Guillaume Le Blond|Le Blond"), (u"\(''R''\)", u"Paul Landois|Landois"), (u"\(''S''\)", u"Jean-Jacques Rousseau|Rousseau"), (u"\(''T''\)", u"Charles Le Roy|Le Roy"), (u"\(''V''\)", u"Marc-Antoine Eidous|Eidous"), (u"\(''X''\)", u"Claude Yvon|Yvon"), (u"\(''Y''\)", u"Antoine Louis|Louis"), (u"\(''Z''\)", u"Jacques-Nicolas Bellin|Bellin"), (u"\(''C\. D\. J\.''\)", u"Louis de Jaucourt|Jaucourt"), (u"\(''D\. J\.''\)", u"Louis de Jaucourt|Jaucourt"), (u"\(''C\. D\. J''\.\)", u"Louis de Jaucourt|Jaucourt"), (u"\(''D\. J''\.\)", u"Louis de Jaucourt|Jaucourt"), (u"\(''D\. J\.''\)", u"Louis de Jaucourt|Jaucourt"), (u"\(''Le chevalier {{sc\|de Jaucourt}}''\.\)", u"Louis de Jaucourt|Jaucourt"), (u"\(''Le Chevalier {{sc\|de Jaucourt}}''\.\)", u"Louis de Jaucourt|Jaucourt"), (u"\(''D\.J''\.\)", u"Louis de Jaucourt|Jaucourt"), (u"\({{sc\|de Jaucourt}}\)", u"Louis de Jaucourt|Jaucourt"), (u"{{sc\|de Jaucourt}}", u"Louis de Jaucourt|Jaucourt"),(u"{{sc\|de Jaucourt\.}}", u"Louis de Jaucourt|Jaucourt"),(u"\(''D\. J''\)", u"Louis de Jaucourt|Jaucourt"), (u"\(''D J''\.\)", u"Louis de Jaucourt|Jaucourt"), (u"\({{sc\|de Jaucourt\.}}\)", u"Louis de Jaucourt|Jaucourt"), (u"\(''Le Chevalier ''\.\)", u"Louis de Jaucourt|Jaucourt"), (u"\(''C\. D\. J\.''\)", u"Louis de Jaucourt|Jaucourt"), (u"\(''\D. J''\.\)", u"Louis de Jaucourt|Jaucourt"), (u"\(''E\. R\. M\.''\)", u"[[Auteur:Jacques-Philippe-Augustin Douchet|Douchet]] & [[Auteur:Nicolas Beauzée|Beauzée]]"), (u"<nowiki>\*</nowiki>", u"Denis Diderot|Diderot"), (u"{{sc\|\*[^}]*}}", u"Denis Diderot|Diderot"), (u"\(—\)", u"Paul Henri Thiry d’Holbach|Baron d’Holbach"), (u"\(''b''\)", u"Gabriel François Venel|Venel") ],
}

Michaud_1843_data = {
    u'prefix dl' : u'Biographie universelle ancienne et moderne/2e éd., 1843',
    U'prefix article' : u'Biographie universelle ancienne et moderne/2e éd., 1843',
    u'need dl prefix code' : True,
    u'need article list' : AL_BY_VOLUME | AL_ALPHA,
    u'suffix article list alpha' : u'Index alphabétique',
    u'cat article' : u'Articles de la Biographie universelle, Michaud, 2e édition',
    u'create link' : { u'regexp' : [ u"(voy\.|''voy\.''|''Voy\.''|Voy\.|Voyez|''Voyez'') {{sc\|([^}]*?)}}((, {{sc\|([^}]*?)}})*)", transform_michaud_link ], u'skip regexp' : u'{{[Mm]ich2L[ ]*\|' },
    u'error page' : u'Discussion:Biographie universelle ancienne et moderne/2e éd., 1843/Erreurs',
}

Celestin_port_data = {
    U'prefix article' : u'Dictionnaire historique, géographique et biographique du Maine-et-Loire',
    u'need dl prefix code' : True,
    u'need article list' : AL_BY_VOLUME | AL_ALPHA,
    u'suffix article list alpha' : u'Index alphabétique',
    u'cat article' : u'Articles du Célestin Port',
    u'create link' : { u'regexp' : [ u"(voy\.|''voy\.''|''Voy\.''|Voy\.|Voyez|''Voyez'') {{sc\|([^}]*?)}}((, {{sc\|([^}]*?)}})*)", transform_michaud_link ], u'skip regexp' : u'{{[Mm]ich2L[ ]*\|' },
    u'error page' : u'Discussion:Dictionnaire historique, géographique et biographique du Maine-et-Loire/Erreurs',
}

page_prefix = {
    u'fr' : u'Page',
}

dl_prefix_code = {}

dl_prefix_code[u'fr'] = u"""{| style="width:100%%;"
|
<div id="dynamic_links" class="nopopups" title="%s" style="height:800px;overflow:auto;margin-left:3em;text-align:left;" >
"""

dl_suffix_code = {}

dl_suffix_code[u'fr'] = u"""</div>
|
<div style="padding:10px;height:800px;overflow:auto;">
<div id="dict_entry" class=text "/>
</div>
|}"""

index_prefix_code = u'<div style="text-align:left;">\n{{colonnes|nombre=4|\n1=\n'
index_suffix_code = u'}}\n</div>'

class VolumeSet(object):
    def __init__(self, descr):
        self.prefix_dl = descr[u'prefix dl']
        self.prefix_article = descr[u'prefix article']
        self.volumes = []
        self.need_dl_prefix_code = descr[u'need dl prefix code']
        self.need_article_list = descr[u'need article list']
        self.suffix_article_list_alpha = descr[u'suffix article list alpha']
        self.cat_article = descr[u'cat article']
        self.create_link = descr.get(u'create link', u'')
        self.error_page = descr[u'error page']
        self.cat_article_regexp = descr.get(u'cat article regexp', [])
        self.cat_article_regexp_self = descr.get(u'cat article regexp self', [])
        self.cat_article_module = descr.get(u'cat article module', '')
        self.cat_article_module_re = {}
        if self.cat_article_module:
            module = __import__(self.cat_article_module)
            for key in module.cat:
                result = []
                for pat in module.cat[key]:
                    result.append(botpywi.quote_regex(pat))
                regexp = u"|".join(result)
                regexp = regexp.replace(u' ', u'\\s')
                regexp = regexp.replace(u"'", u"'*")
                self.cat_article_module_re[key] = re.compile(regexp)
        self.signature = []
        signature = descr.get(u'signature', [])
        for sign, author in signature:
            self.signature.append( (re.compile(sign), author) )
        self.if_not_cated = descr.get(u'if not cated')
        self.error = u''

class Volume(object):
    def __init__(self, descr):
        self.suffix_dl = descr[u'suffix dl']
        self.suffix_article_list = descr[u'suffix article list']
        self.prefix_page = descr[u'prefix page']
        self.create_link = descr.get(u'create link', u'')
        self.pages_text = {}

class Trevoux_1771(VolumeSet):
    def __init__(self):
        VolumeSet.__init__(self, Trevoux_1771_data)
        for i in [ (u'I', 1) ]: #, (u'II', 2), (u'III', 3), (u'IV', 3), (u'V', 4), (u'VI', 5), (u'VII', 6), (u'VIII', 8)]:
            descr_volume = {
                u'suffix dl' : u'Tome %d' % i[1],
                u'suffix article list' : u'Index tome %s' % i[0],
                u'prefix page' : u'Dictionnaire de Trévoux, 1771, %s.djvu' % i[0],
                }
            self.volumes.append(Volume(descr_volume))

class Encyclo_diderot_1(VolumeSet):
    def __init__(self):
        VolumeSet.__init__(self, Encyclo_diderot_1_data)
        for i in range(1, 18):
            basename = u'Diderot - Encyclopedie 1ere edition tome %d.djvu' % i
            descr_volume = {
                u'suffix dl' : u'Volume %d' % i,
                u'suffix article list' : u'Index tome %d' % i,
                u'prefix page' : basename,
                }
            self.volumes.append(Volume(descr_volume))

class Michaud_1843(VolumeSet):
    def __init__(self):
        VolumeSet.__init__(self, Michaud_1843_data)
        for i in range(1, 3):
            basename = u'Michaud - Biographie universelle ancienne et moderne - 1843 - Tome %d.djvu' % i
            descr_volume = {
                u'suffix dl' : 'Tome %d' % i,
                u'suffix article list' : u'Index tome %d' % i,
                u'prefix page' : basename,
                }
            self.volumes.append(Volume(descr_volume))

predef_dict = {
    u'Trévoux 1771' : Trevoux_1771(),
    u'Encyclopédie Diderot 1' : Encyclo_diderot_1(),
    u'Michaud 1843' : Michaud_1843(),
}

class Statistics:
    def __init__(self):
        self.estimed_article = 0
        self.total_article = 0
        self.cated_article = 0
        self.authored_article = 0
        self.total_cat = 0
        self.total_authors = 0

    def ratio(self, total, b):
        if total == 0 or b == 0:
            return 0.0
        return 1.0 - ((float(total) - b) / total)

    def __str__(self):
        text = u'articles: estimed, total, total cat, total authors: %d %d %d %d\n' % (self.estimed_article, self.total_article, self.total_cat, self.total_authors)
        text += u"cated article: %f\n" % self.ratio(self.total_article, self.cated_article)
        text += u"authored article: %f" % self.ratio(self.total_article, self.authored_article)
        return text

class Options:
    def __init__(self):
        self.save_dl = False
        self.save_alpha_index = False
        self.save_volume_index = False
        self.save_article = False
        self.do_links = False

def compare_title(a, b):
    first = int(re.match(u'.*/(\d+)', a[2]).group(1))
    second = int(re.match(u'.*/(\d+)', b[2]).group(1))
    return first - second

def fill_dict_section(volume_set, volume, text, page_nr, dict_section, order, opt):
    for it in re.finditer(u'<section begin=(["]?)(.*?)\\1[ ]*/>', text):
        if not dict_section.has_key(it.group(2)):
            dict_section[it.group(2)] = [ order ]
            order += 1
        elif order - 1 != dict_section[it.group(2)][0]:
            # FIXME: not sufficient if we have twice the same section
            # name consecutively
            volume_set.error += u'# duplicate section [[' + page_prefix[opt.site.code] + u':' + volume.prefix_page + u'/' + unicode(page_nr) + u'|' + unicode(page_nr) + u']], ' + unicode(dict_section[it.group(2)][1]) + u', ' + unicode(dict_section[it.group(2)][-1]) + u', ' + it.group(2) + u'\n'
        dict_section[it.group(2)].append(page_nr)
    return order

def create_link(page, volume_set, volume, opt):
    cl = (volume.create_link or volume_set.create_link) and opt.do_links
    if cl:
        regexp = cl[u'regexp']
        content = page[u'revisions'][0]['*']
        if not re.search(cl[u'skip regexp'], content):
            new_content = re.sub(regexp[0], regexp[1], content)
            if new_content.strip(u'\n') != content.strip(u'\n'):
                save_page(page[u'title'], new_content, opt, old_text = content,
                          comment = u"mise à jour des liens")

def filter_table_entry(entry):
    return entry[2].endswith(u'(nobot)')

def scan_pages(base_page, volume_set, volume, opt):
    gen = query_ext.PreloadingPagesStartswith(base_page, site = opt.site)
    page_list = []
    for p in query_ext.PreloadingContents(gen, site = opt.site):
        page_nr = int(re.match(u'.*/(\d+)', p[u'title']).group(1))
        page_list.append((page_nr, p[u'revisions'][0]['*'], p[u'title']) )
        create_link(p, volume_set, volume, opt)
        volume.pages_text[page_nr] = p[u'revisions'][0]['*']
    page_list.sort(compare_title)

    order = 0
    dict_section = {}
    for p in page_list:
        order = fill_dict_section(volume_set, volume, p[1], p[0],
                                  dict_section, order, opt)

    table = [ (dict_section[key][0], dict_section[key][1:], key, volume.prefix_page) for key in dict_section ]
    table = [ x for x in table if not filter_table_entry(x) ]
    table.sort()
    # format : (order, [page numbers], title, prefix_page)
    return table

def generate_dl(table):
    text = u''
    for p in table:
        text += u'*[[DL#' + unicode(p[1][0]) + u':' + unicode(p[1][-1]) + u'|' + p[2] + u']]\n'
    return text

def get_categories(volume_set, pages_text, stats):
    cats = set()
    for regexp, replace in volume_set.cat_article_regexp:
        if re.search(regexp, pages_text):
            cats.add(replace)
    for regexp in volume_set.cat_article_regexp_self:
        match = re.search(regexp, pages_text)
        if match:
            cats.add(match.group(1))

    for key in volume_set.cat_article_module_re:
        if volume_set.cat_article_module_re[key].search(pages_text):
            cats.add(key)

    cats = [x for x in cats]
    cats.sort(locale.strcoll)
    text = u''
    for c in cats:
        text += u'\n[[Catégorie:Articles de dictionnaire - ' + c + u']]'

    if len(cats):
        stats.cated_article += 1
        stats.total_cat += len(cats)
    else:
        for c in volume_set.if_not_cated:
            text += u'\n[[Catégorie:Articles de dictionnaire - ' + c + u']]'

    return text

def get_authors(volume_set, text, stats):
    authors = set()
    for sign, author in volume_set.signature:
        if sign.search(text):
            if u'[' in author:
                authors.add(author) 
            elif u'|' in author:
                authors.add(u'[[Auteur:' + author + u']]')
            else:
                authors.add(u'[[Auteur:' + author + u'|' + author + u']]')

    if len(authors):
        stats.authored_article += 1
        stats.total_authors += len(authors)

    return u', '.join(authors)

def has_ref(text):
    return re.search(u'<ref>', text)

def generate_article(table, basename, volume_set, volume,
                     idx_volume, opt, stats):
    pagenames = []
    for i, p in enumerate(table):
        pagenames.append({u'title' : basename + u'/' + p[2]})
    pages = {}
    for p in query_ext.PreloadingContents(pagenames, site = opt.site):
        pages[p[u'title']] = p
        
    for i, p in enumerate(table):
        #if i > 10 or p[2] in [ u'ABACE, ABÉCE', u'ABAQUE' ]:
        if idx_volume > 3: # or i <= 2000:
            continue

        stats.total_article += 1

        params = {}
        params[u'index'] = volume.prefix_page
        params[u'from'] = p[1][0]
        params[u'to'] = p[1][-1]
        params[u'from section'] = p[2]
        params[u'to section'] = p[2]
        params[u'prev'] = ''
        if i:
            params[u'prev'] = u'[[../%s/]]' % table[i-1][2]
        else:
            if idx_volume:
                prev_volume = volume_set.volumes[idx_volume - 1]
                if len(prev_volume.table):
                    prev_entry = prev_volume.table[len(prev_volume.table) - 1]
                    params[u'prev'] = u'[[../%s/]]' % prev_entry[2]
        params[u'next'] = ''
        if i < len(table) - 1:
            params[u'next'] = u'[[../%s/]]' % table[i+1][2]
        else:
            if idx_volume < len(volume_set.volumes) - 1:
                next_volume = volume_set.volumes[idx_volume + 1]
                if len(next_volume.table):
                    next_entry = next_volume.table[0]
                    params[u'next'] = u'[[../%s/]]' % next_entry[2]
        ptext  = u'<pages index="%(index)s" from=%(from)d to=%(to)d' % params
        ptext += u' fromsection="%(from section)s"' % params
        ptext += u' tosection="%(to section)s"' % params
        ptext += u' header=1 prev="%(prev)s" next="%(next)s"' % params

        pages_text = u''
        sect = botpywi.quote_regex(p[2])
        sect_begin = u'<section begin=(["]?)%s\\1[ ]*/>' % sect
        sect_end = u'<section end=(["]?)%s\\3[ ]*/>' % sect
        for i in range(params[u'from'], params[u'to'] + 1):
            temp = volume.pages_text[i]
            if i == params[u'from'] or i == params[u'to']:
                temp = re.sub(u'(?ms).*%s(.*)%s.*' % (sect_begin, sect_end),
                              u'\\2', temp)
            pages_text += temp

        article_name = basename + u'/' + p[2]
        old_attrib = []
        if not pages[article_name].has_key(u'missing'):
            old_text = pages[article_name][u'revisions'][0][u'*']
            if re.match(u'(?ms).*<pages[ \n](.*?)/>.*', old_text):
                old_attrib = botpywi.explode_attrib(old_text, u'pages')

        authors = get_authors(volume_set, pages_text, stats)
        for name, attr in old_attrib:
            if name == u'auteur':
                authors = attr
                break
        if authors:
            ptext += u' auteur="%s"' % authors
        ptext += u' />'

        text = ptext

        if has_ref(pages_text):
            text += u'\n----\n<references />'

        if volume_set.cat_article:
            #text += u'\n\n{{DEFAULTSORT:%s}}' % p[2]
            text += u'\n\n[[Catégorie:' + volume_set.cat_article + u']]'

        text += get_categories(volume_set, pages_text, stats)

        if not pages.has_key(article_name):
            volume_set.error += u'# section invalide : "' + p[2] + '"\n'
            continue
        if not pages[article_name].has_key(u'missing'):
            old_text = pages[article_name][u'revisions'][0][u'*']
            if re.match(u'(?ms).*<pages[ \n](.*?)/>.*', old_text):
                attrib = botpywi.explode_attrib(text, u'pages')
                print "existing page with <pages tag:", article_name.encode('utf*8'), "\r",
                if attrib != old_attrib:
                    print "attrib diverges"
                    text = re.sub('(?ms)<pages[\n]*(.*?)/>', ptext, old_text)
                    save_page(article_name, text, opt, old_text = old_text)
            else:
                print "skipping existing page w/o <pages tag:", article_name.encode('utf-8')
        else:
            save_page(article_name, text, opt, old_text = False)

def save_page(title, text, opt, old_text = False, comment = None):
    page = wikipedia.Page(title = title, site = opt.site)
    if old_text:
        if type(old_text) == type(True) and old_text == True:
            if page.exists():
                old_text = page.get()
                if old_text.strip(u'\n') == text.strip(u'\n'):
                    return
            else:
                old_text = u''
        wikipedia.showDiff(old_text, text)
    if not comment:
        comment = u"mise à jour"
    print "saving:", title.encode('utf-8')
    # FIXME: don't save if old_text == text
    page.put_async(text, comment)

# FIXME: share (with a fatal if not found parameter)
def map_letter(letter, title):
    map_l = {
        u'É' : u'E',
        u'Ê' : u'E',
        u'Æ' : u'A',
        u'À' : u'A',
        u'Â' : u'A',
        u'Ç' : u'C',
        u'È' : u'E',
        }
    letter = letter.upper()
    letter = map_l.get(letter, letter)
    if letter in "ABCDEFGHIJKLMNOPQRSTUVWXYZ":
        return letter
    print >> sys.stderr, (u"map_letter, unknown letter: " + letter).encode('utf-8'), title.encode('utf-8')
    return u'A'
    #raise RuntimeError(u"map_letter, unknown letter: " + letter)

def split_alpha(articles):
    result = {}
    for title in articles:
        first_letter = map_letter(title[0], title)
        result.setdefault(first_letter, [])
        result[first_letter].append(title)

    for key in result:
        result[key].sort(locale.strcoll)
    return result

def create_summary(pagename, summary, prefix_code, suffix_code, opt):
    print pagename.encode('utf-8')
    page = wikipedia.Page(title = pagename, site = opt.site)
    if page.exists():
        old_text = page.get()
        text = re.sub(u'(?ms)(<section begin="Sommaire"[ ]*/>\n)(.*)(<section end="Sommaire"[ ]*/>\n)', u'\\1' + summary + u'\\3', old_text)
    else:
        text = u'<div style="text-align:left;">\n{{colonnes|nombre=4|\n1=\n'
        text += u'<section begin="Sommaire"/>\n'
        text += summary
        text += u'<section end="Sommaire"/>\n'
        text += u'}}\n</div>'
    return text

def generate_table_alpha(volume_set, tables, opt):
    articles = []
    for key in tables:
        for p in tables[key][1]:
            articles.append(p[2])

    articles = split_alpha(articles)
    prefix = volume_set.prefix_article + u'/' + volume_set.suffix_article_list_alpha
    for key in articles:
        summary = u''
        # FIXME: assume the alpha index is one level below article
        for title in articles[key]:
            summary += u'*[[../../%s/]]' % title + u'\n'
        pagename = prefix + u'/' + key[0]
        text = create_summary(pagename, summary, index_prefix_code,
                              index_suffix_code, opt)
        save_page(pagename, text, opt, old_text = True)

def generate_table_by_volume(volume_set, volume, table, opt):
    pagename = volume_set.prefix_article + u'/' + volume.suffix_article_list
    page = wikipedia.Page(title = pagename, site = opt.site)
    summary = u''
    for p in table:
        # FIXME: assume the volume index is on the same level as article
        summary += u'* [[../%s/]]' % p[2] + u'\n'
    text = create_summary(pagename, summary, index_prefix_code,
                          index_suffix_code, opt)
    save_page(pagename, text, opt, old_text = True)

def check_article_list(volume_set, tables, opt):
    articles = set()
    for key in tables:
        if volume_set.need_article_list & AL_BY_VOLUME:
            articles.add(tables[key][0].suffix_article_list)
        for d in tables[key][1]:
            articles.add(d[2])

    if volume_set.need_article_list & AL_ALPHA:
        for letter in "ABCDEFGHIJKLMNOPQRSTUVWXYZ":
            articles.add(volume_set.suffix_article_list_alpha + u'/' + letter)

    if volume_set.prefix_dl:
        for volume in volume_set.volumes:
            if volume.suffix_dl:
                articles.add(volume.suffix_dl)

    # so prefix article is not in the set of error
    articles.add(u'')

    print (volume_set.prefix_article + u'/').encode('utf-8')
    # FIXME: wrong, each volume can have its own base article
    extraParams = { u'gapfilterredir' : u'nonredirects' }
    gen = query_ext.PreloadingPagesStartswith(volume_set.prefix_article,
                                              site = opt.site,
                                              extraParams = extraParams)
    for p in gen:
        part = p[u'title'][len(volume_set.prefix_article) + 1:]
        if not part in articles:
            print p[u'title'].encode('utf-8')
            volume_set.error += u'# Article not in list: [[' + p[u'title'] + u']]\n'

def check_duplicate_section(volume_set, tables, opt):
    seen = {}
    for key in tables:
        for d in tables[key][1]:
            if d[2] in seen:
                volume_set.error += u'# duplicate section, [[' + page_prefix[opt.site.code] + u':' + d[3] + u'/' + unicode(d[1][0]) + u'|' + unicode(d[1][0]) + u']], ' + unicode(d[1][-1]) + u' , ' + unicode(d[2]) + u', [[' + page_prefix[opt.site.code] + u':' + seen[d[2]][3] + u'/' + unicode(seen[d[2]][1][0]) + u'|' + unicode(seen[d[2]][1][0]) + u']], ' + unicode(seen[d[2]][1][-1]) + u'\n'
            else:
                seen[d[2]] = d


def do_dict(volume_set, opt, stats):
    tables = {}

    # we need two pass for the prev/next link at volume boundary
    print "first pass, scanning"
    for volume in volume_set.volumes:
        print volume.prefix_page.encode('utf-8')
        base_pages = page_prefix[opt.site.lang] + u':' + volume.prefix_page
        volume.table = scan_pages(base_pages, volume_set, volume, opt)
        tables[volume.prefix_page] = (volume, volume.table)

    print "second pass, generating"
    for idx_volume, volume in enumerate(volume_set.volumes):
        print volume.prefix_page.encode('utf-8')
        base_pages = page_prefix[opt.site.lang] + u':' + volume.prefix_page
        table = volume.table
        stats.estimed_article += len(table)
        if opt.save_dl and volume_set.prefix_dl and volume.suffix_dl:
            text = generate_dl(table)
            out_dl_name = volume_set.prefix_dl + u'/' + volume.suffix_dl
            dl_prefix = dl_suffix = u''
            if volume_set.need_dl_prefix_code:
                dl_prefix = dl_prefix_code[opt.site.lang] % volume.prefix_page
                dl_suffix = dl_suffix_code[opt.site.lang]
            text = create_summary(out_dl_name, text, dl_prefix, dl_suffix, opt)
            save_page(out_dl_name, text, opt, old_text = True)

        if opt.save_volume_index and volume_set.need_article_list & AL_BY_VOLUME:
            generate_table_by_volume(volume_set, volume, table, opt)

        if opt.save_article and volume_set.prefix_article and opt.save_article:
            base_article = volume_set.prefix_article
            generate_article(table, base_article, volume_set, volume,
                             idx_volume, opt, stats)

    if opt.save_alpha_index and volume_set.need_article_list & AL_ALPHA:
        generate_table_alpha(volume_set, tables, opt)

    check_duplicate_section(volume_set, tables, opt)
    check_article_list(volume_set, tables, opt)

    save_page(volume_set.error_page, volume_set.error, opt, old_text = True)

def do_help():
    print sys.argv[0], '-dict|-save_all|-save_dl|-save_alpha_index|-save_volume_index|-save_article|-do_links:'
    exit(1)

if __name__ == "__main__":
    locale.setlocale(locale.LC_COLLATE, 'fr_FR.utf8')
    statistics = Statistics()
    try:
        options = Options()
        options.site = wikipedia.getSite(code = 'fr', fam = 'wikisource')
        for arg in sys.argv[1:]:
            if arg == '-save_all':
                options.save_dl = True
                options.save_alpha_index = True
                options.save_volume_index = True
                options.save_article = True
                options.do_links = True
            elif arg == '-save_dl':
                options.save_dl = True
            elif arg == '-save_alpha_index':
                options.save_alpha_index = True
            elif arg == '-save_volume_index':
                options.save_volume_index = True
            elif arg == '-save_article':
                options.save_article = True
            elif arg == '-do_links':
                options.do_links = True 
            elif arg == '-help':
                do_help()
            elif arg.startswith('-dict'):
                options.dict = unicode(arg[len('-dict:'):], 'utf-8')
            else:
                do_help()
        if not options.dict:
            print 'no dict selected'
            do_help()

        do_dict(predef_dict[options.dict], options, statistics)

        print statistics
    finally:
        wikipedia.stopme()