new function for header

master
Dorian 2 years ago
parent 39c076b309
commit 28e177059d

@ -1,14 +1,36 @@
import wikipedia import wikipedia
wikipedia.set_lang("en")
# wikipedia # wikipedia
# ------------------------------------------------------------------------ # ------------------------------------------------------------------------
def get_wikipage(pagename): def is_header(line):
# return a couple of header_text and header_level
# or return nothing if line is not a header
line = line.strip()
if line == '':
return
i = 0
while line[i] == '=' and line[-1-i] == '=':
i += 1
if i == 0:
return
header_text = line[i:(-1-i)].strip()
header_level = i
return [header_text, header_level]
def get_wikipage(pagename, lang = "en"):
# get wikipedia page content by name of the page # get wikipedia page content by name of the page
wikipedia.set_lang(lang)
print(pagename) print(pagename)
try: try:
results = wikipedia.search(pagename, results=1, suggestion=False) results = wikipedia.search(pagename, results=1, suggestion=False)
try: try:

Loading…
Cancel
Save