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

@author: Albert
"""
import wikipedia as wk
from bs4 import BeautifulSoup

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

 word=raw_input("Wikipedia Search : ")
 results=wk.search(word)
 for i in enumerate(results):
 print i
 try: 
 key=input("Enter the number : ") 
 except AssertionError:
 key=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()