"""
Created on Sat Jul 15 01:41:31 2017

@author: Albert
"""
from __future__ import print_function

import wikipedia as wk
from bs4 import BeautifulSoup


def wiki():
 '''
 Search Anything in wikipedia
 '''

 word = input("Wikipedia Search : ")
 results = wk.search(word)
 for i in enumerate(results):
 print(i)
 try:
 key = int(input("Enter the number : "))
 except AssertionError:
 key = int(input("Please enter corresponding article number : "))

 page = wk.page(results[key])
 url = page.url
 # originalTitle=page.original_title
 pageId = page.pageid
 # references=page.references
 title = page.title
 # soup=BeautifulSoup(page.content,'lxml')
 pageLength = input('''Wiki Page Type : 1.Full 2.Summary : ''')
 if pageLength == 1:
 soup = fullPage(page)
 print(soup)
 else:
 print(title)
 print("Page Id = ", pageId)
 print(page.summary)
 print("Page Link = ", url)
 # print "References : ",references

 pass


def fullPage(page):
 soup = BeautifulSoup(page.content, 'lxml')
 return soup


def randomWiki():
 '''
 This function gives you a list of n number of random articles
 Choose any article.
 '''
 number = input("No: of Random Pages : ")
 lst = wk.random(number)
 for i in enumerate(lst):
 print(i)
 try:
 key = input("Enter the number : ")
 assert key >= 0 and key < number
 except AssertionError:
 key = input("Please enter corresponding article number : ")

 page = wk.page(lst[key])
 url = page.url
 # originalTitle=page.original_title
 pageId = page.pageid
 # references=page.references
 title = page.title
 # soup=BeautifulSoup(page.content,'lxml')
 pageLength = input('''Wiki Page Type : 1.Full 2.Summary : ''')
 if pageLength == 1:
 soup = fullPage(page)
 print(soup)
 else:
 print(title)
 print("Page Id = ", pageId)
 print(page.summary)
 print("Page Link = ", url)
 # print "References : ",references

 pass

# if __name__=="__main__":
# wiki()