From 28e177059d2ef20cde39027efa8a3184350283fe Mon Sep 17 00:00:00 2001
From: Dorian <interact@dorian.systems>
Date: Fri, 24 Feb 2023 15:34:03 +0100
Subject: [PATCH] new function for header

---
 wikipage/__pycache__/page.cpython-38.pyc | Bin 679 -> 985 bytes
 wikipage/page.py                         |  28 ++++++++++++++++++++---
 2 files changed, 25 insertions(+), 3 deletions(-)

diff --git a/wikipage/__pycache__/page.cpython-38.pyc b/wikipage/__pycache__/page.cpython-38.pyc
index 6e1388abcb66d02d9d74a70334a8bdb323b50710..abfc0840faadb69fb34c3299b7d4c57d1af0e357 100644
GIT binary patch
literal 985
zcmYjQ&2G~`5T04viQ^_vl~A>SxaSg|5ErPBC`fRKDixICOSR1|i4(^*yBiuRn*)?%
zZ=5)_N1lVX@Rd{F0C9m?r=V`MGdr{Mv-!U9>rN*CTHPPNM!5^{(;C;8L&g)@`aTH(
z0uzWpFyWj+grX@t(KyA33EYRCe@QVg7@<{Hi)w=1PigBZi2}~y0!q>XXH0bk>{GoG
z3n*|KghRd^x^oNMg8w!O?;(`<ZmU4SDB`8fvKgzmFKl~Nl@(Yz1uL8}qxUoh6)5|d
z-}w0Q53NgU^sw@@QE}dLD>sQ#Sz#QlO(Yjm9U8ebQTxhB<a?QX2i3AAWH^^1=Ap<`
z%+qj}8y@bdY$``ahdkx6?jK1tnuTT}kGYbQY@sDj1(|O%CAA6VGVdG3Q=M=VXQ>YR
z{qQ)R#W^3#(29Qkp=!nYaMiv+UG*jjaDC>Z%UryNL4E3?i)xK-DkjqrvoTl8BW($-
z;r>Pk1Ls)a1tJ($z-Q7Y0|QgX!f$Cjb4qs7E|_q)iIgFv_gygVEKmak`=flXfD8PD
zZy;p{WKSEE>sjK@oFUOb@i1*peaiWYP&(9MgL*mBmhC2*r8|TNG&-g8(K&!smBO{-
zGEnxp&wH$D5<rW@=qS*O@tD9Od}`)i(WS2lcttoz&0NVO<3icL+zYB!4WFeY^0hRF
z2~WqBm#a86RfEu1qe&IKNQGQJQz}zYYmZP~9XmsdyiK*fDv-;O%xhYrTf4F5^P_mY
zsCla6t7f*k?Rk-Jxk~w5()dtA)uG2!NS5{2!?F^VhpIiMcl~4eE_w7F65tS#jNv*m
zY~u~KjvLs;05_T1qT3#-wpEV0NvdYL_FpS*XHwN2{73UlEE2h6hg;ik&G-D<e*rpl
B?&kmi

delta 443
zcmYjN!A`<J5S`g=r8H2H7@`*so+Ku4(vvs!WWt3g!bKalAuUpx(!@YoG3v>LMAILD
zlkqFOa`wjl0l(nMS=7Xt%<S%)_hxo?k$Kat^L*X}vXA*$`;)y|dt(QtDA)`U1Q*&9
zw3xs;l=Y>e)g*KLHmdEfX|$>i#}tDXN)g2HNa|57LXX8DsGHyup5Q1KqtLb?fd=?H
z8brKnh9?NhBb;h6%<$m?o8Yrc0z93>I{Dg!H=qtAKyS^<kmwm9VKkyh!<c8NaUzot
z>i`5*?nwXBCFXQ>qqfBKo*kAXl(SL~Lf4OE_SWsXfg@ZyU9je^)-nRS<M?*Z3DuJ%
zI#N@}JA~_W{<=>0@pRk|?6w;XtCdQ1r6)pFu`0n(<~mMv^@kejJGLkRH;`e0B8L_$
rU=cUb#1acPs8;5oBDRbHNpEbwvqDz`|D^&^q;b^ii*(K{&P@Fq$XjA$

diff --git a/wikipage/page.py b/wikipage/page.py
index 9dc510f..1e9c046 100644
--- a/wikipage/page.py
+++ b/wikipage/page.py
@@ -1,14 +1,36 @@
 import wikipedia
 
-wikipedia.set_lang("en")
-
 #   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
 
+    wikipedia.set_lang(lang)
+
     print(pagename)
+    
     try:
         results = wikipedia.search(pagename, results=1, suggestion=False)
         try: