summaryrefslogtreecommitdiff
path: root/helpcontent2/wiki-to-help/mwlib_mods/custom_nfo.py
blob: 9d60fe6c8e6cdc3ea267e805832b4c7671eba6c8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
""" 
Enable looking for a custom nfo.json and siteinfo.json 

Author: Timo Richter
"""
import mwlib.cdbwiki
class WikiDB(mwlib.cdbwiki.WikiDB):
            def __init__(self, dir, prefix='wiki', lang="en"):
                try:
                    super(WikiDB,self).__init__(dir,prefix,lang)
                except(RuntimeError):
                    print("Warning: "+"could not get siteinfo for language %r" % (lang,))
                    print("Please set up a custom siteinfo and nfo.")
                self.nfo = self.get_data("nfo") or self.nfo
                self.siteinfo = self.get_data("siteinfo") or self.siteinfo

def apply():
    mwlib.cdbwiki.WikiDB=WikiDB