Python_and_the_Web/Scripts/API/wikipedia/main.py at main · Python-World/Python_and_the_Web · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
import wikipedia as wiki

"""
author: Monil [https://www.github.com/MNISAR]

"""


def parse(content, esc):
    """
    type(content) = str
    """
    d = {}
    for c in content.split(esc):
        c = c.split(esc[::-1])
        if len(c) == 1:
            section = c[0]
            content = ""
        elif len(c) == 2:
            section = c[0]
            content = c[1]
        else:
            section = c[0]
            content = esc[::-1].join(c[1:])
        d[section.strip()] = content
    return d


def traversal(content, esc):
    if content != "":
        d = parse(content, esc)
        for k, v in d.items():
            d[k] = traversal(v, esc=esc.strip(" ") + "= ")
        if "" in d:
            del d[""]
        return d
    return None


def get_all_wiki(name):
    search_result = wiki.search(name)
    if len(search_result) == 0:
        d = None
    else:
        page = wiki.page(search_result[0], auto_suggest=True)

        d = {
            "content": traversal(page.content, esc="\n== "),
            "url": page.links,
            "summary": page.summary,
        }
    return d


def main():
    name = input("Enter term to search('' to quit): ")
    while name != "":
        # get name of item to search
        page = get_all_wiki(name)
        parsed = page["content"]
        keys = list(parsed.keys())
        while isinstance(parsed, dict):
            for ind, (topic, content) in enumerate(parsed.items()):
                if content is None:
                    print("\n\n", topic, "\n")
                else:
                    print(ind + 1, ":", topic)
            if len(parsed) == 1:
                parsed = list(parsed.values())[0]
            else:
                tpc_idx = int(input("Enter topic index(-1 to break): "))
                if tpc_idx == -1:
                    break
                parsed = parsed[keys[tpc_idx - 1]]

        print("=" * 20)
        name = input("Enter term to search('' to quit): ")


if __name__ == "__main__":
    main()